/
main.go
61 lines (56 loc) · 1.32 KB
/
main.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
package main
import (
"context"
"net"
"os"
"os/signal"
"syscall"
"time"
"log"
grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
"github.com/joho/godotenv"
pbActivity "github.com/vuonghp92/grpc-mcro-demo/proto/activity"
"github.com/vuonghp92/grpc-mcro-demo/shared/interceptor"
"google.golang.org/grpc"
)
const port = ":50064"
func main() {
godotenv.Load()
srv := grpc.NewServer(grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(
interceptor.XTraceID(),
interceptor.Logging(),
interceptor.XUserID(),
)))
pbActivity.RegisterActivityServiceServer(srv, &ActivityService{
store: NewStoreOnMemory(),
})
go func() {
listener, err := net.Listen("tcp", port)
if err != nil {
log.Fatalf("failed to create listener: %s",
err)
}
log.Println("start server on port", port)
if err := srv.Serve(listener); err != nil {
log.Println("failed to exit serve: ", err)
}
}()
sigint := make(chan os.Signal, 1)
signal.Notify(sigint, syscall.SIGTERM)
<-sigint
log.Println("received a signal of graceful shutdown")
stopped := make(chan struct{})
go func() {
srv.GracefulStop()
close(stopped)
}()
ctx, cancel := context.WithTimeout(
context.Background(), 1*time.Minute)
select {
case <-ctx.Done():
srv.Stop()
case <-stopped:
cancel()
}
log.Println("completed graceful shutdown")
}