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

安装

首先，你需要安装 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

简单的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相关文件.如果要使用，必须自己手动指定tagsserver+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)

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