折腾一下WebSocket的ArrayBuffer传输方式

前言

之前写WebSocket都是基于文本传输的，后来准备升级项目，于是打算尝试一下arraybuffer传输方式，由于是第一次使用javascript处理字符串转arraybuffer，不过真的是一把辛酸泪啊，特此记录。

项目背景

还是之前写的(基于Tio通讯框架的SpringBootLayIM项目)[https://github.com/fanpan26/SpringBootLayIM].

开发过程还原

1.要将客户端的传输方式改为arraybuffer。

     var ws = new WebSocket(tool.options.server + '?access_token=' + token);

     ws.binaryType = 'arraybuffer';

2.我们就要用到框架中的 IWsMsgHandler.onBytes(WsRequest wsRequest, byte[] bytes, ChannelContext channelContext)方法接收客户端的消息

    /**

     * 字节传输

     * */

    public Object onBytes(WsRequest wsRequest, byte[] bytes, ChannelContext channelContext) throws Exception {

        System.out.println("接收到的消息为："+new String(bytes));

        return null;

    }

3.在ws.send之前要先将发送的内容转换。

  //

  var buff = new TextEncoder().encode(str);

很好，这三个步骤做完之后，我们试一下。可以看到，打印结果正常

接收到的消息为：{"username":"雍正王","avatar":"/static/images/demo/huangshang.jpg","id":203328,"type":"friend","content":"22"}

到此为止呢，还是没有任何问题的。按照我之前的思路就是将消息反序列化成对象实例，然后进行相应的逻辑操作。不过呢，后来突发奇想，本来我传递过来的消息就是想原封不动的发送给对方，也就是说程序不需要执行反序列化的过程。那可不可以这样，如下图：

没错就是这样子，然后我打印了一下new TextEncoder().encode(str);返回的内容如下：

可以看到，它是一个 Uint8Array,于是我就看看能否重新构造一个数组，用伪代码实现就是酱紫：

var newBuff = new ArrayBuffer(4+1+bodyArray.buffer.byteLength)

再查资料发现ArrayBuffer得依赖DataView来进行赋值操作，那不就简单啦。先把接收人ID（UInt32）和消息类型（UInt8) 写入，如下：

  var view = new DataView(dataBuff.buffer);

  view.setInt32(0, targetId);

  view.setInt8(4, 1);

那剩下的body怎么写进去呢？按照我想的，应该是有个 setContent(arr,offset:number)方法，可是查了一下，没有。于是乎再查文档，原来Unit8Array有个遍历的方法，那就用这个方法试试吧。

    //从索引第五个开始写

    dataBuff.forEach(function (value, index) {

             view.setInt8(5+index,value);

      });

为了验证我这个思路（呜呜，这个思路还是研究了好久，浪费了很多时间。。。其实中间由于不熟悉DataView和ArrayBuffer导致做了很多尝试的工作，而且都失败了，要么就是报错，要么就是覆盖了消息体）的正确性，我们将后台代码改一下：

       byte[] targetIdBytes = Arrays.copyOf(bytes,4);

       byte[] contents = Arrays.copyOfRange(bytes,5,bytes.length);

       System.out.println("消息体："+new String(contents));

       int targetId = ConvertUtil.byteArrayToInt(targetIdBytes);

       System.out.println("接收人："+targetId);

       System.out.println("消息类型"+bytes[4]);

运行正常。

可是，总觉得在遍历复制一遍有点繁琐。那既然setInt32,setUInt8这些方法是覆盖数组里的值的，那我可不可以这样写呢？使用占位符的方式，也就是说，在不影响消息体的情况下，在转化成byte数组之后，前五位是占位数据，后边才是正确的消息，那么我重写前五位的内容也不会受到什么影响了，而且不用遍历赋值了。说干就干，改成代码如下：

   var str = placeholder + JSON.stringify(d);

   var buff = new TextEncoder().encode(str);

   return buff;

问题就在于这个placeholder的值是什么呢？其实我们使用小写字母代替就可以了。比如 'abcde',看一下转换的结果：

不出我所料，那这样的话，就没问题啦，直接覆盖前五位就可以了。客户端完整代码如下：

  //根据layim提供的data数据，进行解析

            var mine = data.mine,

                to = data.to,

                id = mine.id,

                group = to.type === 'group';

            if (group) {

                id = to.id;

            }

            //构造消息

            var msg = {

                username: mine.username

                , avatar: mine.avatar

                , id: id

                , type: to.type

                , content: mine.content

            }, targetId = to.id

            var dataBuff = this.encode(msg);

            var view1 = new DataView(dataBuff.buffer);

            view1.setInt32(0, targetId);

            view1.setInt8(4, group ? msgType.chatGroup : msgType.chatFriend);

            return view1.buffer;

运行一下，结果没问题，大功告成！

总结

从问题的发出到解决，虽然从博客上来看没有什么难度，但是自己在做的时候，搜了很多资料，尝试了很多次也没有结果，原因是自己自动脑补了一些对象的操作API，基础知识还是很重要的啊，滚回去学习。虽然最终的思路不一定是最好的，或者还有一些其他的问题，但是多尝试一下还是有很多意外的收获。那么问题来了。反序列化和数组的拷贝哪个效率更高一些呢？

  //这个好还是

  byte[] targetIdBytes = Arrays.copyOf(bytes,4);

  byte[] contents = Arrays.copyOfRange(bytes,5,bytes.length);

  //这个好呢？

  Json.toBean(new String(contents))