/
grpc.go
46 lines (37 loc) · 1.08 KB
/
grpc.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
package grpc
import (
"log"
"net"
"os"
"os/signal"
v1 "github.com/cecepsprd/gokit-skeleton/api/proto/v1"
"github.com/cecepsprd/gokit-skeleton/commons/cache"
"github.com/cecepsprd/gokit-skeleton/commons/config"
"github.com/cecepsprd/gokit-skeleton/internal/endpoint"
"github.com/cecepsprd/gokit-skeleton/internal/service"
"google.golang.org/grpc"
)
func RunServer(personSvc service.PersonService, cachePerson cache.PersonCache, cfg config.Config) error {
var (
personEndpoint = endpoint.MakePersonEndpoint(personSvc, cachePerson)
)
grpcServer := NewGRPCServer(personEndpoint, nil)
listen, err := net.Listen("tcp", cfg.App.GrpcPort)
if err != nil {
return err
}
baseGrpcServer := grpc.NewServer()
v1.RegisterPersonServer(baseGrpcServer, grpcServer)
// graceful shutdown
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt)
go func() {
for range c {
log.Println("shutting down gRPC server...")
baseGrpcServer.GracefulStop()
}
}()
// start gRPC server
log.Println("starting Grpc server on port", cfg.App.GrpcPort)
return baseGrpcServer.Serve(listen)
}