• golang RPC包的使用和源码学习(上):基本原理和简单使用



    本篇文章旨在通过学习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)
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
    • 客户端
    //连接 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
    
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
    • 14
    • 15
    • 16
    • 17
    • 18
    • 19
    • 20
    • 21
    • 22

    从上面的代码可以看到,请求服务端时,需要客户端在服务器发起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)
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
    • 14
    • 15
    • 16
    • 17
    • 18
    • 19
    • 20
    • 21
    • 22
    • 23

    上面代码中的==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))
    }
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
    • 14
    • 15
    • 16
    • 客户端
    //连接服务端
    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)
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
  • 相关阅读:
    在您的下一个项目中选择 Golang 和 Node.js 之间的抉择
    ML:机器学习工程化之团队十大角色背景、职责、产出物划分之详细攻略
    趋化因子 CXCL12 中性配体-LIT927--有望治疗炎症疾病
    【安全狗】Linux后渗透常见后门驻留方式分析
    如何将系统盘MBR转GPT?无损教程分享!
    CentOS Python环境搭建
    16:00面试,16:06就出来了,问的问题有点变态。。。
    气膜建筑的可持续性:能源效益与环境影响
    Go 语言 iota 的神奇力量
    Linux软件包管理— yum命令
  • 原文地址:https://blog.csdn.net/u012830303/article/details/126370703