增加 TTL

增加多租户与上下文的集成

Author: YunaiV

lab-12-mybatis/lab-12-mybatis-plus-tenant/pom.xml

@@ -38,6 +38,12 @@
             <scope>test</scope>
         </dependency>
 
+        <dependency>
+            <groupId>com.alibaba</groupId>
+            <artifactId>transmittable-thread-local</artifactId>
+            <version>2.12.2</version>
+        </dependency>
+
     </dependencies>
 
 </project>

lab-12-mybatis/lab-12-mybatis-plus-tenant/src/main/java/cn/iocoder/springboot/lab12/mybatis/config/AsyncConfig.java

@@ -0,0 +1,32 @@
+package cn.iocoder.springboot.lab12.mybatis.config;
+
+import com.alibaba.ttl.TtlRunnable;
+import org.springframework.beans.BeansException;
+import org.springframework.beans.factory.config.BeanPostProcessor;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.scheduling.annotation.EnableAsync;
+import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
+
+@Configuration
+@EnableAsync
+public class AsyncConfig {
+
+    @Bean
+    public BeanPostProcessor executorBeanPostProcessor() {
+        return new BeanPostProcessor() {
+
+            @Override
+            public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
+                if (!(bean instanceof ThreadPoolTaskExecutor)) {
+                    return bean;
+                }
+                ThreadPoolTaskExecutor executor = (ThreadPoolTaskExecutor) bean;
+                executor.setTaskDecorator(TtlRunnable::get);
+                return executor;
+            }
+
+        };
+    }
+
+}

lab-12-mybatis/lab-12-mybatis-plus-tenant/src/main/java/cn/iocoder/springboot/lab12/mybatis/config/MybatisPlusConfig.java

@@ -1,5 +1,6 @@
 package cn.iocoder.springboot.lab12.mybatis.config;
 
+import cn.iocoder.springboot.lab12.mybatis.context.TenantHolder;
 import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
 import com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler;
 import com.baomidou.mybatisplus.extension.plugins.inner.TenantLineInnerInterceptor;
@@ -21,7 +22,8 @@ public MybatisPlusInterceptor mybatisPlusInterceptor() {
 
             @Override
             public Expression getTenantId() {
-                return new LongValue(10);
+                Integer tenantId = TenantHolder.getTenantId();
+                return new LongValue(tenantId);
             }
 
             // 这是 default 方法,默认返回 false 表示所有表都需要拼多租户条件

lab-12-mybatis/lab-12-mybatis-plus-tenant/src/main/java/cn/iocoder/springboot/lab12/mybatis/context/TenantHolder.java

@@ -0,0 +1,19 @@
+package cn.iocoder.springboot.lab12.mybatis.context;
+
+import com.alibaba.ttl.TransmittableThreadLocal;
+
+public class TenantHolder {
+
+    private static final ThreadLocal<Integer> TENANT_ID = new TransmittableThreadLocal<>();
+//    private static final ThreadLocal<Integer> TENANT_ID = new ThreadLocal<>();
+//    private static final ThreadLocal<Integer> TENANT_ID = new InheritableThreadLocal<>();
+
+    public static void setTenantId(Integer tenantId) {
+        TENANT_ID.set(tenantId);
+    }
+
+    public static Integer getTenantId() {
+        return TENANT_ID.get();
+    }
+
+}

lab-12-mybatis/lab-12-mybatis-plus-tenant/src/main/java/cn/iocoder/springboot/lab12/mybatis/core/TtlThreadPoolTaskExecutor.java

@@ -0,0 +1,44 @@
+package cn.iocoder.springboot.lab12.mybatis.core;
+
+import com.alibaba.ttl.TtlCallable;
+import com.alibaba.ttl.TtlRunnable;
+import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
+import org.springframework.util.concurrent.ListenableFuture;
+
+import java.util.Objects;
+import java.util.concurrent.Callable;
+import java.util.concurrent.Future;
+
+@Deprecated
+public class TtlThreadPoolTaskExecutor extends ThreadPoolTaskExecutor {
+
+    @Override
+    public void execute(Runnable task) {
+        super.execute(Objects.requireNonNull(TtlRunnable.get(task)));
+    }
+
+    @Override
+    public void execute(Runnable task, long startTimeout) {
+        super.execute(Objects.requireNonNull(TtlRunnable.get(task)), startTimeout);
+    }
+
+    @Override
+    public Future<?> submit(Runnable task) {
+        return super.submit(Objects.requireNonNull(TtlRunnable.get(task)));
+    }
+
+    @Override
+    public <T> Future<T> submit(Callable<T> task) {
+        return super.submit(Objects.requireNonNull(TtlCallable.get(task)));
+    }
+
+    @Override
+    public ListenableFuture<?> submitListenable(Runnable task) {
+        return super.submitListenable(Objects.requireNonNull(TtlRunnable.get(task)));
+    }
+
+    @Override
+    public <T> ListenableFuture<T> submitListenable(Callable<T> task) {
+        return super.submitListenable(Objects.requireNonNull(TtlCallable.get(task)));
+    }
+}

lab-12-mybatis/lab-12-mybatis-plus-tenant/src/main/java/cn/iocoder/springboot/lab12/mybatis/service/UserService.java

@@ -0,0 +1,29 @@
+package cn.iocoder.springboot.lab12.mybatis.service;
+
+import cn.iocoder.springboot.lab12.mybatis.dataobject.UserDO;
+import cn.iocoder.springboot.lab12.mybatis.mapper.UserMapper;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.scheduling.annotation.Async;
+import org.springframework.scheduling.annotation.AsyncResult;
+import org.springframework.stereotype.Service;
+
+import javax.annotation.Resource;
+import java.util.concurrent.Future;
+
+@Service
+public class UserService {
+
+    private final Logger log = LoggerFactory.getLogger(UserService.class);
+
+    @Resource
+    private UserMapper userMapper;
+
+    @Async
+    public Future<UserDO> getUserAsync(Integer id) {
+        UserDO userDO = userMapper.selectById(id);
+        log.info("[getUserAsync][id({}) user({})]", id, userDO);
+        return AsyncResult.forValue(userDO);
+    }
+
+}

lab-12-mybatis/lab-12-mybatis-plus-tenant/src/main/java/cn/iocoder/springboot/lab12/mybatis/util/TtlExecutorsUtil.java

@@ -0,0 +1,23 @@
+package cn.iocoder.springboot.lab12.mybatis.util;
+
+import com.alibaba.ttl.spi.TtlEnhanced;
+import com.alibaba.ttl.threadpool.agent.TtlAgent;
+import org.springframework.lang.Nullable;
+
+import java.util.concurrent.Executor;
+
+/**
+ * {@link com.alibaba.ttl.threadpool.TtlExecutors} 工具类
+ */
+@Deprecated
+public class TtlExecutorsUtil {
+
+    public static Executor getTtlThreadPoolTaskExecutor(@Nullable Executor executor) {
+        if (TtlAgent.isTtlAgentLoaded() || null == executor || executor instanceof TtlEnhanced) {
+            return executor;
+        }
+//        return new ExecutorTtlWrapper(executor, true);
+        return null;
+    }
+
+}

lab-12-mybatis/lab-12-mybatis-plus-tenant/src/test/java/cn/iocoder/springboot/lab12/mybatis/service/UserServiceTest.java

@@ -0,0 +1,31 @@
+package cn.iocoder.springboot.lab12.mybatis.service;
+
+import cn.iocoder.springboot.lab12.mybatis.Application;
+import cn.iocoder.springboot.lab12.mybatis.context.TenantHolder;
+import cn.iocoder.springboot.lab12.mybatis.dataobject.UserDO;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.junit4.SpringRunner;
+
+import javax.annotation.Resource;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.Future;
+
+import static org.junit.Assert.*;
+
+@RunWith(SpringRunner.class)
+@SpringBootTest(classes = Application.class)
+public class UserServiceTest {
+
+    @Resource
+    private UserService userService;
+
+    @Test
+    public void testGetUserAsync() throws ExecutionException, InterruptedException {
+        TenantHolder.setTenantId(10); // TODO 芋艿：写死
+        Future<UserDO> future = userService.getUserAsync(9);
+        future.get();
+    }
+
+}