package main import ( "context" "fmt" "time" "github.com/nspcc-dev/neofs-api-go/v2/session" sessionGRPC "github.com/nspcc-dev/neofs-api-go/v2/session/grpc" nodeconfig "github.com/nspcc-dev/neofs-node/cmd/neofs-node/config/node" "github.com/nspcc-dev/neofs-node/pkg/morph/event" "github.com/nspcc-dev/neofs-node/pkg/morph/event/netmap" sessionTransportGRPC "github.com/nspcc-dev/neofs-node/pkg/network/transport/session/grpc" sessionSvc "github.com/nspcc-dev/neofs-node/pkg/services/session" "github.com/nspcc-dev/neofs-node/pkg/services/session/storage" "github.com/nspcc-dev/neofs-node/pkg/services/session/storage/persistent" "github.com/nspcc-dev/neofs-node/pkg/services/session/storage/temporary" "github.com/nspcc-dev/neofs-sdk-go/owner" ) type sessionStorage interface { Create(ctx context.Context, body *session.CreateRequestBody) (*session.CreateResponseBody, error) Get(ownerID *owner.ID, tokenID []byte) *storage.PrivateToken RemoveOld(epoch uint64) Close() error } func initSessionService(c *cfg) { if persistentSessionPath := nodeconfig.PersistentSessions(c.appCfg).Path(); persistentSessionPath != "" { persisessions, err := persistent.NewTokenStore(persistentSessionPath, persistent.WithLogger(c.log), persistent.WithTimeout(100*time.Millisecond), persistent.WithEncryptionKey(&c.key.PrivateKey), ) if err != nil { panic(fmt.Errorf("could not create persistent session token storage: %w", err)) } c.privateTokenStore = persisessions } else { c.privateTokenStore = temporary.NewTokenStore() } c.onShutdown(func() { _ = c.privateTokenStore.Close() }) addNewEpochNotificationHandler(c, func(ev event.Event) { c.privateTokenStore.RemoveOld(ev.(netmap.NewEpoch).EpochNumber()) }) server := sessionTransportGRPC.New( sessionSvc.NewSignService( &c.key.PrivateKey, sessionSvc.NewResponseService( sessionSvc.NewExecutionService( c.privateTokenStore, ), c.respSvc, ), ), ) for _, srv := range c.cfgGRPC.servers { sessionGRPC.RegisterSessionServiceServer(srv, server) } }