forked from TrueCloudLab/frostfs-node
69 lines
2 KiB
Go
69 lines
2 KiB
Go
package grpc
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/TrueCloudLab/frostfs-api-go/v2/netmap"
|
|
netmapGRPC "github.com/TrueCloudLab/frostfs-api-go/v2/netmap/grpc"
|
|
netmapsvc "github.com/TrueCloudLab/frostfs-node/pkg/services/netmap"
|
|
)
|
|
|
|
// Server wraps NeoFS API Netmap service and
|
|
// provides gRPC Netmap service server interface.
|
|
type Server struct {
|
|
srv netmapsvc.Server
|
|
}
|
|
|
|
// New creates, initializes and returns Server instance.
|
|
func New(c netmapsvc.Server) *Server {
|
|
return &Server{
|
|
srv: c,
|
|
}
|
|
}
|
|
|
|
// LocalNodeInfo converts gRPC request message and passes it to internal netmap service.
|
|
func (s *Server) LocalNodeInfo(
|
|
ctx context.Context,
|
|
req *netmapGRPC.LocalNodeInfoRequest) (*netmapGRPC.LocalNodeInfoResponse, error) {
|
|
nodeInfoReq := new(netmap.LocalNodeInfoRequest)
|
|
if err := nodeInfoReq.FromGRPCMessage(req); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
resp, err := s.srv.LocalNodeInfo(ctx, nodeInfoReq)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return resp.ToGRPCMessage().(*netmapGRPC.LocalNodeInfoResponse), nil
|
|
}
|
|
|
|
// NetworkInfo converts gRPC request message and passes it to internal netmap service.
|
|
func (s *Server) NetworkInfo(ctx context.Context, req *netmapGRPC.NetworkInfoRequest) (*netmapGRPC.NetworkInfoResponse, error) {
|
|
netInfoReq := new(netmap.NetworkInfoRequest)
|
|
if err := netInfoReq.FromGRPCMessage(req); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
resp, err := s.srv.NetworkInfo(ctx, netInfoReq)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return resp.ToGRPCMessage().(*netmapGRPC.NetworkInfoResponse), nil
|
|
}
|
|
|
|
// NetmapSnapshot converts gRPC request message and passes it to internal netmap service.
|
|
func (s *Server) NetmapSnapshot(ctx context.Context, req *netmapGRPC.NetmapSnapshotRequest) (*netmapGRPC.NetmapSnapshotResponse, error) {
|
|
snapshotReq := new(netmap.SnapshotRequest)
|
|
if err := snapshotReq.FromGRPCMessage(req); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
resp, err := s.srv.Snapshot(ctx, snapshotReq)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return resp.ToGRPCMessage().(*netmapGRPC.NetmapSnapshotResponse), nil
|
|
}
|