mirror of https://github.com/0xERR0R/blocky.git
104 lines
2.9 KiB
Go
104 lines
2.9 KiB
Go
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"github.com/0xERR0R/blocky/config"
|
|
"github.com/0xERR0R/blocky/evt"
|
|
"github.com/0xERR0R/blocky/log"
|
|
"github.com/0xERR0R/blocky/metrics"
|
|
"github.com/0xERR0R/blocky/server"
|
|
"github.com/0xERR0R/blocky/util"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
//nolint:gochecknoglobals
|
|
var (
|
|
done = make(chan bool, 1)
|
|
isConfigMandatory = true
|
|
signals = make(chan os.Signal, 1)
|
|
)
|
|
|
|
func newServeCommand() *cobra.Command {
|
|
return &cobra.Command{
|
|
Use: "serve",
|
|
Args: cobra.NoArgs,
|
|
Short: "start blocky DNS server (default command)",
|
|
RunE: startServer,
|
|
}
|
|
}
|
|
|
|
func startServer(_ *cobra.Command, _ []string) error {
|
|
printBanner()
|
|
|
|
cfg, err := config.LoadConfig(configPath, isConfigMandatory)
|
|
if err != nil {
|
|
return fmt.Errorf("unable to load configuration: %w", err)
|
|
}
|
|
|
|
log.Configure(&cfg.Log)
|
|
|
|
signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM)
|
|
|
|
ctx, cancelFn := context.WithCancel(context.Background())
|
|
defer cancelFn()
|
|
|
|
if cfg.Prometheus.Enable {
|
|
metrics.StartCollection()
|
|
}
|
|
|
|
srv, err := server.NewServer(ctx, cfg)
|
|
if err != nil {
|
|
return fmt.Errorf("can't start server: %w", err)
|
|
}
|
|
|
|
const errChanSize = 10
|
|
errChan := make(chan error, errChanSize)
|
|
|
|
srv.Start(ctx, errChan)
|
|
|
|
var terminationErr error
|
|
|
|
go func() {
|
|
select {
|
|
case <-signals:
|
|
log.Log().Infof("Terminating...")
|
|
util.LogOnError(ctx, "can't stop server: ", srv.Stop(ctx))
|
|
done <- true
|
|
|
|
case err := <-errChan:
|
|
log.Log().Error("server start failed: ", err)
|
|
terminationErr = err
|
|
done <- true
|
|
}
|
|
}()
|
|
|
|
evt.Bus().Publish(evt.ApplicationStarted, util.Version, util.BuildTime)
|
|
<-done
|
|
|
|
return terminationErr
|
|
}
|
|
|
|
func printBanner() {
|
|
log.Log().Info("_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/")
|
|
log.Log().Info("_/ _/")
|
|
log.Log().Info("_/ _/")
|
|
log.Log().Info("_/ _/ _/ _/ _/")
|
|
log.Log().Info("_/ _/_/_/ _/ _/_/ _/_/_/ _/ _/ _/ _/ _/")
|
|
log.Log().Info("_/ _/ _/ _/ _/ _/ _/ _/_/ _/ _/ _/")
|
|
log.Log().Info("_/ _/ _/ _/ _/ _/ _/ _/ _/ _/ _/ _/")
|
|
log.Log().Info("_/ _/_/_/ _/ _/_/ _/_/_/ _/ _/ _/_/_/ _/")
|
|
log.Log().Info("_/ _/ _/")
|
|
log.Log().Info("_/ _/_/ _/")
|
|
log.Log().Info("_/ _/")
|
|
log.Log().Info("_/ _/")
|
|
log.Log().Infof("_/ Version: %-18s Build time: %-18s _/", util.Version, util.BuildTime)
|
|
log.Log().Info("_/ _/")
|
|
log.Log().Info("_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/_/")
|
|
}
|