forked from TrueCloudLab/frostfs-node
[#84] Add netmap service executor and signer
Signed-off-by: Alex Vanin <alexey@nspcc.ru>
This commit is contained in:
parent
f92dc5b27c
commit
2d5cb378a7
2 changed files with 79 additions and 0 deletions
41
pkg/services/netmap/executor.go
Normal file
41
pkg/services/netmap/executor.go
Normal file
|
@ -0,0 +1,41 @@
|
||||||
|
package netmap
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
|
||||||
|
"github.com/nspcc-dev/neofs-api-go/pkg"
|
||||||
|
"github.com/nspcc-dev/neofs-api-go/v2/netmap"
|
||||||
|
"github.com/nspcc-dev/neofs-api-go/v2/session"
|
||||||
|
)
|
||||||
|
|
||||||
|
type executorSvc struct {
|
||||||
|
version *pkg.Version
|
||||||
|
localNodeInfo *netmap.NodeInfo
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewExecutionService(ni *netmap.NodeInfo, v *pkg.Version) netmap.Service {
|
||||||
|
if ni == nil || v == nil {
|
||||||
|
// this should never happen, otherwise it programmers bug
|
||||||
|
panic("can't create netmap execution service")
|
||||||
|
}
|
||||||
|
|
||||||
|
return &executorSvc{
|
||||||
|
version: v,
|
||||||
|
localNodeInfo: ni,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *executorSvc) LocalNodeInfo(
|
||||||
|
_ context.Context,
|
||||||
|
_ *netmap.LocalNodeInfoRequest) (*netmap.LocalNodeInfoResponse, error) {
|
||||||
|
|
||||||
|
body := new(netmap.LocalNodeInfoResponseBody)
|
||||||
|
body.SetVersion(s.version.ToV2())
|
||||||
|
body.SetNodeInfo(s.localNodeInfo)
|
||||||
|
|
||||||
|
resp := new(netmap.LocalNodeInfoResponse)
|
||||||
|
resp.SetBody(body)
|
||||||
|
resp.SetMetaHeader(new(session.ResponseMetaHeader))
|
||||||
|
|
||||||
|
return resp, nil
|
||||||
|
}
|
38
pkg/services/netmap/sign.go
Normal file
38
pkg/services/netmap/sign.go
Normal file
|
@ -0,0 +1,38 @@
|
||||||
|
package netmap
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"crypto/ecdsa"
|
||||||
|
|
||||||
|
"github.com/nspcc-dev/neofs-api-go/v2/netmap"
|
||||||
|
"github.com/nspcc-dev/neofs-node/pkg/services/util"
|
||||||
|
)
|
||||||
|
|
||||||
|
type signService struct {
|
||||||
|
sigSvc *util.SignService
|
||||||
|
|
||||||
|
svc netmap.Service
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewSignService(key *ecdsa.PrivateKey, svc netmap.Service) netmap.Service {
|
||||||
|
return &signService{
|
||||||
|
sigSvc: util.NewUnarySignService(key),
|
||||||
|
svc: svc,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *signService) LocalNodeInfo(
|
||||||
|
ctx context.Context,
|
||||||
|
req *netmap.LocalNodeInfoRequest) (*netmap.LocalNodeInfoResponse, error) {
|
||||||
|
|
||||||
|
resp, err := s.sigSvc.HandleUnaryRequest(ctx, req,
|
||||||
|
func(ctx context.Context, req interface{}) (interface{}, error) {
|
||||||
|
return s.svc.LocalNodeInfo(ctx, req.(*netmap.LocalNodeInfoRequest))
|
||||||
|
},
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return resp.(*netmap.LocalNodeInfoResponse), nil
|
||||||
|
}
|
Loading…
Reference in a new issue