Vert.x + Protobuf二进制协议解析

这一期介绍如何解析二进制私有协议。

先说几句题外话,就是绝大多数情况下,可能根本用不着使用私有二进制协议,除非你的业务对性能极其敏感,否则HTTP足矣。

协议

我们的协议非常简单,先是一个4字节的整数表示数据长度,然后紧接着就是protobuf序列化后的字节数组。proto定义如下:

syntax = "proto2";
package cn.fh.vertx.demo.proto;

option java_multiple_files = true;

message Message {
    required int32 type = 1;
    required string content = 2;
}

在实际业务场景中,消息格式的定义是有讲究的,这里不深究,重点在于vert.x解析。

RecordParser

Vert.x中只有一个二进制协议解析辅助类,即RecordParser,可以很好的解决粘包/拆包问题。它有两种工作模式,一是delimited mode, 即通过固定分隔符分隔数据包,这种用的其实比较少;二是fixed size mode,即固定数据长度模式。诈一看可能很多人会有疑虑,多数协议都是可变长度的啊,这两种模式看起来都不能满足要求。其实模式二支持在处理的过程中随时动态的改变size值,这样就可以间接完成对变长协议的解析。

首先,RecordParser有两类Builder方法,RecordParser.newDelimited()RecordParser.newFixed(4),分别对应上面的模式一和模式二。构造完成以后,跟其它vert.x的方法一样,我们需要定义一个Handler, RecordParser每分割出一段字节数组都会调用一次Handler,业务逻辑处理就在此Handler中进行。对于上面"长度 + 数据"的协议,我们可以首先构造一个以4字节为单位的RecordParser,在收到长度数据后,再动态将解析器修改成指定长度状态,从而完成对数据部分的分割。代码示例如下:

// 先以长度4构造对象
        RecordParser parser = RecordParser.newFixed(4);

        // 设置处理器
        parser.setOutput(new Handler<Buffer>() {
            // 表示当前数据长度
            int size = -1;

            @Override
            public void handle(Buffer buffer) {
                // -1表示当前还没有长度信息,需要从收到的数据中取出长度
                if (-1 == size) {
                    // 取出长度
                    size = buffer.getInt(0);
                    // 动态修改长度
                    parser.fixedSizeMode(size);

                } else {
                    // 如果size != -1, 说明已经接受到长度信息了,接下来的数据就是protobuf可识别的字节数组
                    byte[] buf = buffer.getBytes();
                    Message msg = null;
                    try {
                        msg = Message.parseFrom(buf);
                    } catch (InvalidProtocolBufferException e) {
                        System.out.println(e.getMessage());
                        socket.close();
                        return;
                    }
                    
                    System.out.println(msg);
                    // 处理完后要将长度改回4
                    parser.fixedSizeMode(4);
                    // 重置size变量
                    size = -1;
                }
            }
        });

虽然用起来比较别扭,但这的确是标准使用方法。

如果表示长度的数据不是从0开始的,比如0 ~ 3为消息类型,4 ~ 7才表示body的长度,那么可以在构造解析器时先将长度设为8, 然后取后4字节做为新的长度即可。

其实这个类所做的事正是Netty里LengthFieldBasedFrameDecoder的功能,希望Vert.x以后能直接提供好类似的处理器,不要让用户再手动取长度了,毕竟低层的Netty都支持,你更高级的封装怎么可以没有呢。

这个RecordParser怎么用呢?看一下完整的代码吧:

        Vertx vertx = Vertx.vertx();
        
        // 创建TCP Server
        NetServer server = vertx.createNetServer();
        // 设置Handler
        server.connectHandler(socket -> {
            // 构造parser
            RecordParser parser = RecordParser.newFixed(4);
            parser.setOutput(new Handler<Buffer>() {
                int size = -1;

                @Override
                public void handle(Buffer buffer) {
                    if (-1 == size) {
                        size = buffer.getInt(0);
                        parser.fixedSizeMode(size);
                        
                    } else {
                        byte[] buf = buffer.getBytes();
                        Message msg = null;
                        try {
                            msg = Message.parseFrom(buf);
                        } catch (InvalidProtocolBufferException e) {
                            System.out.println(e.getMessage());
                            socket.close();
                            return;
                        }
                        parser.fixedSizeMode(4);
                        size = -1;
                    }
                }
            });

            socket.handler(parser);
        });
        
        // 监听
        server.listen(8008, "localhost", res -> {
            if (res.succeeded()) {
                System.out.println("tcp server is listening at 8008");

            } else {
                System.out.println(res.cause());
            }
        });

写起来会有一点点奇怪,适应就好。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 194,524评论 5 460
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 81,869评论 2 371
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 141,813评论 0 320
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,210评论 1 263
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,085评论 4 355
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,117评论 1 272
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,533评论 3 381
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,219评论 0 253
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,487评论 1 290
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,582评论 2 309
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,362评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,218评论 3 312
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,589评论 3 299
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,899评论 0 17
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,176评论 1 250
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,503评论 2 341
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,707评论 2 335

推荐阅读更多精彩内容