grpc_microservice.go

package microservice

import (
	context "context"
	"net"

	"github.com/bldsoft/geos/pkg/controller"
	grpc_controller "github.com/bldsoft/geos/pkg/controller/grpc"
	pb "github.com/bldsoft/geos/pkg/controller/grpc/proto"
	"github.com/bldsoft/geos/pkg/microservice/middleware"
	"github.com/bldsoft/gost/log"
	grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
	grpc "google.golang.org/grpc"
	"google.golang.org/grpc/grpclog"
)

type GrpcMicroservice struct {
	address        string
	grpcServer     *grpc.Server
	geoIpService   controller.GeoIpService
	geoNameService controller.GeoNameService
}

func NewGrpcMicroservice(address string, geoIpService controller.GeoIpService, geoNameService controller.GeoNameService) *GrpcMicroservice {
	return &GrpcMicroservice{
		address:        address,
		geoIpService:   geoIpService,
		geoNameService: geoNameService,
	}
}

func (s *GrpcMicroservice) registerServices() {
	geoIpController := grpc_controller.NewGeoIpController(s.geoIpService)
	pb.RegisterGeoIpServiceServer(s.grpcServer, geoIpController)
	geoNameController := grpc_controller.NewGeoNameController(s.geoNameService)
	pb.RegisterGeoNameServiceServer(s.grpcServer, geoNameController)
}

func (s *GrpcMicroservice) Run() error {
	grpclog.SetLoggerV2(middleware.Logger)

	lis, err := net.Listen("tcp", s.address)
	if err != nil {
		return err
	}

	s.grpcServer = grpc.NewServer(grpc.UnaryInterceptor(
		grpc_middleware.ChainUnaryServer(
			middleware.RequestIDMiddleware,
			middleware.RealIPMiddleware,
			middleware.LoggerMiddleware(),
			middleware.RecoveryMiddleware,
		),
	))
	s.registerServices()

	log.Infof("Grpc server started. Listening on %s", s.address)
	defer log.Infof("Grpc server stopped")
	return s.grpcServer.Serve(lis)
}

func (s *GrpcMicroservice) Stop(ctx context.Context) error {
	stopped := make(chan struct{})
	go func() {
		defer close(stopped)
		s.grpcServer.GracefulStop()
	}()
	select {
	case <-stopped:
		return nil
	case <-ctx.Done():
		return ctx.Err()
	}
}