diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/db.lua | 22 | ||||
-rwxr-xr-x | src/nopcollectd | 14 | ||||
-rwxr-xr-x | src/nopsysstated | 10 |
3 files changed, 23 insertions, 23 deletions
@@ -69,17 +69,17 @@ function db.init(db, user, pwd, host, port, table) if cur == nil then return nil, err end - + local timestamp = cur:fetch() if timestamp == nil then timestamp = 0 end return timestamp end - + function mydb:findMissingEntries(lasttimestamp) local lastts = self.con:escape(lasttimestamp) return self.con:execute("SELECT * FROM " .. self.table .. " WHERE timestamp > " .. lastts) end - + function mydb:getNextMissingEntry(cur) local data = {} data = cur:fetch(data, "a") @@ -88,14 +88,14 @@ function db.init(db, user, pwd, host, port, table) end return data end - + function mydb:getEntry(timestamp) local ts = self.con:escape(timestamp) local cur, err = self.con:execute("SELECT * FROM " .. self.table .. " WHERE timestamp = " .. ts) if cur == nil then return nil, err end - + local data = {} data = cur:fetch(data, "a") if data == nil then @@ -103,7 +103,7 @@ function db.init(db, user, pwd, host, port, table) end return data end - + function mydb:addEntry(data) local timestamp = self.con:escape(data.timestamp) local cart = self.con:escape(data.cart) @@ -113,21 +113,21 @@ function db.init(db, user, pwd, host, port, table) local artist = self.con:escape(data.artist) local album = self.con:escape(data.album) local ismusic = self.con:escape(data.ismusic) - + local cur, err = self.con:execute("REPLACE INTO " .. self.table .. " VALUES('" .. timestamp .. "', '" .. cart .. "', '" .. len .. "', '" .. showtitle .. "', '" .. title .. "', '" .. artist .."', '" .. album .. "', '" .. ismusic .. "')") if cur == nil then return nil, err end - + return true end - + function mydb:getLastSysState() local cur, err = self.con:execute("SELECT state FROM " .. self.table .. " WHERE timestamp = (SELECT MAX(timestamp) FROM " .. self.table .. ")") if cur == nil then return nil, err end - + local state = cur:fetch() if state == nil then state = "unknown" end return state @@ -148,7 +148,7 @@ function db.init(db, user, pwd, host, port, table) if self.con then self.con:close() end - + if self.env then self.env:close() end diff --git a/src/nopcollectd b/src/nopcollectd index 0f7c66e..933fb81 100755 --- a/src/nopcollectd +++ b/src/nopcollectd @@ -2,7 +2,7 @@ -- -- rhnop -- --- Copyright (C) 2011-2014 Christian Pointner <equinox@helsinki.at> +-- Copyright (C) 2011-2015 Christian Pointner <equinox@helsinki.at> -- -- This file is part of rhnop. -- @@ -25,24 +25,24 @@ -- the arguments should be: now# nowlen next# nextlen -- -require "socket" +socket = require "socket" package.path = package.path .. ";_rhnoplibdir_/?.lua" db = require "db" +conf = require "conf" if #arg < 1 then io.stderr:write("too few parameters\n") os.exit(1) end -conffile = "nopcollectd." .. arg[1] .. ".conf" -conf = require "conf" +local cnf = conf.load("nopcollectd." .. arg[1] .. ".conf") -src_db = assert(db.init(conf.src_db, conf.src_user, conf.src_pwd, conf.src_host, conf.src_port, conf.src_table)) -dst_db = assert(db.init(conf.dst_db, conf.dst_user, conf.dst_pwd, conf.dst_host, conf.dst_port, conf.dst_table)) +src_db = assert(db.init(cnf.src_db, cnf.src_user, cnf.src_pwd, cnf.src_host, cnf.src_port, cnf.src_table)) +dst_db = assert(db.init(cnf.dst_db, cnf.dst_user, cnf.dst_pwd, cnf.dst_host, cnf.dst_port, cnf.dst_table)) local sock = assert(socket.tcp()) -local ret, err = sock:connect(conf.sync_host, conf.sync_port) +local ret, err = sock:connect(cnf.sync_host, cnf.sync_port) if ret == nil then print(err) return 1 diff --git a/src/nopsysstated b/src/nopsysstated index f1a05f8..dc43c99 100755 --- a/src/nopsysstated +++ b/src/nopsysstated @@ -25,18 +25,18 @@ -- the arguments should be: now# nowlen next# nextlen -- -require "socket" +socket = require "socket" package.path = package.path .. ";_rhnoplibdir_/?.lua" db = require "db" - -conffile = "nopsysstated.conf" conf = require "conf" -db = assert(db.init(conf.db_db, conf.db_user, conf.db_pwd, conf.db_host, conf.db_port, conf.db_table)) +local cnf = conf.load("nopsysstated.conf") + +db = assert(db.init(cnf.db_db, cnf.db_user, cnf.db_pwd, cnf.db_host, cnf.db_port, cnf.db_table)) local sock = assert(socket.tcp()) -local ret, err = sock:connect(conf.host, conf.port) +local ret, err = sock:connect(cnf.host, cnf.port) if ret == nil then print(err) return 1 |