Websocket for Tsung

 

这篇博文距离上次提到要写差不多快两个月了，一方面时间不多，另一方面主观上总觉得写博客总要写的很好才能发表出来。不过，看了很多别人的博客，其实很多时候只是个记录笔记。希望自己也能利用这个平台做好自己的技术积累过程，不要太在意到底写的怎么样。

 

         费话少说，开发Websocket for Tsung的初衷是为了做测试。公司一个基于websocket和XMPP的长连接解决方案，需要做压力测试，而Tsung本身并没有对websocket的支持。要让Tsung支持websocket，其实也有两个思路，一个是将websocket作为session type（Tsung本身支持的类型有：HTTP，WebDAV，XMPP等），一个是将websocket作为server type（Tsung本身支持的类型有：tcp，udp，ssl）。

我一开始的思路是将websocket作为session type，再将我们的XMPP协议及相关逻辑进行封装，好处是可以利用Tsung的插件机制方便的扩展，坏处是不能利用到Tsung本身支持XMPP的特性。所以后来考虑将websocket作为一种server type，但是扩展websocket作为一种server type的话，就需要看Tsung的源码。

这里记录下两种思路的实现。

 

websocket作为session type

找到的唯一的有关Tsung协议插件机制实现的文档见下面这个地址：Writing a Tsung plugin，不过由于这篇文档写的比较早，跟当前的Tsung实现有一些差别。上文中Step 4要完成的Config reader机制有变化。ts_config.erl并不会调用ts_config_websocket.erl中的parse_config方法，而是调用ts_websocket.erl中的parse_config函数，这个函数再通过ts_config_websocket.erl的parse_config来完成相应的功能。具体可参见tsung本身的实现：ts_http.erl，ts_plugin.erl。

 

websocket作为server type

         查看源码后发现，跟底层协议相关处理主要集中在ts_client.erl中。在hand_info函数中，会根据底层协议的类型（tcp，udp，ssl等）进行相应的处理，而我们的websocket协议是基于tcp的，所以我们修改就只针对tcp事件的处理。

         其实思路也很简单，websocket协议作为server type跟tcp不同的是，当上层协议（http或者xmpp等）需要底层协议连接目标服务器时，tcp所要做的就是一个connect调用，而websocket就需要：1）连接目的服务器；2）跟目标服务器完成websocket协议的握手过程。相应的，在发送消息时websocket需要对上层协议的原始消息按websocket协议定义的数据格式进行封装后，再发送出去；在收到消息时，也需要对收到的消息先进行解码，再将数据反馈到上层协议。

 

上面两个方面的实现，代码见：https://github.com/onlychoice/tsung/tree/websocket

 

当前正在分析Tsung的架构和实现，后续的博文会对这方面进行个总结，希望不会拖的太久。