yanic/main.go

package main

import (
	"flag"
	"log"
	"net"
	"net/http"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/ffdo/node-informant/gluon-collector/data"
	"github.com/monitormap/micro-daemon/models"
	"github.com/monitormap/micro-daemon/respond"
	"github.com/monitormap/micro-daemon/websocketserver"
)

var (
	configFile       string
	config           *models.Config
	wsserverForNodes *websocketserver.Server
	multiCollector   *respond.MultiCollector
	nodes            = models.NewNodes()
	//aliases          = models.NewNodes()
)

func main() {
	flag.StringVar(&configFile, "c", "config.yml", "path of configuration file (default:config.yaml)")
	flag.Parse()
	config = models.ConfigReadFile(configFile)

	collectInterval := time.Second * time.Duration(config.Respondd.CollectInterval)

	if config.Nodes.Enable {
		go nodes.Saver(config)
	}
	if config.Nodes.AliasesEnable {
		// FIXME what does this do?
		//go aliases.Saver(config.Nodes.AliasesPath, saveInterval)
	}

	if config.Webserver.Enable {
		if config.Webserver.WebsocketNode {
			wsserverForNodes = websocketserver.NewServer("/nodes")
			go wsserverForNodes.Listen()
		}
		http.Handle("/", http.FileServer(http.Dir(config.Webserver.Webroot)))
	}

	if config.Respondd.Enable {
		multiCollector = respond.NewMultiCollector(collectInterval, func(addr net.UDPAddr, msg interface{}) {
			switch msg := msg.(type) {
			case *data.NodeInfo:
				nodes.Get(msg.NodeId).Nodeinfo = msg
			case *data.NeighbourStruct:
				nodes.Get(msg.NodeId).Neighbours = msg
			case *data.StatisticsStruct:
				nodes.Get(msg.NodeId).Statistics = msg
			default:
				log.Println("unknown message:", msg)
			}
		})
	}

	//TODO bad
	if config.Webserver.Enable {
		log.Fatal(http.ListenAndServe(net.JoinHostPort(config.Webserver.Address, config.Webserver.Port), nil))
	}
	// Wait for End
	sigs := make(chan os.Signal, 1)
	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
	sig := <-sigs
	log.Println("received", sig)

	// Close everything at the end
	if wsserverForNodes != nil {
		wsserverForNodes.Close()
	}
	if multiCollector != nil {
		multiCollector.Close()
	}
}