/
client.go
71 lines (55 loc) · 1.4 KB
/
client.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
package main
import (
"context"
"flag"
"log"
"sort"
"strings"
"time"
example "github.com/rpcxio/rpcx-examples"
"github.com/smallnest/rpcx/client"
)
var (
addr1 = flag.String("addr1", "tcp@localhost:8972", "server address")
addr2 = flag.String("addr2", "tcp@localhost:8973", "server address")
)
func main() {
flag.Parse()
d := client.NewMultipleServersDiscovery([]*client.KVPair{{Key: *addr1}, {Key: *addr2}})
xclient := client.NewXClient("Arith", client.Failtry, client.SelectByUser, d, client.DefaultOption)
defer xclient.Close()
xclient.SetSelector(&alwaysFirstSelector{})
args := &example.Args{
A: 10,
B: 20,
}
for i := 0; i < 10; i++ {
reply := &example.Reply{}
err := xclient.Call(context.Background(), "Mul", args, reply)
if err != nil {
log.Fatalf("failed to call: %v", err)
}
log.Printf("%d * %d = %d", args.A, args.B, reply.C)
time.Sleep(time.Second)
}
}
type alwaysFirstSelector struct {
servers []string
}
func (s *alwaysFirstSelector) Select(ctx context.Context, servicePath, serviceMethod string, args interface{}) string {
var ss = s.servers
if len(ss) == 0 {
return ""
}
return ss[0]
}
func (s *alwaysFirstSelector) UpdateServer(servers map[string]string) {
var ss = make([]string, 0, len(servers))
for k := range servers {
ss = append(ss, k)
}
sort.Slice(ss, func(i, j int) bool {
return strings.Compare(ss[i], ss[j]) <= 0
})
s.servers = ss
}