Go RPC 库

RPC(Remote Poresedure Call)是远程方法调用的缩写。Go的RPC库可以实现通过网络或者其他I/O方式远程调用对象的方法。
服务器注册一个对象，让它作为一个以对象类型命名的服务，让这个对象导出的方法可以被远程调用。一个服务器可以注册多个不同类型的对象，但是不能注册同一类型的多个对象。
一个能够被远程调用的方法应该像下面这样：

1. func (t *T) MethodName(argType T1, replyType *T2) error

复制代码

一个简单的例子：实现简单的kv存储，并（在同一台机器上）通过RPC调用Put和Get方法。

1. package main
3. import (
4.         "fmt"
5.         "log"
6.         "net"
7.         "net/http"
8.         "net/rpc"
9.         "sync"
10.         "time"
11. )
13. // RPC struct definition
15. type PutArgs struct {
16.         Key string
17.         Val string
18. }
20. type PutReply struct {
21.         Ok bool
22. }
24. type GetArgs struct {
25.         Key string
26. }
28. type GetReply struct {
29.         Val string
30. }
32. type Kv struct {
33.         data map[string]string
34.         mu   sync.Mutex
35. }
37. // server
39. func (kv *Kv) Get(args *GetArgs, reply *GetReply) error {
40.         kv.mu.Lock()
41.         defer kv.mu.Unlock()
42.         reply.Val = kv.data[args.Key]
43.         return nil
44. }
46. func (kv *Kv) Put(args *PutArgs, reply *PutReply) error {
47.         kv.mu.Lock()
48.         defer kv.mu.Unlock()
49.         kv.data[args.Key] = args.Val
50.         reply.Ok = true
51.         return nil
52. }
54. func KvServer() {
55.         kv := new(Kv)
56.         kv.data = make(map[string]string)
57.         rpc.Register(kv)
58.         rpc.HandleHTTP()
59.         l, err := net.Listen("tcp", ":1234")
60.         if err != nil {
61.                 log.Fatal("listen error: ", err)
62.         }
63.         http.Serve(l, nil)
64. }
66. // client interface
68. func get(key string) string {
69.         client, err := rpc.DialHTTP("tcp", "127.0.0.1"+":1234")
70.         if err != nil {
71.                 log.Fatal("dialing:", err)
72.         }
73.         args := GetArgs{key}
74.         reply := GetReply{}
75.         e := client.Call("Kv.Get", args, &reply)
76.         if e != nil {
77.                 log.Fatal("get error", e)
78.         }
79.         client.Close()
80.         return reply.Val
81. }
83. func put(key string, val string) bool {
84.         client, err := rpc.DialHTTP("tcp", "127.0.0.1"+":1234")
85.         if err != nil {
86.                 log.Fatal("dialing:", err)
87.         }
88.         args := PutArgs{key, val}
89.         reply := PutReply{}
90.         e := client.Call("Kv.Put", args, &reply)
91.         if e != nil {
92.                 log.Fatal("put error", e)
93.         }
94.         client.Close()
95.         return reply.Ok
96. }
98. // test
100. func main() {
101.         go KvServer()
102.         time.Sleep(time.Second) // wait for server to start
103.         fmt.Println(put("key1", "value1"))
104.         fmt.Println(put("key2", "value2"))
105.         fmt.Println(put("key3", "value3"))
106.         fmt.Println(get("key1"))
107.         fmt.Println(get("key2"))
108.         fmt.Println(get("key3"))
109.         fmt.Println(get("value1"))
110. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！