package main import ( "encoding/json" "flag" "log" "net" "net/http" "os" "os/signal" "reflect" "strings" "syscall" "time" "github.com/monitormap/micro-daemon/models" "github.com/monitormap/micro-daemon/respond" "github.com/monitormap/micro-daemon/websocketserver" ) var ( wsserverForNodes = websocketserver.NewServer("/nodes") respondDaemon *respond.Daemon nodes = models.NewNodes() aliases = models.NewNodes() listenAddr string listenPort string collectInterval time.Duration httpDir string outputNodesFile string outputAliasesFile string saveInterval time.Duration ) func main() { var collectSeconds, saveSeconds int flag.StringVar(&listenAddr, "host", "", "path aliases.json file") flag.StringVar(&listenPort, "port", "8080", "path aliases.json file") flag.IntVar(&collectSeconds, "collectInterval", 15, "interval for data collections") flag.StringVar(&httpDir, "httpdir", "webroot", "a implemented static file webserver") flag.StringVar(&outputNodesFile, "path-nodes", "webroot/nodes.json", "path nodes.json file") flag.StringVar(&outputAliasesFile, "path-aliases", "webroot/aliases.json", "path aliases.json file") flag.IntVar(&saveSeconds, "saveInterval", 60, "interval for data saving") flag.Parse() collectInterval = time.Second * time.Duration(collectSeconds) saveInterval = time.Second * time.Duration(saveSeconds) go wsserverForNodes.Listen() go nodes.Saver(outputNodesFile, saveInterval) go aliases.Saver(outputAliasesFile, saveInterval) respondDaemon = respond.NewDaemon(func(coll *respond.Collector, res *respond.Response) { var result map[string]interface{} json.Unmarshal(res.Raw, &result) nodeID, _ := result["node_id"].(string) if nodeID == "" { log.Println("unable to parse node_id") return } node := nodes.Get(nodeID) // Set result elem := reflect.ValueOf(node).Elem() field := elem.FieldByName(strings.Title(coll.CollectType)) if !reflect.DeepEqual(field, result) { wsserverForNodes.SendAll(node) } field.Set(reflect.ValueOf(result)) }) go respondDaemon.ListenAndSend(collectInterval) http.Handle("/", http.FileServer(http.Dir(httpDir))) //TODO bad log.Fatal(http.ListenAndServe(net.JoinHostPort(listenAddr, listenPort), 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 wsserverForNodes.Close() respondDaemon.Close() }