这篇博文距离上次提到要写差不多快两个月了,一方面时间不多,另一方面主观上总觉得写博客总要写的很好才能发表出来。不过,看了很多别人的博客,其实很多时候只是个记录笔记。希望自己也能利用这个平台做好自己的技术积累过程,不要太在意到底写的怎么样。
费话少说,开发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 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
转自:http://jzhihui.iteye.com/blog/1402968