86 lines
2.4 KiB
Go
86 lines
2.4 KiB
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")
|
|
}
|