frostfs-node/cmd/frostfs-ir/main.go

135 lines
3.1 KiB
Go
Raw Normal View History

2020-07-24 13:54:03 +00:00
package main
import (
"context"
2020-07-24 13:54:03 +00:00
"flag"
"fmt"
"os"
"sync"
"sync/atomic"
2020-07-24 13:54:03 +00:00
"git.frostfs.info/TrueCloudLab/frostfs-node/internal/logs"
irMetrics "git.frostfs.info/TrueCloudLab/frostfs-node/internal/metrics"
"git.frostfs.info/TrueCloudLab/frostfs-node/misc"
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/innerring"
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/logger"
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/sdnotify"
"github.com/spf13/viper"
"go.uber.org/zap"
2020-07-24 13:54:03 +00:00
)
const (
// ErrorReturnCode returns when application crashed at initialization stage.
2020-07-24 13:54:03 +00:00
ErrorReturnCode = 1
// SuccessReturnCode returns when application closed without panic.
2020-07-24 13:54:03 +00:00
SuccessReturnCode = 0
EnvPrefix = "FROSTFS_IR"
)
var (
wg = new(sync.WaitGroup)
intErr = make(chan error) // internal inner ring errors
logPrm = new(logger.Prm)
innerRing *innerring.Server
pprofCmp *pprofComponent
metricsCmp *httpComponent
log *logger.Logger
cfg *viper.Viper
configFile *string
configDir *string
cmode = &atomic.Bool{}
audit = &atomic.Bool{}
2020-07-24 13:54:03 +00:00
)
func exitErr(err error) {
if err != nil {
fmt.Println(err)
os.Exit(ErrorReturnCode)
}
}
func main() {
configFile = flag.String("config", "", "path to config")
configDir = flag.String("config-dir", "", "path to config directory")
versionFlag := flag.Bool("version", false, "frostfs-ir node version")
2020-07-24 13:54:03 +00:00
flag.Parse()
if *versionFlag {
fmt.Print(misc.BuildInfo("FrostFS Inner Ring node"))
2020-07-24 13:54:03 +00:00
os.Exit(SuccessReturnCode)
}
var err error
cfg, err = newConfig()
2020-07-24 13:54:03 +00:00
exitErr(err)
cmode.Store(cfg.GetBool("node.kludge_compatibility_mode"))
metrics := irMetrics.NewInnerRingMetrics()
err = logPrm.SetLevelString(
cfg.GetString("logger.level"),
)
exitErr(err)
err = logPrm.SetDestination(
cfg.GetString("logger.destination"),
)
exitErr(err)
logPrm.SamplingHook = metrics.LogMetrics().GetSamplingHook()
logPrm.PrependTimestamp = cfg.GetBool("logger.timestamp")
log, err = logger.NewLogger(logPrm)
2020-07-24 13:54:03 +00:00
exitErr(err)
ctx, cancel := context.WithCancel(context.Background())
pprofCmp = newPprofComponent()
pprofCmp.init()
metricsCmp = newMetricsComponent()
metricsCmp.init()
audit.Store(cfg.GetBool("audit.enabled"))
2020-07-24 13:54:03 +00:00
innerRing, err = innerring.New(ctx, log, cfg, intErr, metrics, cmode, audit)
exitErr(err)
2020-07-24 13:54:03 +00:00
pprofCmp.start()
metricsCmp.start()
2020-07-24 13:54:03 +00:00
// start inner ring
err = innerRing.Start(ctx, intErr)
exitErr(err)
2020-07-24 13:54:03 +00:00
log.Info(logs.CommonApplicationStarted,
zap.String("version", misc.Version))
2020-07-24 13:54:03 +00:00
watchForSignal(cancel)
<-ctx.Done() // graceful shutdown
log.Debug(logs.FrostFSNodeWaitingForAllProcessesToStop)
wg.Wait()
log.Info(logs.FrostFSIRApplicationStopped)
2020-07-24 13:54:03 +00:00
}
func shutdown() {
innerRing.Stop()
if err := metricsCmp.shutdown(); err != nil {
log.Debug(logs.FrostFSIRCouldNotShutdownHTTPServer,
zap.String("error", err.Error()),
)
}
if err := pprofCmp.shutdown(); err != nil {
log.Debug(logs.FrostFSIRCouldNotShutdownHTTPServer,
zap.String("error", err.Error()),
)
}
if err := sdnotify.ClearStatus(); err != nil {
log.Error(logs.FailedToReportStatusToSystemd, zap.Error(err))
}
}