GolangRPC并发处理技巧分享

package main
<p>import (
"log"
"net"
"net/rpc"
)</p><p>type Args struct {
A, B int
}</p><p>type Calculator int</p><p>func (c <em>Calculator) Multiply(args Args, reply </em>int) error {
<em>reply = args.A </em> args.B
log.Printf("计算 %d <em> %d = %d", args.A, args.B, </em>reply)
return nil
}</p><p>func main() {
rpc.Register(new(Calculator))
listener, err := net.Listen("tcp", ":1234")
if err != nil {
log.Fatal("监听失败:", err)
}
defer listener.Close()</p><pre class="brush:php;toolbar:false;">log.Println("RPC服务器启动在 :1234")
for {
    conn, err := listener.Accept()
    if err != nil {
        continue
    }
    go rpc.ServeConn(conn) // 每个连接由独立Goroutine处理
}

}

package main
<p>import (
"log"
"net/rpc"
"sync"
)</p><p>func callRPC(wg <em>sync.WaitGroup, client </em>rpc.Client, i, j int) {
defer wg.Done()
args := Args{i, j}
var result int
err := client.Call("Calculator.Multiply", args, &result)
if err != nil {
log.Printf("调用失败: %v", err)
return
}
log.Printf("%d * %d = %d", i, j, result)
}</p><p>func main() {
client, err := rpc.Dial("tcp", "127.0.0.1:1234")
if err != nil {
log.Fatal("连接失败:", err)
}
defer client.Close()</p><pre class="brush:php;toolbar:false;">var wg sync.WaitGroup
for i := 1; i <= 5; i++ {
    wg.Add(1)
    go callRPC(&wg, client, i, i+1)
}
wg.Wait()

}