/
grpc_server.go
70 lines (56 loc) · 2.72 KB
/
grpc_server.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package api
import (
"fmt"
"net"
"runtime/debug"
"github.com/aws/aws-sdk-go-v2/service/s3"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
"github.com/DIMO-Network/shared/db"
"github.com/DIMO-Network/vehicle-signal-decoding/internal/config"
pkggrpc "github.com/DIMO-Network/vehicle-signal-decoding/pkg/grpc"
grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
"github.com/grpc-ecosystem/go-grpc-middleware/v2/interceptors/recovery"
"github.com/rs/zerolog"
"google.golang.org/grpc"
)
func StartGrpcServer(logger zerolog.Logger, dbs func() *db.ReaderWriter, s *config.Settings, s3Client *s3.Client) {
lis, err := net.Listen("tcp", fmt.Sprintf(":%v", s.GRPCPort))
if err != nil {
logger.Fatal().Msgf("Failed to listen on port %v: %v", s.GRPCPort, err)
}
vehicleSignalDecodingService := NewGrpcService(&logger, dbs, s3Client, s)
templateConfigService := NewTemplateConfigService(&logger, dbs)
pidConfigService := NewPidConfigService(&logger, dbs)
deviceSettingsService := NewDeviceSettingsConfigService(&logger, dbs)
dbcConfigService := NewDbcConfigService(&logger, dbs)
vehicleTemplateService := NewVehicleTemplateService(&logger, dbs)
templateDeviceDefinitionService := NewTemplateDeviceDefinitionService(&logger, dbs)
aftermarketDeviceTemplateService := NewAftermarketDeviceTemplateService(&logger, dbs)
grpcRecovery := GRPCPanicker{Logger: &logger}
server := grpc.NewServer(
grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(
recovery.UnaryServerInterceptor(recovery.WithRecoveryHandler(grpcRecovery.GRPCPanicRecoveryHandler)),
)),
)
pkggrpc.RegisterVehicleSignalDecodingServiceServer(server, vehicleSignalDecodingService)
pkggrpc.RegisterTemplateConfigServiceServer(server, templateConfigService)
pkggrpc.RegisterPidConfigServiceServer(server, pidConfigService)
pkggrpc.RegisterDeviceSettingsServiceServer(server, deviceSettingsService)
pkggrpc.RegisterDbcConfigServiceServer(server, dbcConfigService)
pkggrpc.RegisterVehicleTemplateServiceServer(server, vehicleTemplateService)
pkggrpc.RegisterTemplateDeviceDefinitionServiceServer(server, templateDeviceDefinitionService)
pkggrpc.RegisterAftermarketDeviceTemplateServiceServer(server, aftermarketDeviceTemplateService)
logger.Info().Str("port", s.GRPCPort).Msgf("started grpc server on port: %v", s.GRPCPort)
if err := server.Serve(lis); err != nil {
logger.Fatal().Msgf("Failed to serve over port %v: %v", s.GRPCPort, err)
}
}
type GRPCPanicker struct {
Logger *zerolog.Logger
}
func (pr *GRPCPanicker) GRPCPanicRecoveryHandler(p any) (err error) {
//appmetrics.GRPCPanicsCount.Inc()
pr.Logger.Err(fmt.Errorf("%s", p)).Str("stack", string(debug.Stack())).Msg("grpc recovered from panic")
return status.Errorf(codes.Internal, "%s", p)
}