脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → Golang → Go RPC框架

Go实现简易RPC框架的方法步骤

 更新时间：2019年03月13日 14:25:14   作者：Jiahonzheng  
本文旨在讲述 RPC 框架设计中的几个核心问题及其解决方法，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧

本文旨在讲述 RPC 框架设计中的几个核心问题及其解决方法，并基于 Golang 反射技术，构建了一个简易的 RPC 框架。

项目地址：Tiny-RPC

RPC

RPC（Remote Procedure Call），即远程过程调用，可以理解成，服务 A 想调用不在同一内存空间的服务 B 的函数，由于不在一个内存空间，不能直接调用，需要通过网络来表达调用的语义和传达调用的数据。

服务端

RPC 服务端需要解决 2 个问题：

  • 由于客户端传送的是 RPC 函数名，服务端如何维护 函数名 与 函数实体 之间的映射
  • 服务端如何根据 函数名 实现对应的 函数实体 的调用

核心流程

  • 维护函数名到函数的映射
  • 在接收到来自客户端的函数名、参数列表后，解析参数列表为反射值，并执行对应函数
  • 对函数执行结果进行编码，并返回给客户端

方法注册

服务端需要维护 RPC 函数名到 RPC 函数实体的映射，我们可以使用 map 数据结构来维护映射关系。

type Server struct {
 addr string
 funcs map[string]reflect.Value
}

// Register a method via name
func (s *Server) Register(name string, f interface{}) {
 if _, ok := s.funcs[name]; ok {
 return
 }
 s.funcs[name] = reflect.ValueOf(f)
}

执行调用

一般来说，客户端在调用 RPC 时，会将 函数名 和 参数列表 作为请求数据，发送给服务端。

由于我们使用了 map[string]reflect.Value 来维护函数名与函数实体之间的映射，则我们可以通过 Value.Call() 来调用与函数名相对应的函数。

package main

import (
 "fmt"
 "reflect"
)

func main() {
 // Register methods
 funcs := make(map[string]reflect.Value)
 funcs["add"] = reflect.ValueOf(add)

 // When receives client's request
 req := []reflect.Value{reflect.ValueOf(1), reflect.ValueOf(2)}
 vals := funcs["add"].Call(req)
 var rsp []interface{}
 for _, val := range vals {
 rsp = append(rsp, val.Interface())
 }

 fmt.Println(rsp)
}

func add(a, b int) (int, error) {
 return a + b, nil
}

具体实现

由于篇幅的限制，此处没有贴出服务端实现的具体代码，细节请查看项目地址。

客户端

RPC 客户端需要解决 1 个问题：

  • 由于函数的具体实现在服务端，客户端只有函数的原型，客户端如何通过 函数原型 调用其 函数实体

核心流程

  • 对调用者传入的函数参数进行编码，并传送给服务端
  • 对服务端响应数据进行解码，并返回给调用者

生成调用

我们可以通过 reflect.MakeFunc 为指定的函数原型绑定一个函数实体。

package main

import (
 "fmt"
 "reflect"
)

func main() {
 add := func(args []reflect.Value) []reflect.Value {
 result := args[0].Interface().(int) + args[1].Interface().(int)
 return []reflect.Value{reflect.ValueOf(result)}
 }

 var addptr func(int, int) int
 container := reflect.ValueOf(&addptr).Elem()
 v := reflect.MakeFunc(container.Type(), add)
 container.Set(v)

 fmt.Println(addptr(1, 2))
}

具体实现

由于篇幅的限制，此处没有贴出客户端实现的具体代码，细节请查看项目地址。

数据传输格式

我们需要定义服务端与客户端交互的数据格式。

type Data struct {
 Name string    // service name
 Args []interface{} // request's or response's body except error
 Err string    // remote server error
}

与交互数据相对应的编码与解码函数。

func encode(data Data) ([]byte, error) {
 var buf bytes.Buffer
 encoder := gob.NewEncoder(&buf)
 if err := encoder.Encode(data); err != nil {
 return nil, err
 }
 return buf.Bytes(), nil
}

func decode(b []byte) (Data, error) {
 buf := bytes.NewBuffer(b)
 decoder := gob.NewDecoder(buf)
 var data Data
 if err := decoder.Decode(&data); err != nil {
 return Data{}, err
 }
 return data, nil
}

同时，我们需要定义简单的 TLV 协议（固定长度消息头 + 变长消息体），规范数据的传输。

// Transport struct
type Transport struct {
 conn net.Conn
}

// NewTransport creates a transport
func NewTransport(conn net.Conn) *Transport {
 return &Transport{conn}
}

// Send data
func (t *Transport) Send(req Data) error {
 b, err := encode(req) // Encode req into bytes
 if err != nil {
 return err
 }
 buf := make([]byte, 4+len(b))
 binary.BigEndian.PutUint32(buf[:4], uint32(len(b))) // Set Header field
 copy(buf[4:], b)                  // Set Data field
 _, err = t.conn.Write(buf)
 return err
}

// Receive data
func (t *Transport) Receive() (Data, error) {
 header := make([]byte, 4)
 _, err := io.ReadFull(t.conn, header)
 if err != nil {
 return Data{}, err
 }
 dataLen := binary.BigEndian.Uint32(header) // Read Header filed
 data := make([]byte, dataLen)       // Read Data Field
 _, err = io.ReadFull(t.conn, data)
 if err != nil {
 return Data{}, err
 }
 rsp, err := decode(data) // Decode rsp from bytes
 return rsp, err
}

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

  • Go strconv包实现字符串和基本数据类型转换的实例详解

    Go strconv包实现字符串和基本数据类型转换的实例详解

    在Go语言（Golang）的编程实践中,strconv包是一个非常重要的标准库,它提供了在基本数据类型（如整型、浮点型、布尔型）和字符串之间的转换功能,本文给大家介绍了关于Go语言字符串转换strconv,需要的朋友可以参考下
    2024-09-09
  • golang gorm学习之如何指定数据表

    golang gorm学习之如何指定数据表

    在sql中首先要指定是从哪张表中查询,所以这篇文章小编就来带大家一起看一下gorm是如何根据model来自动解析表名的,感兴趣的小伙伴可以了解下
    2023-08-08
  • 使用Golang创建单独的WebSocket会话

    使用Golang创建单独的WebSocket会话

    WebSocket是一种在Web开发中非常常见的通信协议,它提供了双向、持久的连接,适用于实时数据传输和实时通信场景,本文将介绍如何使用 Golang 创建单独的 WebSocket 会话,包括建立连接、消息传递和关闭连接等操作,需要的朋友可以参考下
    2023-12-12
  • 在Go中复制文件最流行的3种方法

    在Go中复制文件最流行的3种方法

    今天小编就为大家分享一篇关于在Go中复制文件最流行的3种方法，小编觉得内容挺不错的，现在分享给大家，具有很好的参考价值，需要的朋友一起跟随小编来看看吧
    2018-10-10
  • go语言实现并发网络爬虫的示例代码

    go语言实现并发网络爬虫的示例代码

    本文主要介绍了go语言实现并发网络爬虫的示例代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2023-03-03
  • GoLang中生成UUID唯一标识的实现方法

    GoLang中生成UUID唯一标识的实现方法

    UUID是让分散式系统中的所有元素,都能有唯一的辨识信息,本文主要介绍了GoLang中生成UUID唯一标识的实现方法,具有一定的参考价值,感兴趣的可以了解一下
    2024-08-08
  • Golang 使用Map实现去重与set的功能操作

    Golang 使用Map实现去重与set的功能操作

    这篇文章主要介绍了Golang 使用 Map 实现去重与 set 的功能操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2021-04-04
  • Golang易错知识点汇总

    Golang易错知识点汇总

    这篇文章汇总了在开发和刷面试题过程中遇到的Golang容易搞错的知识点，关键部分也都为大家写了代码示例，感兴趣的小伙伴可以了解一下
    2022-09-09
  • 详解Go语言单元测试中如何解决MySQL存储依赖问题

    详解Go语言单元测试中如何解决MySQL存储依赖问题

    MySQL 存储就是一个非常常见的外部依赖，这篇文章主要来和大家一起探讨在 Go 语言中编写单元测试时，如何解决 MySQL 存储依赖，需要的可以参考一下
    2023-07-07
  • go解析svn log生成的xml格式的文件

    go解析svn log生成的xml格式的文件

    这篇文章主要介绍了go解析svn log生成的xml格式的文件的方法，非常的实用，有需要的小伙伴可以参考下。
    2015-04-04

最新评论