67 lines
1.7 KiB
Go
67 lines
1.7 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
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"
|
|
"net"
|
|
"gitea.pena/SQuiz/core/internal/initialize"
|
|
"gitea.pena/SQuiz/core/internal/proto/notifyer"
|
|
"time"
|
|
)
|
|
|
|
type GRPC struct {
|
|
grpc *grpc.Server
|
|
logger *zap.Logger
|
|
}
|
|
|
|
func NewGRPC(logger *zap.Logger) (*GRPC, error) {
|
|
grpcStreamInterceptor := grpc.StreamInterceptor(grpc_middleware.ChainStreamServer(
|
|
grpc_zap.StreamServerInterceptor(logger),
|
|
grpc_recovery.StreamServerInterceptor(),
|
|
))
|
|
|
|
grpcUnaryInterceptor := grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(
|
|
grpc_zap.UnaryServerInterceptor(logger),
|
|
grpc_recovery.UnaryServerInterceptor(),
|
|
))
|
|
|
|
return &GRPC{
|
|
grpc: grpc.NewServer(grpcStreamInterceptor, grpcUnaryInterceptor, grpc.ConnectionTimeout(5*time.Second)),
|
|
logger: logger,
|
|
}, nil
|
|
}
|
|
|
|
func (g *GRPC) Run(connectionString string) {
|
|
g.logger.Info("Starting GRPC Server", zap.String("host", connectionString))
|
|
|
|
if err := g.listen(connectionString); err != nil && err != grpc.ErrServerStopped {
|
|
g.logger.Error("GRPC Listen error", zap.Error(err))
|
|
}
|
|
}
|
|
|
|
func (g *GRPC) Stop(_ context.Context) error {
|
|
g.grpc.GracefulStop()
|
|
g.logger.Info("Shutting down GRPC server...")
|
|
|
|
return nil
|
|
}
|
|
|
|
func (g *GRPC) Register(reg initialize.GRpcControllers) *GRPC {
|
|
notifyer.RegisterQuizServiceServer(g.grpc, reg.MailNotify)
|
|
// another
|
|
return g
|
|
}
|
|
|
|
func (g *GRPC) listen(address string) error {
|
|
listener, err := net.Listen("tcp", address)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return g.grpc.Serve(listener)
|
|
}
|