This commit is contained in:
ubq323 2021-01-28 21:01:40 +00:00
commit b63b675e95
2 changed files with 21 additions and 9 deletions

View File

@ -12,3 +12,4 @@ bin = @["onstat"]
requires "nim >= 1.4.2"
requires "https://github.com/GULPF/tiny_sqlite#8fe760d9"
requires "karax >= 1.2.1"
requires "cligen >= 1"

View File

@ -11,15 +11,13 @@ import net
import sequtils
import strformat
import std/exitprocs
import cligen
import ./db
macro includeFile(x: string): string = newStrLitNode(readFile(x.strVal))
const css = includeFile("./src/style.css")
let database = openDatabase("./monitoring.sqlite3")
migrate(database)
close(database)
var threadDB {.threadvar.}: Option[DbConn]
proc getDB(): DbConn {.gcsafe.} =
@ -144,8 +142,21 @@ proc timerCallback(fd: AsyncFD): bool =
asyncCheck pollTargets()
false
proc run(dbPath="./monitoring.sqlite3", port=7800, interval=30000, urls: seq[string]) =
## Run onstat. Note that the URLs you configure will be persisted in the monitoring database. To remove them, you must manually update this.
let database = openDatabase(dbPath)
migrate(database)
for url in urls:
echo &"Adding {url}"
database.exec("INSERT INTO sites (url) VALUES (?)", url)
close(database)
echo "Starting up"
asyncCheck pollTargets()
addTimer(30000, false, timerCallback)
addTimer(interval, false, timerCallback)
var server = newAsyncHttpServer()
waitFor server.serve(Port(7800), onRequest)
waitFor server.serve(Port(port), onRequest)
dispatch(run, help={
"dbPath": "path to SQLite3 database for historical data logging",
"port": "port to serve HTTP on",
"interval": "interval at which to poll other services (milliseconds)"
})