java - 如何在 Java 中使用 AspectJ 记录父级和子级的线程 ID

Question

我是 AspectJ 的新手。能够创建一个简单的 JUnit 和 Aspect 类来记录 ThreadId，从而记录父线程 Id。但我不知道如何记录 Child ThreadId。

鉴于以下代码片段，我想使用 AspectJ 记录父级和子级的线程 ID。

JUnit:

@Test
public void testExecutorService() {
    ExecutorService service = Executors.newSingleThreadExecutor();
    Runnable task = new Runnable() {
        @Override
        public void run() {
            System.out.println("working on the task");
        }
    };
    service.submit(task);
}

方面:以下方面记录父线程Id。

before() :
    call(* ExecutorService+.submit(..))
 {
    System.out.println("Parent Thread Id: "+Thread.currentThread().getId());
    //System.out.println("Child Thread Id: "+??); //?? - how to capture child thread id?
 }

我知道它在这里使用“之前”建议，并且它正在拦截提交方法，这也可能是一个问题。 如何使用正确的切入点表达式记录子线程 ID 和父线程 ID？

Answer 1

您可以做的是拦截提交的任务参数并增强该任务以打印您想要的内容:

Object around(Runnable task) : call(* ExecutorService+.submit(Runnable, ..)) && args( task)
{
    final long parentID = Thread.currentThread().getId();
    Runnable newTask =  () -> {
            System.out.println("Parent Thread Id: "+ parentID);
            System.out.println("Child Thread Id: "+Thread.currentThread().getId()); //?? - how to capture child thread id?
            task.run();
    };
    return proceed(newTask);
}

一旦执行异步任务，就会打印执行该任务的线程的 id。

运行示例:

Main.java:

public class Main {
    public static void main(String[] args) {
        new Test().testExecutorService();
    }
}

测试.java:

public class Test {
    public void testExecutorService() {
        ExecutorService service = Executors.newSingleThreadExecutor();
        service.submit(() ->  System.out.println("working on the task"));
        try {
            service.awaitTermination(2, TimeUnit.SECONDS);
            service.shutdown();
        }catch (InterruptedException ie) {
            Thread.currentThread().interrupt();
        }
    }
}

示例.aj

public aspect Example {
     
     Object around(Runnable task) : call(public Future<?> java.util.concurrent.ExecutorService+.submit(Runnable)) 
                        && args(task)
     {
         final long parentID = Thread.currentThread().getId();
         Runnable newTask =  () -> {
                 System.out.println("Parent Thread Id: "+ parentID);
                 System.out.println("Child Thread Id: "+Thread.currentThread().getId()); 
                 task.run();
         };
         return proceed(newTask);
     }
}