JAVA RPC (六) 之thrift反序列化RPC消息体

java

我们来看一下服务端的简单实现,直接上thrift代码,很直观的来看一看thrift的server到底干了些什么

 1  public boolean process(TProtocol in, TProtocol out) throws TException {

2 TMessage msg = in.readMessageBegin();

3 ProcessFunction fn = (ProcessFunction)this.processMap.get(msg.name);

4 if (fn == null) {

5 TProtocolUtil.skip(in, (byte)12);

6 in.readMessageEnd();

7 TApplicationException x = new TApplicationException(1, "Invalid method name: '" + msg.name + "'");

8 out.writeMessageBegin(new TMessage(msg.name, (byte)3, msg.seqid));

9 x.write(out);

10 out.writeMessageEnd();

11 out.getTransport().flush();

12 return true;

13 } else {

14 fn.process(msg.seqid, in, out, this.iface);

15 return true;

16 }

17 }

TMessage msg = in.readMessageBegin();这段代码的意思是从client端获取到二进制数据时候,获取Tmessage实例,和client的wirteMessage方法进行对应,聪明的小伙伴这个时候一定会想到这个是怎么解析的
为什么没有解析消息体长度和消息体内容的部分代码,不要着急我们一步一步渗透,readMessageBegin核心代码如下

 1  public TMessage readMessageBegin() throws TException {

2 int size = this.readI32();

3 if (size < 0) {

4 int version = size & -65536;

5 if (version != -2147418112) {

6 throw new TProtocolException(4, "Bad version in readMessageBegin");

7 } else {

8 return new TMessage(this.readString(), (byte)(size & 255), this.readI32());

9 }

10 } else if (this.strictRead_) {

11 throw new TProtocolException(4, "Missing version in readMessageBegin, old client?");

12 } else {

13 return new TMessage(this.readStringBody(size), this.readByte(), this.readI32());

14 }

15 }

int size = this.readI32();是做了些什么,接着往下看

 1 public int readI32() throws TException {

2 byte[] buf = this.i32rd;

3 int off = 0;

4 if (this.trans_.getBytesRemainingInBuffer() >= 4) {

5 buf = this.trans_.getBuffer();

6 off = this.trans_.getBufferPosition();

7 this.trans_.consumeBuffer(4);

8 } else {

9 this.readAll(this.i32rd, 0, 4);

10 }

11

12 return (buf[off] & 255) << 24 | (buf[off + 1] & 255) << 16 | (buf[off + 2] & 255) << 8 | buf[off + 3] & 255;

13 }

这个地方代码就很清晰了,原来是从trans里面或取得字节流数据。

1 private int readAll(byte[] buf, int off, int len) throws TException {

2 this.checkReadLength(len);

3 return this.trans_.readAll(buf, off, len);

4 }

继续能翻到最终的代码解析部分

private void readFrame() throws TTransportException {

this.transport_.readAll(this.i32buf, 0, 4);

int size = decodeFrameSize(this.i32buf);

if (size < 0) {

throw new TTransportException("Read a negative frame size (" + size + ")!");

} else if (size > this.maxLength_) {

throw new TTransportException("Frame size (" + size + ") larger than max length (" + this.maxLength_ + ")!");

} else {

byte[] buff = new byte[size];

this.transport_.readAll(buff, 0, size);

this.readBuffer_.reset(buff);

}

}

这里就很明显了,首先read一个四个字节的长度,这个int类型代表着后面要跟着接收的消息体的长度,来源是client的发送内容,然后在根据消息体长度在读出来所有的内容,然后缓存到readBuffer_中,然后再读取内容的时候直接从readBuffer_中获取字节流就可以了。

好了,服务端接收二进制数据解析的关键点就在这里了,接下来开始,我将用大量的篇幅开始讲解我的企业级RPC框架,满满的干货,感兴趣的小伙伴去我码云给个star吧!!!

https://gitee.com/a1234567891/koalas-rpc

koalas-RPC 个人作品,提供大家交流学习,有意见请私信,欢迎拍砖。客户端采用thrift协议,服务端支持netty和thrift的TThreadedSelectorServer半同步半异步线程模型,支持动态扩容,服务上下线,权重动态,可用性配置,页面流量统计等,持续为个人以及中小型公司提供可靠的RPC框架技术方案

更多学习内容请加高级java QQ群:825199617

 













以上是 JAVA RPC (六) 之thrift反序列化RPC消息体 的全部内容, 来源链接: utcz.com/z/392320.html

回到顶部