golang RPC包

时间:2022-09-29 whynogome 人气:0

本篇文章旨在通过学习rpc包和github上的一个rpc小项目，熟悉和学习golang中各个包的使用

工作流程

通过阅读官方文档，了解了rpc的基本工作模式

第一步，建立一个用于远程调用的包，存放仅供远程调用使用的方法和类型-
第二步，实例化包的对象，并在rpc中注册该包，以便之后的调用
第三步，建立一个服务端，接收客户端的请求，使用编码器解析请求后，根据请求中的方法和参数，调用第二步注册的实例的方法，然后使用编码器把返回值加密后，返回给客户端
第四步，建立一个客户端，连接服务端，成功后，向连接发送使用编码器加密后的数据，然后等待服务端响应（同步或异步）。响应成功后，使用编码器解析服务端返回的数据

第三步第四步中，多次用到的编码器，是rpc包的关键。默认情况下，rpc包使用的是go特有的encoding/gob包进行数据的编码和解码。但是当我们服务端和客户端使用了不同的语言时，若加密方法无法兼容，就会出现问题，所以rpc包支持自定义编码器。

工作模式

go的rpc除了支持常规常规的tpc+端口的远程调用方式，也支持基于http的远程调用实现。但是，我都用rpc了，还用个毛的http形式。不过作为一种形式，我们出于礼貌的简单了解下。

http模式

官方文档的例子，就是使用的http形式的rpc，如下

服务端

//实例化rpc远程调用的方法所属对象
arith := new(Arith)
//注册对象
rpc.Register(arith)
//把rpc监听 对应到http处理器。即指定http请求addr+port时，调用的方法
rpc.HandleHTTP()
//获取监听地址
l, e := net.Listen("tcp", ":1234")
if e != nil {
	log.Fatal("listen error:", e)
}
//开启一个go程，持续处理监听数据
go http.Serve(l, nil)

客户端

//连接 rpc的http服务端
client, err := rpc.DialHTTP("tcp", serverAddress + ":1234")
if err != nil {
	log.Fatal("dialing:", err)
}
//同步调用
// 实例化rpc传入参数
args := &server.Args{7,8}
//声明rpc 回复参数。传入和回复参数，必须与调用方法中的参入类型一致
var reply int
//调用rpc注册的方法
err = client.Call("Arith.Multiply", args, &reply)
if err != nil {
	log.Fatal("arith error:", err)
}
fmt.Printf("Arith: %d*%d=%d", args.A, args.B, reply)
//或：异步调用
// Asynchronous call
quotient := new(Quotient)
divCall := client.Go("Arith.Divide", args, quotient, nil)
replyCall := <-divCall.Done	// will be equal to divCall

从上面的代码可以看到，请求服务端时，需要客户端在服务器发起http请求，如果直接在浏览器或者其他工具发起http请求则报错，因为此时rpc.HandleHTTP方法指定的默认方法，使用的是默认gob编码器，且只接收connect类型的请求。查看源代码，如下

直接发起的http请求，无法使用go独有的gob包编码，rpc服务端也就无法使用默认gob包解码。

所以我们要写一个新的方法代替rpc.HandleHTTp，把http请求绑定到一个使用其他解码器的方法上，如下

//注册路由和对应的handler
http.HandleFunc("/json", func(rw http.ResponseWriter, r *http.Request) {
//声明一个客户端连接对象   
var conn io.ReadWriteCloser = struct {
      io.Writer
      io.ReadCloser
   }{
      ReadCloser: r.Body,
      Writer:     rw,
   }
//也可以使用如下方法获取接管客户端连接，http处理器不在管理该链接，使用完毕后需要自行关闭链接
// conn, _, err := rw.(http.Hijacker).Hijack()
  // if err != nil {
   //   log.Print("rpc hijacking fail: ", err.Error())
    //  return
   // }
   io.WriteString(conn, "HTTP/1.0 rpc-ok\n\n")
   //server.ServeConn(conn)
 //rpc.ServeRequest,指定编码器，以同步的方式处理请求一次，编码器内不关闭链接，由http服务处理。适用于http形式的请求，因为http是无状态的，每次请求都是一个新的链接
//rpc.ServeCodec,指定编码器，for循环接收客户端链接的消息，每次消息处理开启一个go程，相当于异步处理，直到客户端关闭或解码错误，跳出循环，关闭连接。适用客户端，一次连接多次发送数据
   rpc.ServeRequest(jsonrpc.NewServerCodec(conn))})
//监听http请求
http.ListenAndServe("127.0.0.1:1234", nil)

上面代码中的==jsonrpc.NewServerCodec(conn)==是一个官方定义好的json格式的编码器。我们在http中发送数据时，只要使用json格式，就能被服务端解析

执行go文件，在post模拟url请求，如下

服务器模式

服务端

//声明和注册rpc方法对象
//获取监听信息
lis, err := net.Listen("tcp", ":8082")
	if err != nil {
		log.Fatal(err)
}
//循环读取监听到的数据
for {
	//获取一个客户端连接
   conn, err := lis.Accept()
   if err != nil {
      continue
   }
   //开启一个go程序，使用自定义编码器处理当前获取连接
   go s.Server.ServeCodec(jsonrpc.NewServerCodec(conn))
}

客户端

//连接服务端
conn, err := net.Dial("tcp", ":8082")if err != nil {
	log.Fatal(err)
}
defer conn.Close()
//使用json编码器新建客户端
client := &Client{rpc.NewClientWithCodec(jsonrpc.NewServerCodec(conn))}
//声明rpc方法中传入和输出的参数
resq := message.ArithRequest{A: 20, B: 5}
resp := message.ArithResponse{}
//调用rpc方法
err = client.Call("ArithService.Add", &resq, &resp)
log.Printf("Arith.Add(%v, %v): %v ,Error: %v", resq.A, resq.B, resp.C, err)

加载全部内容