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

package main

import (
	"fmt"
	"log"

	container "github.com/nspcc-dev/neofs-api-go/v2/container/grpc"
	object "github.com/nspcc-dev/neofs-api-go/v2/object/grpc"
	session "github.com/nspcc-dev/neofs-api-go/v2/session/grpc"
	containerGRPC "github.com/nspcc-dev/neofs-node/pkg/network/transport/container/grpc"
	objectGRPC "github.com/nspcc-dev/neofs-node/pkg/network/transport/object/grpc"
	sessionGRPC "github.com/nspcc-dev/neofs-node/pkg/network/transport/session/grpc"
	"github.com/nspcc-dev/neofs-node/pkg/util/grace"
)

func fatalOnErr(err error) {
	if err != nil {
		log.Fatal(err)
	}
}

func main() {
	c := defaultCfg()

	init_(c)

	bootUp(c)

	wait(c)

	shutdown(c)
}

func init_(c *cfg) {
	c.ctx = grace.NewGracefulContext(nil)

	initGRPC(c)

	initAccountingService(c)

	container.RegisterContainerServiceServer(c.cfgGRPC.server, containerGRPC.New(new(containerSvc)))
	session.RegisterSessionServiceServer(c.cfgGRPC.server, sessionGRPC.New(new(sessionSvc)))
	object.RegisterObjectServiceServer(c.cfgGRPC.server, objectGRPC.New(new(objectSvc)))
}

func bootUp(c *cfg) {
	serveGRPC(c)
}

func wait(c *cfg) {
	<-c.ctx.Done()
}

func shutdown(c *cfg) {
	c.cfgGRPC.server.GracefulStop()
	fmt.Println("gRPC server stopped")

	c.wg.Wait()
}