package main import ( "net/http" "time" "git.frostfs.info/TrueCloudLab/frostfs-node/internal/logs" httputil "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/http" "go.uber.org/zap" ) type httpComponent struct { srv *httputil.Server address string name string handler http.Handler shutdownDur time.Duration enabled bool } const ( enabledKeyPostfix = ".enabled" addressKeyPostfix = ".address" shutdownTimeoutKeyPostfix = ".shutdown_timeout" ) func (c *httpComponent) init() { log.Info("init " + c.name) c.enabled = cfg.GetBool(c.name + enabledKeyPostfix) c.address = cfg.GetString(c.name + addressKeyPostfix) c.shutdownDur = cfg.GetDuration(c.name + shutdownTimeoutKeyPostfix) if c.enabled { c.srv = httputil.New( httputil.HTTPSrvPrm{ Address: c.address, Handler: c.handler, }, httputil.WithShutdownTimeout(c.shutdownDur), ) } else { log.Info(c.name + " is disabled, skip") c.srv = nil } } func (c *httpComponent) start() { if c.srv != nil { log.Info("start " + c.name) wg.Add(1) go func() { defer wg.Done() exitErr(c.srv.Serve()) }() } } func (c *httpComponent) shutdown() error { if c.srv != nil { log.Info("shutdown " + c.name) return c.srv.Shutdown() } return nil } func (c *httpComponent) needReload() bool { enabled := cfg.GetBool(c.name + enabledKeyPostfix) address := cfg.GetString(c.name + addressKeyPostfix) dur := cfg.GetDuration(c.name + shutdownTimeoutKeyPostfix) return enabled != c.enabled || enabled && (address != c.address || dur != c.shutdownDur) } func (c *httpComponent) reload() { log.Info("reload " + c.name) if c.needReload() { log.Info(c.name + " config updated") if err := c.shutdown(); err != nil { log.Debug(logs.FrostFSIRCouldNotShutdownHTTPServer, zap.String("error", err.Error()), ) } else { c.init() c.start() } } }