记一次rpcx拦截器的实现+使用TLS证书启动

吃馍夹菜 · · 650 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

安装

首先,你需要安装 rpcx:

go get -u -v github.com/smallnest/rpcx/...

这一步只会安装 rpcx 的基础功能。如果你想要使用 etcd 作为注册中心,你需要加上etcd这个标签。 (see build tags)

go get -u -v -tags "etcd" github.com/smallnest/rpcx/...

如果你想要使用 quic ,你也需要加上quic这个标签。

go get -u -v -tags "quic etcd" github.com/smallnest/rpcx/...

方便起见,我推荐你安装所有的tags,即使你现在并不需要他们:

go get -u -v -tags "reuseport quic kcp zookeeper etcd consul ping" github.com/smallnest/rpcx/...

tag:

  • quic: 支持 quic 协议
  • kcp: 支持 kcp 协议
  • zookeeper: 支持 zookeeper 注册中心
  • etcd: 支持 etcd 注册中心
  • consul: 支持 consul 注册中心
  • ping: 支持 网络质量负载均衡
  • reuseport: 支持 reuseport

上面内容摘自:https://doc.rpcx.io/part1/qui...

简单的Service实现

实现一个 Sevice 就像写一个单纯的 Go 结构体:

import "context"

type Args struct {
    A int
    B int
}

type Reply struct {
    C int
}

type Arith int

func (t *Arith) Mul(ctx context.Context, args *Args, reply *Reply) error {
    reply.C = args.A * args.B
    return nil
}

注册一个简单服务

s := server.NewServer()
s.RegisterName("Arith", new(Arith), "")
s.Serve("tcp", ":8972") // 可以启动quic(udp)/tcp/http服务

手写一个拦截器

其实是实现plugin的PostReadRequest方法

type SignPlugin struct {
}

// 拦截器
func (signPlugin *SignPlugin) PostReadRequest(ctx context.Context, req *protocol.Message, err error) error {
    //fmt.Println(req.ServiceMethod)
    //fmt.Println(req.ServicePath)
    // 我这是使用json传输的
    p := &你的结构体
    err = json.Unmarshal(req.Payload, p)

    // todo 你的逻辑(数据验证,数据处理等)
    decodeMap := 你处理后的数据(如果不用处理,该步骤可以省略)
    req.Payload = decodeMap
    return err
}

把拦截器add到启动服务中

s := server.NewServer()
siginPlugin := &controller.SignPlugin{}
s.Plugins.Add(siginPlugin) // 把拦截器注册到服务中
s.RegisterName("Arith", new(Arith), "")
s.Serve("tcp", ":8972") // 可以启动quic(udp)/tcp/http服务

到此拦截器已经实现.
下面附上client代码

    option := client.DefaultOption
    option.SerializeType = protocol.JSON // 传输数据格式
    // #1
    d := client.NewPeer2PeerDiscovery("tcp@127.0.0.1:8972", "")
    // #2
    xclient := client.NewXClient("Arith", client.Failtry, client.RandomSelect, d,option)
    defer xclient.Close()

    // #3
    args := &example.Args{
        A: 10,
        B: 20,
    }

    // #4
    reply := &example.Reply{}

    // #5
    err := xclient.Call(context.Background(), "Mul", args, reply)
    if err != nil {
        log.Fatalf("failed to call: %v", err)
    }

    log.Printf("%d * %d = %d", args.A, args.B, reply.C)

使用quic(UDP)和TLS证书注意:

  • 使用quic(UDP)时,s.Serve("tcp", "localhost:8972"). address参数必须要写完整,localhost不能省略
  • 使用quic(UDP)时,rpcx对quic协议的支持是通过build tags实现的.默认不会编译quic相关文件.如果要使用,必须自己手动指定tags

    • server+client都需要
    go run -tags quic main.go
  • 使用quic(UDP)时,必须要使用TLS证书

    • server代码
    // 加载tls证书,使用quic必要步骤
    cert, err := tls.LoadX509KeyPair("公钥路径", "私钥路径")
    if err != nil {
        panic("加载证书失败:" + err.Error())
    }
    configs := &tls.Config{Certificates: []tls.Certificate{cert}}
    s = server.NewServer(server.WithTLSConfig(configs))
    • client代码
    conf := &tls.Config{
        InsecureSkipVerify: true, // true=跳过tls证书验证
    }
    
    option := client.DefaultOption
    option.SerializeType = protocol.JSON // 传输数据格式
    option.TLSConfig = conf // 使用tls证书
    d := client.NewPeer2PeerDiscovery("tcp@127.0.0.1:8972", "")
      xclient := client.NewXClient("Arith", client.Failtry, client.RandomSelect, d,option)

有疑问加站长微信联系(非本文作者)

本文来自:Segmentfault

感谢作者:吃馍夹菜

查看原文:记一次rpcx拦截器的实现+使用TLS证书启动

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

650 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传