Golang原生rpc(rpc服务端源码解读)

Golang原生rpc服务端源码解读

什么是RPC

RPC是Remote Procedure Call的缩写，译为远程过程调用。它允许像调用本地函数一样调用远程函数。

在分布式系统中，不同的机器上运行着不同的进程，这些进程需要相互通信才能协同工作。RPC技术使得分布式系统中的进程间通信变得简单易行，让开发分布式系统的复杂性得以降低。

Golang原生rpc服务端源码解读

在Golang中，可以使用net/rpc包提供的原生RPC服务，实现远程过程调用。下面是一个简单的RPC服务端源码解读。

导入相关的包

import (
    "errors"
    "log"
    "net"
    "net/rpc"
)

定义服务端结构体

定义服务端结构体，并添加方法作为RPC服务端提供的函数。此处的方法Arith.Multiply即为RPC服务端提供的函数。

type Arith struct{}

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

启动RPC服务端

创建TCP服务，并使用rpc.ServeConn将服务进行绑定，使其能够接收来自客户端的请求。

func main() {
    arith := new(Arith) // 实例化服务端结构体
    rpc.Register(arith) // 注册RPC服务端的方法
    rpc.HandleHTTP()

    // 启动TCP服务
    l, e := net.Listen("tcp", ":1234")
    if e != nil {
        log.Fatal("listen error:", e)
    }

    defer l.Close()

    // 监听来自客户端的请求，并对请求进行处理
    for {
        conn, err := l.Accept()
        if err != nil {
            log.Fatal("accept error:", err)
        }
        go rpc.ServeConn(conn)
    }
}

示例说明

以下是两个示例说明：

示例1：远程调用RPC服务端函数

在客户端中，可以使用net/rpc包提供的Dial函数，建立与RPC服务端的连接，并进行远程调用：

conn, err := rpc.Dial("tcp", "localhost:1234")
if err != nil {
    log.Fatal("dialing:", err)
}

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

log.Printf("Arith: %d*%d=%d", args.A, args.B, reply)

上述代码中，Dial函数建立了TCP连接，并指定服务端的地址。Call函数用于调用服务端的Arith.Multiply函数，并传递参数args，最终的返回值会存储到reply变量中。

示例2：并发远程调用RPC服务端函数

在客户端中，还可以使用go关键字实现并发调用RPC服务端函数：

done := make(chan int, 2)

go func() {
    conn, err := rpc.Dial("tcp", "localhost:1234")
    if err != nil {
        log.Fatal("dialing:", err)
    }

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

    log.Printf("Arith: %d*%d=%d", args.A, args.B, reply)
    done <- 1
}()

go func() {
    conn, err := rpc.Dial("tcp", "localhost:1234")
    if err != nil {
        log.Fatal("dialing:", err)
    }

    args := &Args{2, 3}
    var reply int
    err = conn.Call("Arith.Multiply", args, &reply)
    if err != nil {
        log.Fatal("arith error:", err)
    }

    log.Printf("Arith: %d*%d=%d", args.A, args.B, reply)
    done <- 1
}()

<-done
<-done

上述代码中，定义了一个done通道，用于协调并发调用RPC服务端的函数。通过go关键字，实例化了两个子协程，分别调用了Arith.Multiply函数，并将结果存储到reply变量。最后，通过<-done等待子协程执行完毕。