yanic/cmd/respond-collector/main.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,
		)
	}
}