0

我必须测试基于 Jetty 的服务器。该服务器可以使用自己的协议、HTTP、HTTPS，最后它开始支持 SPDY。我有一些基于httplib/的压力测试http.client——每个线程都以相似的 URL 开头（查询字符串中的一些数据是可变的），将执行时间添加到全局变量，并且每隔几秒钟显示一些统计信息。代码如下：

t_start = time.time()
connection.request("GET", path)
resp = connection.getresponse()
t_stop = time.time()
check_response(resp)
QRY_TIMES.append(t_stop - t_start)

使用本机协议的客户端共享httplibAPI，因此connection可能是本机的，HTTPConnection或者HTTPSConnection.

现在我想使用spdylay模块添加 SPDY 测试。但是它的界面是不透明的，我不知道如何将它的不透明变成类似于httplib界面的东西。我已经根据示例制作了测试客户端，但是虽然第二个参数spdylay.urlfetch()是类名而不是对象，但我不知道如何在我的测试中使用它。我已经on_close()在我的类的方法中添加了测试 extends spdylay.BaseSPDYStreamHandler，但它与其他测试不兼容。如果是实例，我会在spdylay.urlfetch()通话之外使用它。

如何spydlay在基于httplib接口的代码中使用？

4

1 回答 1

0

我唯一的想法是使用全局字典，其中url是键，handler对象是值。这并不理想，因为：

  • 具有相同的新查询url将覆盖以前的响应
  • 很容易忘记handler从全局字典中解脱出来

但它有效！

import sys
import spdylay

CLIENT_RESULTS = {}


class MyStreamHandler(spdylay.BaseSPDYStreamHandler):
    def __init__(self, url, fetcher):
        super().__init__(url, fetcher)
        self.headers = []
        self.whole_data = []

    def on_header(self, nv):
        self.headers.append(nv)

    def on_data(self, data):
        self.whole_data.append(data)

    def get_response(self, charset='UTF8'):
        return (b''.join(self.whole_data)).decode(charset)

    def on_close(self, status_code):
        CLIENT_RESULTS[self.url] = self


def spdy_simply_get(url):
    spdylay.urlfetch(url, MyStreamHandler)
    data_handler = CLIENT_RESULTS[url]
    result = data_handler.get_response()
    del CLIENT_RESULTS[url]
    return result


if __name__ == '__main__':
    if '--test' in sys.argv:
        spdy_response = spdy_simply_get('https://localhost:8443/test_spdy/get_ver_xml.hdb')

我希望有人可以spdy_simply_get(url)做得更好。

于 2014-01-24T08:04:19.290 回答