Go实现简易RPC框架的方法步骤
作者:Jiahonzheng 发布时间:2024-04-26 17:16:33
本文旨在讲述 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
}
来源:https://juejin.im/post/5c85ec1e5188257e937dd36a
猜你喜欢
- python实现四舍五入""" 四舍五入 :param
- 本文实例讲述了JS与jQuery判断文本框还剩多少字符可以输入的方法。分享给大家供大家参考,具体如下:javascript部分:functi
- SQL 联合查询与XML解析实例
- 首先还是应该科普下函数参数传递机制,传值和传引用是什么意思?函数参数传递机制问题在本质上是调用函数(过程)和被调用函数(过程)在调用发生时进
- Flask Script扩展提供向Flask插入外部脚本的功能,包括运行一个开发用的服务器,一个定制的Python shell,设置数据库的
- Swagger是一个API开发者的工具框架,用于生成、描述、调用和可视化RESTful风格的Web服务。总体目标是使客户端和文件系统服务器以
- 近年,不论是正在快速增长的直播,远程教育以及IM聊天场景,还是在常规企业级系统中用到的系统提醒,对websocket的需求越来越大,对web
- 该代码用的是paramiko模块,python版本是python2.7下面上源码# -*- coding: utf-8 -*-import
- 一、同一台电脑需要安装2个MYSQL,以mysql-5.7.39和mysql-8.0.30为例; 1.下载:https://dow
- 最近我在用梯度下降算法绘制神经网络的数据时,遇到了一些算法性能的问题。梯度下降算法的代码如下(伪代码):def gradient_desce
- 你用过css么?当然,我是指你喜欢做网页的话,用过?很好,那你用过它的特效么?没有?那请跟我来。让我先通俗的介绍一下css,cs
- 实现二维平面上散点的绘制,并可以给每个散点标记序号或者名称:import numpy as npimport matplotlib.pypl
- 作者认为最快的学习区块链的方式是自己创建一个,本文就跟随作者用Python来创建一个区块链。对数字货币的崛起感到新奇的我们,并且想知道其背后
- 本例已经实现的数据库password,数据库的表以及表结构如下:表中已经插入的信息:实现思路无非是用户完成账户密码输入并点击登录
- 如下所示:sum = 0n = 99while n > 0: sum = sum + n n = n - 2p
- 用过软件的朋友都知道,进度条是一个优秀软件的重要组成部分。它的存在能够使用户及时掌握程序的运行进度,确认应用程序正常工作。可是ASP中似乎没
- 注:本文是应Alan邀请为《CSS布局实录》写的一个web标准入门指导。书已经上市近一年了,现在摘选出来,给初学者一个参考。希望了解更多实现
- 关于这篇文章有几句话想说,首先给大家道歉,之前学的时候真的觉得下述的是比较厉害的东西,但是后来发现真的是基础中的基础,内容还不是很完全。再看
- asp压缩access数据库,具体asp代码见下: Class DatabaseTools &n
- 1.随机翻转(水平和垂直)torchvision.transforms.RandomVerticalFlip函数和torchvision.t