GolangRPC并发处理技巧分享

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

package main
import (
"log"
"net/rpc"
"sync"
)
func callRPC(wg sync.WaitGroup, client 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)
}
func main() {
client, err := rpc.Dial("tcp", "127.0.0.1:1234")
if err != nil {
log.Fatal("连接失败:", err)
}
defer client.Close()
var wg sync.WaitGroup
for i := 1; i <= 5; i++ {
    wg.Add(1)
    go callRPC(&wg, client, i, i+1)
}
wg.Wait()
}