78 lines
1.8 KiB
Go
78 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"github.com/FreifunkBremen/respond-collector/database"
|
|
"github.com/FreifunkBremen/respond-collector/models"
|
|
"github.com/FreifunkBremen/respond-collector/respond"
|
|
"github.com/FreifunkBremen/respond-collector/rrd"
|
|
"github.com/FreifunkBremen/respond-collector/webserver"
|
|
)
|
|
|
|
var (
|
|
configFile string
|
|
config *models.Config
|
|
collector *respond.Collector
|
|
db *database.DB
|
|
nodes *models.Nodes
|
|
)
|
|
|
|
func main() {
|
|
var importPath string
|
|
flag.StringVar(&importPath, "import", "", "import global statistics from the given RRD file, requires influxdb")
|
|
flag.StringVar(&configFile, "config", "config.toml", "path of configuration file (default:config.yaml)")
|
|
flag.Parse()
|
|
config = models.ReadConfigFile(configFile)
|
|
|
|
if config.Influxdb.Enable {
|
|
db = database.New(config)
|
|
defer db.Close()
|
|
|
|
if importPath != "" {
|
|
importRRD(importPath)
|
|
return
|
|
}
|
|
}
|
|
|
|
nodes = models.NewNodes(config)
|
|
nodes.Start()
|
|
|
|
if config.Respondd.Enable {
|
|
collector = respond.NewCollector(db, nodes, config.Respondd.Interface)
|
|
collector.Start(config.Respondd.CollectInterval.Duration)
|
|
defer collector.Close()
|
|
}
|
|
|
|
if config.Webserver.Enable {
|
|
log.Println("starting webserver on", config.Webserver.Bind)
|
|
srv := webserver.New(config.Webserver.Bind, config.Webserver.Webroot)
|
|
go srv.Close()
|
|
}
|
|
|
|
// Wait for INT/TERM
|
|
sigs := make(chan os.Signal, 1)
|
|
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
|
|
sig := <-sigs
|
|
log.Println("received", sig)
|
|
}
|
|
|
|
func importRRD(path string) {
|
|
log.Println("importing RRD from", path)
|
|
for ds := range rrd.Read(path) {
|
|
db.AddPoint(
|
|
database.MeasurementGlobal,
|
|
nil,
|
|
map[string]interface{}{
|
|
"nodes": uint32(ds.Nodes),
|
|
"clients.total": uint32(ds.Clients),
|
|
},
|
|
ds.Time,
|
|
)
|
|
}
|
|
}
|