如何在CompletableFuture.runAsync()执行前统一调用前置方法并隐藏实现

1. 封装自定义工具类 #

搞个工具类，比如叫AsyncTaskWrapper，里面写自己的runAsync方法，内部自动把你要前置执行的逻辑和开发者传入的任务绑在一起。这样其他人用这个工具类的方法就行，根本不用管前置那堆代码。

代码示例：

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.Executor;
import java.util.concurrent.Executors;
import java.util.function.Runnable;

public class AsyncTaskWrapper {
    // 和原生方法对齐，默认用公共线程池
    public static CompletableFuture<Void> runAsync(Runnable task) {
        return CompletableFuture.runAsync(wrapTask(task));
    }

    // 支持自定义线程池的版本
    public static CompletableFuture<Void> runAsync(Runnable task, Executor executor) {
        return CompletableFuture.runAsync(wrapTask(task), executor);
    }

    // 核心逻辑：把前置方法和原任务打包
    private static Runnable wrapTask(Runnable originalTask) {
        return () -> {
            // 这里就是你要在异步任务前跑的方法，对外完全隐藏
            methodToRunBeforeAsync();
            originalTask.run();
        };
    }

    // 藏起来的前置方法，外部碰不到
    private static void methodToRunBeforeAsync() {
        // 这里写你的前置逻辑，比如初始化线程上下文、打个埋点日志啥的
        System.out.println("前置方法执行，当前线程：" + Thread.currentThread().getName());
    }
}

2. 替换原有调用方式 #

让其他开发者直接用AsyncTaskWrapper.runAsync()代替原生的CompletableFuture.runAsync()就行，他们完全不用知道有个methodToRunBeforeAsync()存在。

调用示例：

public class MyClass {
    public static void myMethod() {
        System.out.println("业务方法执行，当前线程：" + Thread.currentThread().getName());
    }

    public static void main(String[] args) {
        // 用自定义工具类替代原生方法
        AsyncTaskWrapper.runAsync(MyClass::myMethod);
        // 需要自定义线程池的话也支持
        AsyncTaskWrapper.runAsync(MyClass::myMethod, Executors.newFixedThreadPool(5));
    }
}

3. 额外提醒 #

• 如果项目里已经到处用了原生的CompletableFuture.runAsync()，直接用IDE的批量替换功能，把所有CompletableFuture.runAsync换成AsyncTaskWrapper.runAsync就行，改动成本极低。
• 前置方法是工具类的私有方法，外部根本调用不了，实现细节完全藏起来了。
• 这种方式完全兼容原生CompletableFuture的所有特性，后续链式调用thenRun()、exceptionally()之类的都不受影响。

内容的提问来源于stack exchange，提问作者Rabid