79 lines
2.1 KiB
Go
79 lines
2.1 KiB
Go
|
package server
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"net"
|
||
|
|
|
||
|
|
grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
|
||
|
|
grpc_zap "github.com/grpc-ecosystem/go-grpc-middleware/logging/zap"
|
||
|
|
grpc_recovery "github.com/grpc-ecosystem/go-grpc-middleware/recovery"
|
||
|
|
"go.uber.org/zap"
|
||
|
|
"google.golang.org/grpc"
|
||
|
|
"penahub.gitlab.yandexcloud.net/external/treasurer/internal/errors"
|
||
|
|
"penahub.gitlab.yandexcloud.net/external/treasurer/internal/models"
|
||
|
|
"penahub.gitlab.yandexcloud.net/external/treasurer/internal/proto/treasurer"
|
||
|
|
)
|
||
|
|
|
||
|
|
type DepsGRPC struct {
|
||
|
|
Logger *zap.Logger
|
||
|
|
}
|
||
|
|
|
||
|
|
type GRPC struct {
|
||
|
|
logger *zap.Logger
|
||
|
|
grpc *grpc.Server
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewGRPC(deps DepsGRPC) (*GRPC, errors.Error) {
|
||
|
|
if deps.Logger == nil {
|
||
|
|
return nil, errors.NewWithMessage("Logger is nil on <NewGRPC>", errors.ErrInvalidArgs)
|
||
|
|
}
|
||
|
|
|
||
|
|
grpcStreamInterceptor := grpc.StreamInterceptor(grpc_middleware.ChainStreamServer(
|
||
|
|
grpc_zap.StreamServerInterceptor(deps.Logger),
|
||
|
|
grpc_recovery.StreamServerInterceptor(),
|
||
|
|
))
|
||
|
|
|
||
|
|
grpcUnaryInterceptor := grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(
|
||
|
|
grpc_zap.UnaryServerInterceptor(deps.Logger),
|
||
|
|
grpc_recovery.UnaryServerInterceptor(),
|
||
|
|
))
|
||
|
|
|
||
|
|
return &GRPC{
|
||
|
|
grpc: grpc.NewServer(grpcStreamInterceptor, grpcUnaryInterceptor),
|
||
|
|
logger: deps.Logger,
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (receiver *GRPC) Run(config *models.ConfigurationGRPC) {
|
||
|
|
connectionString := fmt.Sprintf("%s:%s", config.Host, config.Port)
|
||
|
|
|
||
|
|
receiver.logger.Info("Starting GRPC Server", zap.String("host", connectionString))
|
||
|
|
|
||
|
|
if err := receiver.listen(connectionString); err != nil && err != grpc.ErrServerStopped {
|
||
|
|
receiver.logger.Error("GRPC Listen error", zap.Error(err))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (receiver *GRPC) Stop(_ context.Context) error {
|
||
|
|
receiver.grpc.GracefulStop()
|
||
|
|
receiver.logger.Info("Shutting down GRPC server...")
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (receiver *GRPC) Register(server treasurer.TreasurerServiceServer) *GRPC {
|
||
|
|
treasurer.RegisterTreasurerServiceServer(receiver.grpc, server)
|
||
|
|
|
||
|
|
return receiver
|
||
|
|
}
|
||
|
|
|
||
|
|
func (receiver *GRPC) listen(address string) error {
|
||
|
|
listener, err := net.Listen("tcp", address)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
return receiver.grpc.Serve(listener)
|
||
|
|
}
|