/
server.go
125 lines (99 loc) · 3.09 KB
/
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
// Copyright 2017 AMIS Technologies
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package rpc
import (
"crypto/tls"
"net"
"net/http"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials"
"github.com/getamis/sirius/metrics"
grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
)
// NewServer creates a gRPC server with pre-configured services
func NewServer(opts ...ServerOption) *Server {
server := &Server{}
for _, opt := range opts {
opt(server)
}
server.createGRPCServer()
server.registerAPIs()
server.initMetrics()
return server
}
// API provides APIs for specific gRPC server
//go:generate mockery -name API
type API interface {
Bind(server *grpc.Server)
}
// Server represents a gRPC server
type Server struct {
grpcServer *grpc.Server
credentials *tls.Config
grpcMetrics metrics.ServerMetrics
streamInterceptors []grpc.StreamServerInterceptor
unaryInterceptors []grpc.UnaryServerInterceptor
serverOpts []grpc.ServerOption
apis []API
}
func (s *Server) Serve(l net.Listener) error {
return s.grpcServer.Serve(l)
}
func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
s.grpcServer.ServeHTTP(w, r)
}
func (s *Server) Shutdown() {
s.grpcServer.GracefulStop()
type handler interface {
Shutdown()
}
for _, api := range s.apis {
h, ok := api.(handler)
if ok {
h.Shutdown()
}
}
}
// ----------------------------------------------------------------------------
func (s *Server) createGRPCServer() {
options := s.serverOpts
// credentials
if s.credentials != nil {
tls := credentials.NewTLS(s.credentials)
options = append(options, grpc.Creds(tls))
}
var streamInterceptors []grpc.StreamServerInterceptor
var unaryInterceptors []grpc.UnaryServerInterceptor
// metrics
if s.grpcMetrics != nil {
streamInterceptors = append(streamInterceptors, s.grpcMetrics.StreamServerInterceptor())
unaryInterceptors = append(unaryInterceptors, s.grpcMetrics.UnaryServerInterceptor())
}
streamInterceptors = append(streamInterceptors, s.streamInterceptors...)
unaryInterceptors = append(unaryInterceptors, s.unaryInterceptors...)
// chain interceptors
options = append(options, grpc.StreamInterceptor(grpc_middleware.ChainStreamServer(streamInterceptors...)))
options = append(options, grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(unaryInterceptors...)))
s.grpcServer = grpc.NewServer(options...)
}
func (s *Server) registerAPIs() {
for _, api := range s.apis {
api.Bind(s.grpcServer)
}
}
func (s *Server) initMetrics() {
if s.grpcMetrics != nil {
s.grpcMetrics.InitializeMetrics(s.grpcServer)
}
}