Golang

关注公众号 jb51net

关闭
首页 > 脚本专栏 > Golang > golang RPC包

golang RPC包原理和使用详细介绍

作者:whynogome

golang的rpc支持三个级别的RPC:TCP、HTTP、JSONRPC。但Go的RPC包是独一无二的RPC,它和传统的RPC系统不同,它只支持Go开发的服务器与客户端之间的交互,因为在内部,它们采用了Gob来编码

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

工作流程

通过阅读官方文档,了解了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)

到此这篇关于golang RPC包原理和使用详细介绍的文章就介绍到这了,更多相关golang RPC包内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文