forked from TrueCloudLab/frostfs-node
50 lines
1.4 KiB
Go
50 lines
1.4 KiB
Go
package grpcreputation
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/TrueCloudLab/frostfs-api-go/v2/reputation"
|
|
reputation2 "github.com/TrueCloudLab/frostfs-api-go/v2/reputation/grpc"
|
|
reputationrpc "github.com/TrueCloudLab/frostfs-node/pkg/services/reputation/rpc"
|
|
)
|
|
|
|
// Server wraps FrostFS API v2 Reputation service server
|
|
// and provides gRPC Reputation service server interface.
|
|
type Server struct {
|
|
srv reputationrpc.Server
|
|
}
|
|
|
|
// New creates, initializes and returns Server instance.
|
|
func New(srv reputationrpc.Server) *Server {
|
|
return &Server{
|
|
srv: srv,
|
|
}
|
|
}
|
|
|
|
func (s *Server) AnnounceLocalTrust(ctx context.Context, r *reputation2.AnnounceLocalTrustRequest) (*reputation2.AnnounceLocalTrustResponse, error) {
|
|
req := new(reputation.AnnounceLocalTrustRequest)
|
|
if err := req.FromGRPCMessage(r); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
resp, err := s.srv.AnnounceLocalTrust(ctx, req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return resp.ToGRPCMessage().(*reputation2.AnnounceLocalTrustResponse), nil
|
|
}
|
|
|
|
func (s *Server) AnnounceIntermediateResult(ctx context.Context, r *reputation2.AnnounceIntermediateResultRequest) (*reputation2.AnnounceIntermediateResultResponse, error) {
|
|
req := new(reputation.AnnounceIntermediateResultRequest)
|
|
if err := req.FromGRPCMessage(r); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
resp, err := s.srv.AnnounceIntermediateResult(ctx, req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return resp.ToGRPCMessage().(*reputation2.AnnounceIntermediateResultResponse), nil
|
|
}
|