异步任务扩展 - ThreadLocal上下文拷贝示例

Author: lvyahui8

spring-boot-data-aggregator-example/src/main/java/io/github/lvyahui8/spring/example/context/ExampleAppContext.java

@@ -28,6 +28,10 @@ public static String getUsername() {
         return LOGGED_USER.get() != null ? LOGGED_USER.get().getUsername() : null;
     }
 
+    public static User getUser() {
+        return LOGGED_USER.get();
+    }
+
     public static void remove() {
         LOGGED_USER.remove();
     }

spring-boot-data-aggregator-example/src/main/java/io/github/lvyahui8/spring/example/wrapper/CustomAsyncQueryTaskWrapper.java

@@ -0,0 +1,42 @@
+package io.github.lvyahui8.spring.example.wrapper;
+
+import io.github.lvyahui8.spring.aggregate.service.AsyncQueryTaskWrapper;
+import io.github.lvyahui8.spring.example.context.ExampleAppContext;
+import io.github.lvyahui8.spring.example.context.RequestContext;
+import io.github.lvyahui8.spring.example.model.User;
+import lombok.extern.slf4j.Slf4j;
+
+/**
+ * @author lvyahui (lvyahui8@gmail.com,lvyahui8@126.com)
+ * @since 2020/1/5 13:10
+ */
+@Slf4j
+public class CustomAsyncQueryTaskWrapper implements AsyncQueryTaskWrapper {
+
+    private Long tenantId;
+    private User user;
+
+    @Override
+    public void beforeSubmit() {
+        /* 提交任务前, 先从当前线程拷贝出ThreadLocal中的数据到任务中 */
+        log.info("asyncTask beforeSubmit. threadName: {}",Thread.currentThread().getName());
+        this.tenantId = RequestContext.getTenantId();
+        this.user = ExampleAppContext.getUser();
+    }
+
+    @Override
+    public void beforeExecute(Thread taskFrom) {
+        /* 任务提交后, 执行前, 在异步线程中用数据恢复ThreadLocal(Context) */
+        log.info("asyncTask beforeExecute. threadName: {}, taskFrom: {}",Thread.currentThread().getName(),taskFrom.getName());
+        RequestContext.setTenantId(tenantId);
+        ExampleAppContext.setLoggedUser(user);
+    }
+
+    @Override
+    public void afterExecute(Thread taskFrom) {
+        /* 任务执行完成后, 清理异步线程中的ThreadLocal(Context) */
+        log.info("asyncTask afterExecute. threadName: {}, taskFrom: {}",Thread.currentThread().getName(),taskFrom.getName());
+        RequestContext.removeTenantId();
+        ExampleAppContext.remove();
+    }
+}

spring-boot-data-aggregator-example/src/main/resources/application.properties

@@ -1,5 +1,5 @@
 spring.main.banner-mode=off
 io.github.lvyahui8.spring.base-packages=io.github.lvyahui8.spring.example
 io.github.lvyahui8.spring.ignore-exception=false
-io.github.lvyahui8.spring.task-wrapper-class=io.github.lvyahui8.spring.aggregate.service.AsyncQueryTaskWrapperAdapter
+io.github.lvyahui8.spring.task-wrapper-class=io.github.lvyahui8.spring.example.wrapper.CustomAsyncQueryTaskWrapper
 example.logging=true