下单延迟10s撤单性能测试

FunTester

发布于 2021-12-02 12:22:46

3910

发布于 2021-12-02 12:22:46

文章被收录于专栏：FunTesterFunTester

我曾经天真以为，性能测试场景我已经无敌了，可现实又轻轻拍了我一下。

研发提出了一个下单之后延迟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方法中从延迟队列中获取到期的订单号，然后进行撤单操作。

下面是delayQueue功能演示视频

实现

订单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()
    }