Initial commit

Initial public review release v0.10.0

cmd/neofs-node/main.go

@@ -0,0 +1,146 @@
+package main
+
+import (
+	"context"
+	"crypto/ecdsa"
+	"crypto/elliptic"
+	"crypto/rand"
+	"flag"
+	"os"
+	"time"
+
+	"github.com/nspcc-dev/neofs-api-go/service"
+	state2 "github.com/nspcc-dev/neofs-api-go/state"
+	crypto "github.com/nspcc-dev/neofs-crypto"
+	"github.com/nspcc-dev/neofs-node/lib/fix"
+	"github.com/nspcc-dev/neofs-node/lib/fix/config"
+	"github.com/nspcc-dev/neofs-node/lib/fix/web"
+	"github.com/nspcc-dev/neofs-node/lib/fix/worker"
+	"github.com/nspcc-dev/neofs-node/lib/muxer"
+	"github.com/nspcc-dev/neofs-node/misc"
+	"github.com/nspcc-dev/neofs-node/modules/node"
+	"github.com/nspcc-dev/neofs-node/services/public/state"
+	"github.com/pkg/errors"
+	"github.com/spf13/viper"
+	"go.uber.org/dig"
+	"go.uber.org/zap"
+	"google.golang.org/grpc"
+)
+
+type params struct {
+	dig.In
+
+	Debug  web.Profiler   `optional:"true"`
+	Metric web.Metrics    `optional:"true"`
+	Worker worker.Workers `optional:"true"`
+	Muxer  muxer.Mux
+	Logger *zap.Logger
+}
+
+var (
+	healthCheck bool
+	configFile  string
+)
+
+func runner(ctx context.Context, p params) error {
+	// create combined service, that would start/stop all
+	svc := fix.NewServices(p.Debug, p.Metric, p.Muxer, p.Worker)
+
+	p.Logger.Info("start services")
+	svc.Start(ctx)
+
+	<-ctx.Done()
+
+	p.Logger.Info("stop services")
+	svc.Stop()
+
+	return nil
+}
+
+func check(err error) {
+	if err != nil {
+		panic(err)
+	}
+}
+
+// FIXME: this is a copypaste from node settings constructor
+func keyFromCfg(v *viper.Viper) (*ecdsa.PrivateKey, error) {
+	switch key := v.GetString("node.private_key"); key {
+	case "":
+		return nil, errors.New("`node.private_key` could not be empty")
+	case "generated":
+		return ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
+	default:
+		return crypto.LoadPrivateKey(key)
+	}
+}
+
+func runHealthCheck() {
+	if !healthCheck {
+		return
+	}
+
+	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+	defer cancel()
+
+	cfg, err := config.NewConfig(config.Params{
+		File:    configFile,
+		Prefix:  misc.Prefix,
+		Name:    misc.NodeName,
+		Version: misc.Version,
+
+		AppDefaults: setDefaults,
+	})
+	check(err)
+
+	addr := cfg.GetString("node.address")
+
+	key, err := keyFromCfg(cfg)
+	if err != nil {
+		check(err)
+	}
+
+	con, err := grpc.DialContext(ctx, addr,
+		// TODO: we must provide grpc.WithInsecure() or set credentials
+		grpc.WithInsecure())
+	check(err)
+
+	req := new(state.HealthRequest)
+	req.SetTTL(service.NonForwardingTTL)
+	if err := service.SignRequestData(key, req); err != nil {
+		check(err)
+	}
+
+	res, err := state2.NewStatusClient(con).
+		HealthCheck(ctx, req)
+	check(errors.Wrapf(err, "address: %q", addr))
+
+	var exitCode int
+
+	if !res.Healthy {
+		exitCode = 2
+	}
+	_, _ = os.Stdout.Write([]byte(res.Status + "\n"))
+	os.Exit(exitCode)
+}
+
+func main() {
+	flag.BoolVar(&healthCheck, "health", healthCheck, "run health-check")
+
+	// todo: if configFile is empty, we can check './config.yml' manually
+	flag.StringVar(&configFile, "config", configFile, "use config.yml file")
+	flag.Parse()
+
+	runHealthCheck()
+
+	fix.New(&fix.Settings{
+		File:    configFile,
+		Name:    misc.NodeName,
+		Prefix:  misc.Prefix,
+		Runner:  runner,
+		Build:   misc.Build,
+		Version: misc.Version,
+
+		AppDefaults: setDefaults,
+	}, node.Module).RunAndCatch()
+}