forked from TrueCloudLab/frostfs-node
144 lines
4.8 KiB
Go
144 lines
4.8 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
|
|
"github.com/nspcc-dev/neofs-api-go/v2/accounting"
|
|
accountingGRPC "github.com/nspcc-dev/neofs-api-go/v2/accounting/grpc"
|
|
containerGRPC "github.com/nspcc-dev/neofs-api-go/v2/container"
|
|
container "github.com/nspcc-dev/neofs-api-go/v2/container/grpc"
|
|
objectGRPC "github.com/nspcc-dev/neofs-api-go/v2/object"
|
|
object "github.com/nspcc-dev/neofs-api-go/v2/object/grpc"
|
|
"github.com/nspcc-dev/neofs-api-go/v2/session"
|
|
sessionGRPC "github.com/nspcc-dev/neofs-api-go/v2/session/grpc"
|
|
accountingTransportGRPC "github.com/nspcc-dev/neofs-node/pkg/network/transport/accounting/grpc"
|
|
containerTransport "github.com/nspcc-dev/neofs-node/pkg/network/transport/container/grpc"
|
|
objectTransport "github.com/nspcc-dev/neofs-node/pkg/network/transport/object/grpc"
|
|
sessionTransport "github.com/nspcc-dev/neofs-node/pkg/network/transport/session/grpc"
|
|
accountingService "github.com/nspcc-dev/neofs-node/pkg/services/accounting"
|
|
"github.com/pkg/errors"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type accountingSvcExec struct{}
|
|
|
|
type sessionSvc struct{}
|
|
|
|
type containerSvc struct{}
|
|
|
|
type objectSvc struct{}
|
|
|
|
func unimplementedErr(srv, call string) error {
|
|
return errors.Errorf("unimplemented API service call %s.%s", srv, call)
|
|
}
|
|
|
|
func (s *accountingSvcExec) Balance(context.Context, *accounting.BalanceRequestBody) (*accounting.BalanceResponseBody, error) {
|
|
return new(accounting.BalanceResponseBody), nil
|
|
}
|
|
|
|
func (s *sessionSvc) Create(context.Context, *session.CreateRequest) (*session.CreateResponse, error) {
|
|
return nil, unimplementedErr("Session", "Create")
|
|
}
|
|
|
|
func (s *containerSvc) Put(context.Context, *containerGRPC.PutRequest) (*containerGRPC.PutResponse, error) {
|
|
return nil, unimplementedErr("Container", "Put")
|
|
}
|
|
|
|
func (s *containerSvc) Delete(context.Context, *containerGRPC.DeleteRequest) (*containerGRPC.DeleteResponse, error) {
|
|
return nil, unimplementedErr("Container", "Delete")
|
|
}
|
|
|
|
func (s *containerSvc) Get(context.Context, *containerGRPC.GetRequest) (*containerGRPC.GetResponse, error) {
|
|
return nil, unimplementedErr("Container", "Get")
|
|
}
|
|
|
|
func (s *containerSvc) List(context.Context, *containerGRPC.ListRequest) (*containerGRPC.ListResponse, error) {
|
|
return nil, unimplementedErr("Container", "List")
|
|
}
|
|
|
|
func (s *containerSvc) SetExtendedACL(context.Context, *containerGRPC.SetExtendedACLRequest) (*containerGRPC.SetExtendedACLResponse, error) {
|
|
return nil, unimplementedErr("Container", "SetExtendedACL")
|
|
}
|
|
|
|
func (s *containerSvc) GetExtendedACL(context.Context, *containerGRPC.GetExtendedACLRequest) (*containerGRPC.GetExtendedACLResponse, error) {
|
|
return nil, unimplementedErr("Container", "GetExtendedACL")
|
|
}
|
|
|
|
func (s *objectSvc) Get(context.Context, *objectGRPC.GetRequest) (objectGRPC.GetObjectStreamer, error) {
|
|
return nil, unimplementedErr("Object", "Get")
|
|
}
|
|
|
|
func (s *objectSvc) Put(context.Context) (objectGRPC.PutObjectStreamer, error) {
|
|
return nil, unimplementedErr("Object", "Put")
|
|
}
|
|
|
|
func (s *objectSvc) Head(context.Context, *objectGRPC.HeadRequest) (*objectGRPC.HeadResponse, error) {
|
|
return nil, unimplementedErr("Object", "Put")
|
|
}
|
|
|
|
func (s *objectSvc) Search(context.Context, *objectGRPC.SearchRequest) (objectGRPC.SearchObjectStreamer, error) {
|
|
return nil, unimplementedErr("Object", "Search")
|
|
}
|
|
|
|
func (s *objectSvc) Delete(context.Context, *objectGRPC.DeleteRequest) (*objectGRPC.DeleteResponse, error) {
|
|
return nil, unimplementedErr("Object", "Delete")
|
|
}
|
|
|
|
func (s *objectSvc) GetRange(context.Context, *objectGRPC.GetRangeRequest) (objectGRPC.GetRangeObjectStreamer, error) {
|
|
return nil, unimplementedErr("Object", "GetRange")
|
|
}
|
|
|
|
func (s *objectSvc) GetRangeHash(context.Context, *objectGRPC.GetRangeHashRequest) (*objectGRPC.GetRangeHashResponse, error) {
|
|
return nil, unimplementedErr("Object", "GetRangeHash")
|
|
}
|
|
|
|
func serveGRPC(c *cfg) {
|
|
lis, err := net.Listen("tcp", c.grpcAddr)
|
|
fatalOnErr(err)
|
|
|
|
srv := grpc.NewServer()
|
|
|
|
metaHdr := new(session.ResponseMetaHeader)
|
|
xHdr := new(session.XHeader)
|
|
xHdr.SetKey("test X-Header key")
|
|
xHdr.SetValue("test X-Header value")
|
|
metaHdr.SetXHeaders([]*session.XHeader{xHdr})
|
|
|
|
accountingGRPC.RegisterAccountingServiceServer(srv,
|
|
accountingTransportGRPC.New(
|
|
accountingService.NewSignService(
|
|
c.key,
|
|
accountingService.NewExecutionService(new(accountingSvcExec), metaHdr),
|
|
),
|
|
),
|
|
)
|
|
container.RegisterContainerServiceServer(srv, containerTransport.New(new(containerSvc)))
|
|
sessionGRPC.RegisterSessionServiceServer(srv, sessionTransport.New(new(sessionSvc)))
|
|
object.RegisterObjectServiceServer(srv, objectTransport.New(new(objectSvc)))
|
|
|
|
go func() {
|
|
c.wg.Add(1)
|
|
defer func() {
|
|
c.wg.Done()
|
|
}()
|
|
|
|
if err := srv.Serve(lis); err != nil {
|
|
fmt.Println("gRPC server error", err)
|
|
}
|
|
}()
|
|
|
|
go func() {
|
|
c.wg.Add(1)
|
|
defer func() {
|
|
fmt.Println("gRPC server stopped gracefully")
|
|
fmt.Println("net listener stopped", lis.Addr())
|
|
c.wg.Done()
|
|
}()
|
|
|
|
<-c.ctx.Done()
|
|
|
|
srv.GracefulStop()
|
|
}()
|
|
}
|