Move to frostfs-node

Signed-off-by: Pavel Karpy <p.karpy@yadro.com>

cmd/frostfs-ir/main.go

@@ -0,0 +1,144 @@
+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")
+	versionFlag := flag.Bool("version", false, "frostfs-ir node version")
+	flag.Parse()
+
+	if *versionFlag {
+		fmt.Print(misc.BuildInfo("NeoFS Inner Ring node"))
+
+		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)
+	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.Prm
+
+		prm.Address = addr
+		prm.Handler = item.handler()
+
+		httpServers = append(httpServers,
+			httputil.New(prm,
+				httputil.WithShutdownTimeout(
+					cfg.GetDuration(item.cfgPrefix+".shutdown_timeout"),
+				),
+			),
+		)
+	}
+
+	return httpServers
+}