做 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 是一个包含了接受者的函数。
参考