FunTester 利用闭包实现自定义等待方法

FunTester · 2022年01月06日 · 890 次阅读

我在最近学习多线程编程的过程中,时长需要测试各类多线程工具类或者框架在各种线程数下的性能。基本的思路就是准备数据,编写用例方法,执行用例方法,最后等待结束,统计计算测试数据。

其中倒数第二个步骤:等待结束,之前我在Java 自定义异步功能实践文章中用到看java.util.concurrent.Phaser这个多线程同步类完成。但是在最近的异步编程实践中,我用到了更多闭包的语法,例如:

public static void main(String[] args) {
    //自定义异步任务
    def funtester = {
        fun {
            //syncDoingsomething()
        }
    }
    //单开10个线程执行异步任务
    10.times {funtester()}
}

通过闭包的语法将异步任务的内容封装成另外一个方法,然后通过多次执行异步方法来完成多线程任务,这样写比较简单,而且非常具有可读性,如果使用java.util.concurrent.Phaser作为第二个参数,显得有些凌乱,不够优雅,比如:

public static void main(String[] args) {
    def phaser = new Phaser(10)
    //自定义异步任务
    def funtester = {
        fun({
            //syncDoingsomething()
        }, phaser)
    }
    //单开10个线程执行异步任务
    10.times {funtester()}
}

所以我就有了一个单独实现自定义等待方法的需求,就是设置一个或者多个判定条件,一直等待这个判定条件和条件组合符合预期,才释放当前线程,继续运行下去。经过了之前的异步和闭包的学习,我很快找到了一个简单的解决方案。如下:

/**
 * 自定义等待,间隔0.5s
 *
 * @param f
 */
public static void waitFor(Supplier<Boolean> f) {
    while (!f.get()) {
        sleep(0.5);
    }
}

使用方法介绍:场景,我们需要 10 个线程分别计算某一类数据结果,然后将数据结果存入一个列表中(需线程安全,不然会偶发数量对不上),按照上面的思路,就是执行完 10 个异步任务,然后等待所有线程执行完。我使用了列表长度作为等待对象,代码如下:

waitFor({ts.size() > 10})
//下面也行
waitFor {ts.size() > 10}

最近会对 Java 各种队列进行压测,会用到这个功能,欢迎持续关注。

PS:本文使用 Groovy,若无特殊需求,我一般使用 Groovy 作为脚本语言。

Have Fun ~ Tester !

如果觉得我的文章对您有用,请随意打赏。您的支持将鼓励我继续创作!
暂无回复。
需要 登录 后方可回复, 如果你还没有账号请点击这里 注册