forked from TrueCloudLab/frostfs-node
145 lines
3 KiB
Go
145 lines
3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"github.com/TrueCloudLab/frostfs-node/misc"
|
|
"github.com/TrueCloudLab/frostfs-node/pkg/innerring"
|
|
httputil "github.com/TrueCloudLab/frostfs-node/pkg/util/http"
|
|
"github.com/TrueCloudLab/frostfs-node/pkg/util/logger"
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
"github.com/spf13/viper"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
const (
|
|
// ErrorReturnCode returns when application crashed at initialization stage.
|
|
ErrorReturnCode = 1
|
|
|
|
// SuccessReturnCode returns when application closed without panic.
|
|
SuccessReturnCode = 0
|
|
)
|
|
|
|
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")
|
|
flag.Parse()
|
|
|
|
if *versionFlag {
|
|
fmt.Print(misc.BuildInfo("FrostFS Inner Ring node"))
|
|
|
|
os.Exit(SuccessReturnCode)
|
|
}
|
|
|
|
cfg, err := newConfig(*configFile, *configDir)
|
|
exitErr(err)
|
|
|
|
var logPrm logger.Prm
|
|
|
|
err = logPrm.SetLevelString(
|
|
cfg.GetString("logger.level"),
|
|
)
|
|
exitErr(err)
|
|
|
|
log, err := logger.NewLogger(&logPrm)
|
|
exitErr(err)
|
|
|
|
ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM, syscall.SIGHUP)
|
|
defer cancel()
|
|
|
|
intErr := make(chan error) // internal inner ring errors
|
|
|
|
httpServers := initHTTPServers(cfg, log)
|
|
|
|
innerRing, err := innerring.New(ctx, log, cfg, intErr)
|
|
exitErr(err)
|
|
|
|
// start HTTP servers
|
|
for i := range httpServers {
|
|
srv := httpServers[i]
|
|
go func() {
|
|
exitErr(srv.Serve())
|
|
}()
|
|
}
|
|
|
|
// start inner ring
|
|
err = innerRing.Start(ctx, intErr)
|
|
exitErr(err)
|
|
|
|
log.Info("application started",
|
|
zap.String("version", misc.Version))
|
|
|
|
select {
|
|
case <-ctx.Done():
|
|
case err := <-intErr:
|
|
log.Info("internal error", zap.String("msg", err.Error()))
|
|
}
|
|
|
|
innerRing.Stop()
|
|
|
|
// shut down HTTP servers
|
|
for i := range httpServers {
|
|
srv := httpServers[i]
|
|
|
|
go func() {
|
|
err := srv.Shutdown()
|
|
if err != nil {
|
|
log.Debug("could not shutdown HTTP server",
|
|
zap.String("error", err.Error()),
|
|
)
|
|
}
|
|
}()
|
|
}
|
|
|
|
log.Info("application stopped")
|
|
}
|
|
|
|
func initHTTPServers(cfg *viper.Viper, log *logger.Logger) []*httputil.Server {
|
|
items := []struct {
|
|
cfgPrefix string
|
|
handler func() http.Handler
|
|
}{
|
|
{"pprof", httputil.Handler},
|
|
{"prometheus", promhttp.Handler},
|
|
}
|
|
|
|
httpServers := make([]*httputil.Server, 0, len(items))
|
|
|
|
for _, item := range items {
|
|
if !cfg.GetBool(item.cfgPrefix + ".enabled") {
|
|
log.Info(item.cfgPrefix + " is disabled, skip")
|
|
continue
|
|
}
|
|
|
|
addr := cfg.GetString(item.cfgPrefix + ".address")
|
|
|
|
var prm httputil.HTTPSrvPrm
|
|
|
|
prm.Address = addr
|
|
prm.Handler = item.handler()
|
|
|
|
httpServers = append(httpServers,
|
|
httputil.New(prm,
|
|
httputil.WithShutdownTimeout(
|
|
cfg.GetDuration(item.cfgPrefix+".shutdown_timeout"),
|
|
),
|
|
),
|
|
)
|
|
}
|
|
|
|
return httpServers
|
|
}
|