mirror of
https://github.com/crazy-max/diun.git
synced 2025-12-21 13:23:09 +01:00
100 lines
3.2 KiB
Go
100 lines
3.2 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"os/signal"
|
|
"path"
|
|
|
|
"github.com/crazy-max/diun/v4/internal/app"
|
|
"github.com/crazy-max/diun/v4/internal/config"
|
|
"github.com/crazy-max/diun/v4/internal/logging"
|
|
"github.com/crazy-max/diun/v4/pkg/utl"
|
|
"github.com/pkg/profile"
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
// ServeCmd holds serve command args and flags
|
|
type ServeCmd struct {
|
|
Cfgfile string `name:"config" env:"CONFIG" help:"Diun configuration file."`
|
|
ProfilerPath string `name:"profiler-path" env:"PROFILER_PATH" help:"Base path where profiling files are written."`
|
|
Profiler string `name:"profiler" env:"PROFILER" help:"Profiler to use."`
|
|
LogLevel string `name:"log-level" env:"LOG_LEVEL" default:"info" help:"Set log level."`
|
|
LogJSON bool `name:"log-json" env:"LOG_JSON" default:"false" help:"Enable JSON logging output.'"`
|
|
LogCaller bool `name:"log-caller" env:"LOG_CALLER" default:"false" help:"Add file:line of the caller to log output."`
|
|
LogNoColor bool `name:"log-nocolor" env:"LOG_NOCOLOR" default:"false" help:"Disables the colorized output."`
|
|
GRPCAuthority string `name:"grpc-authority" env:"GRPC_AUTHORITY" default:":42286" help:"Address used to expose the gRPC server."`
|
|
}
|
|
|
|
func (s *ServeCmd) Run(ctx *Context) error {
|
|
var diun *app.Diun
|
|
|
|
// Logging
|
|
logging.Configure(logging.Options{
|
|
LogLevel: s.LogLevel,
|
|
LogJSON: s.LogJSON,
|
|
LogCaller: s.LogCaller,
|
|
LogNoColor: s.LogNoColor,
|
|
})
|
|
log.Info().Str("version", version).Msgf("Starting %s", ctx.Meta.Name)
|
|
|
|
// Handle os signals
|
|
channel := make(chan os.Signal, 1)
|
|
signal.Notify(channel, os.Interrupt, utl.SIGTERM)
|
|
go func() {
|
|
sig := <-channel
|
|
diun.Close()
|
|
log.Warn().Msgf("Caught signal %v", sig)
|
|
os.Exit(0)
|
|
}()
|
|
|
|
// Load configuration
|
|
cfg, err := config.Load(s.Cfgfile)
|
|
if err != nil {
|
|
log.Fatal().Err(err).Msg("Cannot load configuration")
|
|
}
|
|
log.Debug().Msg(cfg.String())
|
|
|
|
// Profiler
|
|
if len(s.Profiler) > 0 && len(s.ProfilerPath) > 0 {
|
|
profilerPath := path.Clean(s.ProfilerPath)
|
|
if err = os.MkdirAll(profilerPath, os.ModePerm); err != nil {
|
|
log.Fatal().Err(err).Msg("Cannot create profiler folder")
|
|
}
|
|
profilePath := profile.ProfilePath(profilerPath)
|
|
switch s.Profiler {
|
|
case "cpu":
|
|
defer profile.Start(profile.CPUProfile, profilePath).Stop()
|
|
case "mem":
|
|
defer profile.Start(profile.MemProfile, profilePath).Stop()
|
|
case "alloc":
|
|
defer profile.Start(profile.MemProfileAllocs, profilePath).Stop()
|
|
case "heap":
|
|
defer profile.Start(profile.MemProfileHeap, profilePath).Stop()
|
|
case "routines":
|
|
defer profile.Start(profile.GoroutineProfile, profilePath).Stop()
|
|
case "mutex":
|
|
defer profile.Start(profile.MutexProfile, profilePath).Stop()
|
|
case "threads":
|
|
defer profile.Start(profile.ThreadcreationProfile, profilePath).Stop()
|
|
case "block":
|
|
defer profile.Start(profile.BlockProfile, profilePath).Stop()
|
|
default:
|
|
log.Fatal().Msgf("Unknown profiler: %s", s.Profiler) //nolint:gocritic // defer not set if profiler is unknown
|
|
}
|
|
}
|
|
|
|
// Init
|
|
diun, err = app.New(ctx.Meta, cfg, s.GRPCAuthority)
|
|
if err != nil {
|
|
log.Fatal().Err(err).Msgf("Cannot initialize %s", ctx.Meta.Name)
|
|
}
|
|
|
|
// Start
|
|
err = diun.Start()
|
|
if err != nil {
|
|
log.Fatal().Err(err).Msgf("Cannot start %s", ctx.Meta.Name)
|
|
}
|
|
|
|
return nil
|
|
}
|