yanic/cmd/serve.go

package cmd

import (
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/bdlm/log"
	"github.com/spf13/cobra"

	"github.com/FreifunkBremen/yanic/database"
	allDatabase "github.com/FreifunkBremen/yanic/database/all"
	allOutput "github.com/FreifunkBremen/yanic/output/all"
	"github.com/FreifunkBremen/yanic/respond"
	"github.com/FreifunkBremen/yanic/runtime"
	"github.com/FreifunkBremen/yanic/webserver"
)

// Config represents the whole configuration
type ServeConfig struct {
	Respondd  respond.Config
	Webserver webserver.Config
	Nodes     runtime.NodesConfig
	Database  database.Config
}

// serveCmd represents the serve command
var serveCmd = &cobra.Command{
	Use:     "serve",
	Short:   "Runs the yanic server",
	Example: "yanic serve --config /etc/yanic.toml",
	Run: func(cmd *cobra.Command, args []string) {
		config := &ServeConfig{}
		if err := ReadConfigFile(configPath, config); err != nil {
			log.Panicf("unable to load config file: %s", err)
		}

		if err := allDatabase.Start(config.Database); err != nil {
			log.Panicf("could not connect to database: %s", err)
		}
		defer allDatabase.Close()

		nodes = runtime.NewNodes(&config.Nodes)
		nodes.Start()

		if err := allOutput.Start(nodes, config.Nodes); err != nil {
			log.Panicf("error on init outputs: %s", err)
		}
		defer allOutput.Close()

		if config.Webserver.Enable {
			log.Infof("starting webserver on %s", config.Webserver.Bind)
			srv := webserver.New(config.Webserver.Bind, config.Webserver.Webroot)
			go webserver.Start(srv)
			defer srv.Close()
		}

		if config.Respondd.Enable {
			// Delaying startup to start at a multiple of `duration` since the zero time.
			if duration := config.Respondd.Synchronize.Duration; duration > 0 {
				now := time.Now()
				delay := duration - now.Sub(now.Truncate(duration))
				log.Infof("delaying %0.1f seconds", delay.Seconds())
				time.Sleep(delay)
			}

			collector = respond.NewCollector(allDatabase.Conn, nodes, config.Respondd.SitesDomains(), config.Respondd.Interfaces)
			collector.Start(config.Respondd.CollectInterval.Duration)
			defer collector.Close()
		}

		// Wait for INT/TERM
		sigs := make(chan os.Signal, 1)
		signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
		sig := <-sigs
		log.Infof("received %s", sig)

	},
}

func init() {
	RootCmd.AddCommand(serveCmd)
	serveCmd.Flags().StringVarP(&configPath, "config", "c", "config.toml", "Path to configuration file")
}