Go RPC 库
RPC(Remote Poresedure Call)是远程方法调用的缩写。Go的RPC库可以实现通过网络或者其他I/O方式远程调用对象的方法。
服务器注册一个对象，让它作为一个以对象类型命名的服务，让这个对象导出的方法可以被远程调用。一个服务器可以注册多个不同类型的对象，但是不能注册同一类型的多个对象。
一个能够被远程调用的方法应该像下面这样：
func (t *T) MethodName(argType T1, replyType *T2) error
一个简单的例子：实现简单的kv存储，并（在同一台机器上）通过RPC调用Put和Get方法。
package main
import (
"fmt"
"log"
"net"
"net/http"
"net/rpc"
"sync"
"time"
)
// RPC struct definition
type PutArgs struct {
Key string
Val string
}
type PutReply struct {
Ok bool
}
type GetArgs struct {
Key string
}
type GetReply struct {
Val string
}
type Kv struct {
data map[string]string
mu sync.Mutex
}
// server
func (kv *Kv) Get(args *GetArgs, reply *GetReply) error {
kv.mu.Lock()
defer kv.mu.Unlock()
reply.Val = kv.data[args.Key]
return nil
}
func (kv *Kv) Put(args *PutArgs, reply *PutReply) error {
kv.mu.Lock()
defer kv.mu.Unlock()
kv.data[args.Key] = args.Val
reply.Ok = true
return nil
}
func KvServer() {
kv := new(Kv)
kv.data = make(map[string]string)
rpc.Register(kv)
rpc.HandleHTTP()
l, err := net.Listen("tcp", ":1234")
if err != nil {
log.Fatal("listen error: ", err)
}
http.Serve(l, nil)
}
// client interface
func get(key string) string {
client, err := rpc.DialHTTP("tcp", "127.0.0.1"+":1234")
if err != nil {
log.Fatal("dialing:", err)
}
args := GetArgs{key}
reply := GetReply{}
e := client.Call("Kv.Get", args, &reply)
if e != nil {
log.Fatal("get error", e)
}
client.Close()
return reply.Val
}
func put(key string, val string) bool {
client, err := rpc.DialHTTP("tcp", "127.0.0.1"+":1234")
if err != nil {
log.Fatal("dialing:", err)
}
args := PutArgs{key, val}
reply := PutReply{}
e := client.Call("Kv.Put", args, &reply)
if e != nil {
log.Fatal("put error", e)
}
client.Close()
return reply.Ok
}
// test
func main() {
go KvServer()
time.Sleep(time.Second) // wait for server to start
fmt.Println(put("key1", "value1"))
fmt.Println(put("key2", "value2"))
fmt.Println(put("key3", "value3"))
fmt.Println(get("key1"))
fmt.Println(get("key2"))
fmt.Println(get("key3"))
fmt.Println(get("value1"))
}