折腾一下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))
