/
worc.go
82 lines (68 loc) · 2 KB
/
worc.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
package worc
import (
"fmt"
"log"
"reflect"
"golang.org/x/net/context"
"google.golang.org/grpc"
"google.golang.org/grpc/resolver"
"github.com/Wang-Kai/hi"
)
var serviceConns = newSafeMap()
func init() {
hiBuilder := hi.NewResolverBuilder([]string{"127.0.0.1:2379"})
resolver.Register(&hiBuilder)
}
// StartServiceConns start grpc conns with balancer
func StartServiceConns(address string, serviceList []string) {
for _, serviceName := range serviceList {
go func(name string) {
var dialAddr = fmt.Sprintf("hi://foo/%s", name)
conn, err := grpc.Dial(dialAddr, grpc.WithBalancerName("round_robin"), grpc.WithInsecure(), grpc.WithMaxMsgSize(1024*1024*128))
if err != nil {
log.Printf(`connect to '%s' service failed: %v`, name, err)
}
serviceConns.Set(name, conn)
}(serviceName)
}
}
// CloseServiceConns close all established conns
func CloseServiceConns() {
for _, conn := range serviceConns.List() {
conn.Close()
}
}
// CallRPC is helper func that make life easier
// ctx: context
// client: grpc client
// serviceName: name of service
// metod: method name that you want to use
// req: grpc request
func CallRPC(ctx context.Context, client interface{}, serviceName string, method string, req interface{}) (ret interface{}, err error) {
defer func() {
if x := recover(); x != nil {
err = fmt.Errorf("call RPC '%s' error: %v", method, x)
}
}()
conn := serviceConns.Get(serviceName)
if conn == nil {
return nil, fmt.Errorf("service conn '%s' not found", serviceName)
}
// get NewServiceClient's reflect.Value
vClient := reflect.ValueOf(client)
var vParameter []reflect.Value
vParameter = append(vParameter, reflect.ValueOf(conn))
// c[0] is serviceServer reflect.Value
c := vClient.Call(vParameter)
// rpc param
v := make([]reflect.Value, 2)
v[0] = reflect.ValueOf(ctx)
v[1] = reflect.ValueOf(req)
// rpc method call
f := c[0].MethodByName(method)
resp := f.Call(v)
if !resp[1].IsNil() {
return nil, resp[1].Interface().(error)
}
return resp[0].Interface(), nil
}