logmania/cmd/server.go

package cmd

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

	log "github.com/sirupsen/logrus"
	"github.com/spf13/cobra"

	"github.com/genofire/logmania/bot"
	"github.com/genofire/logmania/lib"
	"github.com/genofire/logmania/notify"
	allNotify "github.com/genofire/logmania/notify/all"
	configNotify "github.com/genofire/logmania/notify/config"
	"github.com/genofire/logmania/receive"
	allReceiver "github.com/genofire/logmania/receive/all"
)

var (
	configPath  string
	config      *lib.Config
	notifyState *configNotify.NotifyState
	notifier    notify.Notifier
	receiver    receive.Receiver
	logChannel  chan *log.Entry
	logmaniaBot *bot.Bot
)

// serverCmd represents the serve command
var serverCmd = &cobra.Command{
	Use:     "server",
	Short:   "Runs the logmania server",
	Example: "logmania server --config /etc/yanic.toml",
	Run: func(cmd *cobra.Command, args []string) {
		log.SetFormatter(&log.TextFormatter{
			DisableTimestamp: true,
		})
		config, err := lib.ReadConfig(configPath)
		if config == nil || err != nil {
			log.Panicf("Could not load '%s' for configuration.", configPath)
		}

		notifyState := configNotify.ReadStateFile(config.Notify.StateFile)
		go notifyState.Saver(config.Notify.StateFile)

		logmaniaBot = bot.NewBot(notifyState)

		notifier = allNotify.Init(&config.Notify, notifyState, logmaniaBot)
		log.AddHook(notifier)
		logChannel = make(chan *log.Entry)

		go func() {
			for a := range logChannel {
				notifier.Fire(a)
			}
		}()

		if config.Notify.AlertCheck.Duration > time.Duration(time.Second) {
			go notifyState.Alert(config.Notify.AlertCheck.Duration, notifier.Fire)
		}

		log.Info("starting logmania")

		receiver = allReceiver.Init(&config.Receive, logChannel)

		go receiver.Listen()

		// Wait for system signal
		sigchan := make(chan os.Signal, 1)
		signal.Notify(sigchan, syscall.SIGTERM, syscall.SIGHUP, syscall.SIGQUIT, syscall.SIGUSR1)
		for sig := range sigchan {
			switch sig {
			case syscall.SIGTERM:
				log.Panic("terminated of logmania")
				os.Exit(0)
			case syscall.SIGQUIT:
				quit()
			case syscall.SIGHUP:
				quit()
			case syscall.SIGUSR1:
				reload()
			}
		}
	},
}

func quit() {
	receiver.Close()
	notifier.Close()
	log.Info("quit of logmania")
	os.Exit(0)
}

func reload() {
	log.Info("reload config file")
	config, err := lib.ReadConfig(configPath)
	if config == nil || err != nil {
		log.Errorf("reload: could not load '%s' for new configuration. Skip reload.", configPath)
		return
	}
	receiver.Close()
	receiver = allReceiver.Init(&config.Receive, logChannel)
	go receiver.Listen()

	notifier.Close()
	notifier = allNotify.Init(&config.Notify, notifyState, logmaniaBot)
}

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