我曾经天真以为，性能测试场景我已经无敌了，可现实又轻轻拍了我一下。
研发提出了一个下单之后延迟 10s 撤单的压测需求，着实让我迷糊了一会儿，最后参考了 Java 的延迟队列java.util.concurrent.DelayQueue实现了这个需求。
需求
在某个场景中，我之前的设计是下单，成功的话撤单，原因是为了避免测试用户资产被耗尽。
最近研发对服务功能逻辑进行了优化，在回归测试中，发现这个用例测试结果与之前差异较大。研发判断可能是下单、撤单间隔时间较短导致问题。所以需要进行延迟撤单的压测。
思路
这里使用了java.util.concurrent.DelayQueue这个线程安全队列类，看名字其实也能明白用途。还有一个方案java.util.Timer，但是会创建额外的线程，而且难以控制撤单的速率，所以这次就放弃了。
首先每次下单时候，如果成功会把订单号通过一个java.util.concurrent.Delayed实现类记录到java.util.concurrent.DelayQueue队列中，然后在com.oktest.okex.algo.trade.OrderAndCancelGrid3.FunTester#doing方法中从延迟队列中获取到期的订单号，然后进行撤单操作。
实现
订单 ID 是java.lang.String类型，java.util.concurrent.Delayed实现类如下：
private static class FunDelay implements Delayed {
int time
String id
FunDelay(String id) { this.time = getMark() + delay this.id = id }
@Override public long getDelay(TimeUnit unit) { return time - getMark(); }
@Override public int compareTo(Delayed o) { FunDelay item = (FunDelay) o; if (this.time < item.time) { return -1; } else { return 1; } } }
复制代码
测试用例
大部分内容复用了之前的设计，重写了after方法，处理用户结束测试之后，延迟队列中尚未处理的订单 ID。
private static class FunTester extends FixedThread<Trade> {
DelayQueue<FunDelay> ids = new DelayQueue<>()
FunTester(Trade o, int limit) { super(o, limit, true) }
@Override protected void doing() throws Exception { def res = f.order(********) if (0 == res.getInteger("code")) { def array = res.getJSONArray("data").get(0) def id = array.getString("Id") ids.add(new FunDelay(id)) } else { output(res.toString()) } int i = 0 while (true) { if (i++ > 10) break def poll = ids.poll() if (poll == null) break f.stop("***", poll.id) } }
@Override protected void after() { super.after() sleep(1.0) f.clearGrid() }
@Override ThreadBase clone() { return new FunTester(f, limit) }
}
复制代码
执行
执行方法依旧复用之前的内容，添加第 4 个参数，来参数化延迟的时间。
static int delay;
public static void main(String[] args) { ClientManage.init(10, 5, 0, "", 0) def util = new ArgsUtil(args) def thread = util.getIntOrdefault(0, 1) def times = util.getIntOrdefault(1, 1000) def runuptime = util.getIntOrdefault(2, 0) delay = util.getIntOrdefault(3, 3) Constant.RUNUP_TIME = runuptime Common.PRINT_RESPONSE = false Common.VERIFY = false FunLibrary.LOG_KEY = false Common.MOCO = true Common.setHost() def line = DataUtils.getMocoTokens() def tasks = [] thread.times { def base = getBase(line.get(getRandomInt(500))) def drive = new Trade(base) tasks << new FunTester(drive, times) }
new Concurrent(tasks, "下单和延迟10s撤单").start()
over() }
复制代码
测试结果
我使用了默认延迟 3s 的参数。下面是执行过程中的日志截图：
图中可以看到，17s 开始下单，20s 开始撤单。需求完美实现！
欢迎关注 FunTester，Have Fun ~ Tester ！
评论