-
Notifications
You must be signed in to change notification settings - Fork 0
/
grpc_server.go
64 lines (53 loc) · 1.51 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
package grpcx
import (
"context"
"github.com/grpc-ecosystem/go-grpc-middleware"
"google.golang.org/grpc"
"google.golang.org/grpc/reflection"
"log"
"net"
)
// GrpcServer struct
type GrpcServer struct {
*grpc.Server
opts []grpc.UnaryServerInterceptor
routes []func(server *GrpcServer)
}
// NewGrpcServer create new GrpcServer with default configuration
func NewGrpcServer() *GrpcServer {
server := &GrpcServer{
opts: []grpc.UnaryServerInterceptor{
serverInterceptor,
},
}
return server
}
func (gs *GrpcServer) Use(f ...func(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error)) {
for _, vf := range f {
gs.opts = append(gs.opts, vf)
}
}
func (gs *GrpcServer) Register(f ...func(s *GrpcServer)) {
gs.routes = append(gs.routes, f...)
}
func (gs *GrpcServer) Run(addr string) (err error) {
// 初始化server, 将多个拦截器构建成一个拦截器
gs.Server = grpc.NewServer(
grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(gs.opts...)),
)
// 注册路由
for _, vf := range gs.routes {
vf(gs)
}
lis, err := net.Listen("tcp", addr)
if err != nil {
log.Printf("failed to listen: %v", err)
return
}
// 在给定的gRPC服务器上注册服务器反射服务
reflection.Register(gs.Server)
// Serve方法在lis上接受传入连接,为每个连接创建一个ServerTransport和server的goroutine。
// 该goroutine读取gRPC请求,然后调用已注册的处理程序来响应它们
err = gs.Server.Serve(lis)
return
}