optimize

Author: lvyahui8

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

@@ -0,0 +1,21 @@
+package io.github.lvyahui8.spring.example.context;
+
+/**
+ * @author lvyahui (lvyahui8@gmail.com,lvyahui8@126.com)
+ * @since 2019/7/22 23:11
+ */
+public class RequestContext {
+    private static ThreadLocal<Long> TENANT_ID = new ThreadLocal<>();
+
+    public static Long getTenantId() {
+        return TENANT_ID.get();
+    }
+
+    public static void setTenantId(Long tenantId) {
+        TENANT_ID.set(tenantId);
+    }
+
+    public static void removeTenantId() {
+        TENANT_ID.remove();
+    }
+}

spring-boot-data-aggregator-example/src/main/java/io/github/lvyahui8/spring/example/service/HomepageService.java

@@ -0,0 +1,34 @@
+package io.github.lvyahui8.spring.example.service;
+
+import io.github.lvyahui8.spring.example.model.Category;
+import io.github.lvyahui8.spring.example.model.Post;
+import io.github.lvyahui8.spring.example.model.User;
+
+import java.util.List;
+
+/**
+ *
+ * @author lvyahui (lvyahui8@gmail.com,lvyahui8@126.com)
+ * @since 2019/7/22 22:25
+ */
+public interface HomepageService {
+    /**
+     * 获取顶部菜单类目
+     *
+     * @return 顶部类目
+     */
+    List<Category> topMenu();
+
+    /**
+     * 文章列表
+     *
+     * @return xx
+     */
+    List<Post> postList();
+
+    /**
+     * 粉丝数据
+     * @return xxx
+     */
+    List<User> allFollowers();
+}

spring-boot-data-aggregator-example/src/main/java/io/github/lvyahui8/spring/example/service/impl/HomepageServiceImpl.java

@@ -0,0 +1,50 @@
+package io.github.lvyahui8.spring.example.service.impl;
+
+import io.github.lvyahui8.spring.annotation.DataProvider;
+import io.github.lvyahui8.spring.example.context.RequestContext;
+import io.github.lvyahui8.spring.example.model.Category;
+import io.github.lvyahui8.spring.example.model.Post;
+import io.github.lvyahui8.spring.example.model.User;
+import io.github.lvyahui8.spring.example.service.HomepageService;
+import org.springframework.stereotype.Service;
+import org.springframework.util.Assert;
+
+import java.util.List;
+
+/**
+ * @author lvyahui (lvyahui8@gmail.com,lvyahui8@126.com)
+ * @since 2019/7/22 23:08
+ */
+@Service
+public class HomepageServiceImpl implements HomepageService {
+
+    @DataProvider("topMenu")
+    @Override
+    public List<Category> topMenu() {
+        /* will be null */
+        Long tenantId = RequestContext.getTenantId();
+        Assert.notNull(tenantId,"tenantId must be not null");
+        // ... The content hereafter will be omitted.
+        return null;
+    }
+
+    @DataProvider("postList")
+    @Override
+    public List<Post> postList() {
+        /* will be null */
+        Long tenantId = RequestContext.getTenantId();
+        Assert.notNull(tenantId,"tenantId must be not null");
+        // ... The content hereafter will be omitted.
+        return null;
+    }
+
+    @DataProvider("allFollowers")
+    @Override
+    public List<User> allFollowers() {
+        /* will be null */
+        Long tenantId = RequestContext.getTenantId();
+        Assert.notNull(tenantId,"tenantId must be not null");
+        // ... The content hereafter will be omitted.
+        return null;
+    }
+}

spring-boot-data-aggregator-example/src/test/java/io/github/lvyahui8/spring/example/DataBeanAggregateQueryFacadeTest.java

@@ -2,9 +2,12 @@
 
 import io.github.lvyahui8.spring.aggregate.facade.DataBeanAggregateQueryFacade;
 import io.github.lvyahui8.spring.aggregate.func.Function2;
+import io.github.lvyahui8.spring.aggregate.func.Function3;
 import io.github.lvyahui8.spring.annotation.DataConsumer;
 import io.github.lvyahui8.spring.autoconfigure.BeanAggregateProperties;
 import io.github.lvyahui8.spring.example.context.ExampleAppContext;
+import io.github.lvyahui8.spring.example.context.RequestContext;
+import io.github.lvyahui8.spring.example.model.Category;
 import io.github.lvyahui8.spring.example.model.Post;
 import io.github.lvyahui8.spring.example.model.User;
 import lombok.extern.slf4j.Slf4j;
@@ -97,4 +100,25 @@ public User apply(@DataConsumer("loggedUsername") String loggedUsername,
             ExampleAppContext.remove();
         }
     }
+
+    @Test
+    public void testThreadLocal() throws Exception {
+        try {
+            RequestContext.setTenantId(10000L);
+            Object result = dataBeanAggregateQueryFacade.get(null,
+                    new Function3<List<Category>, List<Post>, List<User>, Object>() {
+                @Override
+                public Object apply(
+                        @DataConsumer("topMenu") List<Category> categories,
+                        @DataConsumer("postList") List<Post> posts,
+                        @DataConsumer("allFollowers") List<User> users) {
+                    return new Object[] {
+                            categories,posts,users
+                    };
+                }
+            });
+        } finally {
+            RequestContext.removeTenantId();
+        }
+    }
 }