Go语言集群计算实践：利用net/rpc构建分布式系统

// common.go
package main

// Args 定义RPC方法的参数结构体
type Args struct {
    A, B int
}

// Reply 定义RPC方法的返回值结构体
type Reply struct {
    C int
}

// Arith 是一个示例服务，包含可远程调用的方法
type Arith int

// Add 方法用于执行加法操作
func (t *Arith) Add(args *Args, reply *Reply) error {
    reply.C = args.A + args.B
    return nil
}

// server.go
package main

import (
    "log"
    "net"
    "net/rpc"
    "net/rpc/jsonrpc" // 也可以使用jsonrpc
)

func main() {
    // 注册Arith服务
    arith := new(Arith)
    rpc.Register(arith)

    // 监听TCP端口
    tcpAddr, err := net.ResolveTCPAddr("tcp", ":1234")
    if err != nil {
        log.Fatal("ResolveTCPAddr error:", err)
    }

    listener, err := net.ListenTCP("tcp", tcpAddr)
    if err != nil {
        log.Fatal("ListenTCP error:", err)
    }
    log.Println("RPC server listening on :1234")

    for {
        conn, err := listener.Accept()
        if err != nil {
            log.Println("Accept error:", err)
            continue
        }
        // 使用gob编码处理RPC连接
        go rpc.ServeConn(conn)
        // 如果想使用jsonrpc，可以这样：
        // go jsonrpc.ServeConn(conn)
    }
}

// client.go
package main

import (
    "fmt"
    "log"
    "net/rpc"
    "net/rpc/jsonrpc" // 如果服务器使用jsonrpc，客户端也需对应
)

func main() {
    // 连接到RPC服务器
    client, err := rpc.Dial("tcp", "localhost:1234")
    // 如果服务器使用jsonrpc，可以这样：
    // client, err := jsonrpc.Dial("tcp", "localhost:1234")
    if err != nil {
        log.Fatal("Dialing error:", err)
    }

    // 定义参数和返回值
    args := &Args{A: 7, B: 8}
    var reply Reply

    // 调用远程的Add方法
    err = client.Call("Arith.Add", args, &reply)
    if err != nil {
        log.Fatal("Arith.Add error:", err)
    }
    fmt.Printf("Arith: %d+%d=%d\n", args.A, args.B, reply.C)

    // 异步调用示例
    // var asyncReply Reply
    // call := client.Go("Arith.Add", args, &asyncReply, nil)
    // <-call.Done // 等待调用完成
    // if call.Error != nil {
    //  log.Fatal("Async Arith.Add error:", call.Error)
    // }
    // fmt.Printf("Async Arith: %d+%d=%d\n", args.A, args.B, asyncReply.C)

    client.Close()
}