最近新申请了一些数据库资源,代码也已经改动完毕,测了几个数据,感觉改动没有什么问题。不过,切换数据源如果只是简单测几条数据,貌似很不严谨的样子
,毕竟线上的流量比较大,说不准就被压垮了呢。所以,还是有必要做一下压力测试。拿线上的流量来做测试是最好的了,一方面可以看改动后的程序和数据库能否支撑目前的请求,另一方面,也可以和当前的性能进行比较。因为时间仓促,没有去调研现有的工具,而是自己写了一个。我的思路是这样的:线上的请求过来都会打日志(完整的日志),一个脚本一直在扫描这个日志,当出现了请求之后,就发往改动好的程序代码,这样就可以模拟两套数据源和代码在同一时刻的响应时间了(可能中间还是会差个几秒,但是可以忽略不计了)。
#coding:utf-8
#!/usr/bin/python
import json
import redis
import subprocess
def getData():
popen = subprocess.Popen('tail -n 1 -f restlog.log',stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
while True:
line = popen.stdout.readline()
if "request" not in line: #如果不是请求,直接跳出
continue
data = line.split("request body:")
#下面一段封装具体的请求,因为涉及到公司的一些代码,暂且省略了
代码运行了三四个小时,做了充分的数据对比,还不错。