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

145 lines
2.9 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"
"net/http"
2020-07-24 13:54:03 +00:00
"os"
"os/signal"
"syscall"
2020-07-24 13:54:03 +00:00
"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"
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
)
func exitErr(err error) {
if err != nil {
fmt.Println(err)
os.Exit(ErrorReturnCode)
}
}
func main() {
configFile := flag.String("config", "", "path to config")
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("NeoFS Inner Ring node"))
2020-07-24 13:54:03 +00:00
os.Exit(SuccessReturnCode)
}
cfg, err := newConfig(*configFile)
exitErr(err)
var logPrm logger.Prm
err = logPrm.SetLevelString(
cfg.GetString("logger.level"),
)
exitErr(err)
log, err := logger.NewLogger(&logPrm)
2020-07-24 13:54:03 +00:00
exitErr(err)
ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM, syscall.SIGHUP)
defer cancel()
intErr := make(chan error) // internal inner ring errors
2020-07-24 13:54:03 +00:00
httpServers := initHTTPServers(cfg, log)
2020-07-24 13:54:03 +00:00
innerRing, err := innerring.New(ctx, log, cfg, intErr)
exitErr(err)
2020-07-24 13:54:03 +00:00
// start HTTP servers
for i := range httpServers {
srv := httpServers[i]
go func() {
exitErr(srv.Serve())
}()
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("application started",
zap.String("version", misc.Version))
2020-07-24 13:54:03 +00:00
select {
case <-ctx.Done():
case err := <-intErr:
log.Info("internal error", zap.String("msg", err.Error()))
}
2020-07-24 13:54:03 +00:00
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()),
)
}
}()
}
2020-07-24 13:54:03 +00:00
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.Prm
prm.Address = addr
prm.Handler = item.handler()
httpServers = append(httpServers,
httputil.New(prm,
httputil.WithShutdownTimeout(
cfg.GetDuration(item.cfgPrefix+".shutdown_timeout"),
),
),
)
}
return httpServers
}