forked from TrueCloudLab/frostfs-node
[#1255] object: Add persistent storage usage
Use persistent storage usage in the node if it was configured so. Signed-off-by: Pavel Karpy <carpawell@nspcc.ru>
This commit is contained in:
parent
9cda3121ab
commit
90a8c52bdb
4 changed files with 57 additions and 6 deletions
|
@ -1,16 +1,51 @@
|
|||
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) {
|
||||
c.privateTokenStore = temporary.NewTokenStore()
|
||||
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())
|
||||
})
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue