/
grpc.go
63 lines (53 loc) · 1.19 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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
package tuna
import (
"context"
"net"
"os"
"os/signal"
"github.com/kataras/golog"
"google.golang.org/grpc"
)
// GRPC is responsible for handling gRPC server.
type GRPC struct {
server *grpc.Server
context context.Context
}
// Context returns the context used in gRPC.
func (g *GRPC) Context() context.Context {
return g.context
}
// Server returns gRPC server.
func (g *GRPC) Server() *grpc.Server {
return g.server
}
// Run starts the gRPC server on given address.
func (g *GRPC) Run(listener net.Listener) {
// graceful shutdown in interrupt signal
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt)
go func() {
for range c {
golog.Error("grpc: Server closed")
g.Shutdown()
<-g.context.Done()
}
}()
// start gRPC server
golog.Info("grpc: Server Started")
err := g.server.Serve(listener)
if err != nil {
golog.Fatal(err)
}
}
// Shutdown shutting down grpc server gracefully.
func (g *GRPC) Shutdown() {
g.server.GracefulStop()
}
// New instantiate the GRPC handler.
func New(ctx context.Context, opt ...grpc.ServerOption) *GRPC {
opt = append(opt, grpc.UnaryInterceptor(serverLogInterceptor))
return &GRPC{
server: grpc.NewServer(opt...),
context: ctx,
}
}