Go 中使用 RPC

做 MIT-6.824 的 Lab 1 MapReduce 任务，Master 和 Worker 之间通信使用 RPC 调用。写一个 Demo 熟悉一下。

示例

package main

import "net"
import "net/rpc"
import "net/http"
import "errors"
import "log"
import "fmt"

type Args struct {
A, B int
}

type Qutient struct {
Quo, Rem int
}

// 别名
type Arith int

func (t *Arith) Multiply(args *Args, reply *int) error {
*reply = args.A * args.B
return nil
}

func (t* Arith) Divide(args *Args, quo *Qutient) error {
if args.B == 0 {
return errors.New("divide by zero")
}
quo.Quo = args.A / args.B
quo.Rem = args.A % args.B
return nil
}

func main() {
arith := new(Arith)
rpc.Register(arith)
rpc.HandleHTTP()
l, e := net.Listen("tcp", ":1234")
if e != nil {
log.Fatal("listen error:", e)
}
go http.Serve(l, nil)

// 调用
client, err := rpc.DialHTTP("tcp", "127.0.0.1" + ":1234")
if err != nil {
log.Fatal("dialing:", err)
}

args := &Args{7,8}
var reply int
err2 := client.Call("Arith.Multiply", args, &reply)
if err2 != nil {
log.Fatal("Arith error:", err2)
}

fmt.Printf("Arith: %d*%d=%d \n", args.A, args.B, reply)
}

注意的地方

1、Go 语言里可以使用 type 关键字来把一个类型来转换成另外一个类型而保持数据的本质不变。代码中的 Arith 是另外一种类型，量纲上是数字。

2、Go 中的 methods 是一个包含了接受者的函数。

参考

http://nil.csail.mit.edu/6.824/2015/labs/lab-1.html

/detail/2573043205.html

https://golang.org/pkg/net/rpc/