修改拦截器定义方式, 并支持多个拦截器注解排序

Author: lvyahui8

spring-boot-data-aggregator-autoconfigure/src/main/java/io/github/lvyahui8/spring/autoconfigure/BeanAggregateAutoConfiguration.java

@@ -3,6 +3,7 @@
 import io.github.lvyahui8.spring.aggregate.config.RuntimeSettings;
 import io.github.lvyahui8.spring.aggregate.facade.DataBeanAggregateQueryFacade;
 import io.github.lvyahui8.spring.aggregate.facade.impl.DataBeanAggregateQueryFacadeImpl;
+import io.github.lvyahui8.spring.aggregate.interceptor.AggregateQueryInterceptor;
 import io.github.lvyahui8.spring.aggregate.interceptor.AggregateQueryInterceptorChain;
 import io.github.lvyahui8.spring.aggregate.interceptor.impl.AggregateQueryInterceptorChainImpl;
 import io.github.lvyahui8.spring.aggregate.model.DataProvideDefinition;
@@ -24,14 +25,16 @@
 import org.springframework.context.ApplicationContextAware;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
+import org.springframework.core.Ordered;
 import org.springframework.core.annotation.AnnotationUtils;
+import org.springframework.core.annotation.Order;
 import org.springframework.scheduling.concurrent.CustomizableThreadFactory;
 import org.springframework.util.Assert;
 import org.springframework.util.StringUtils;
 
 import java.lang.reflect.Method;
 import java.lang.reflect.Modifier;
-import java.util.Set;
+import java.util.*;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.LinkedBlockingDeque;
 import java.util.concurrent.ThreadPoolExecutor;
@@ -104,7 +107,7 @@ public DataBeanAggregateQueryService dataBeanAggregateQueryService (
      * @return
      */
     @Bean(name = "aggregateExecutorService")
-    @ConditionalOnMissingBean(name = "aggregateExecutorService")
+    @ConditionalOnMissingBean(name = "aggregateExecutorService",value=ExecutorService.class)
     public ExecutorService aggregateExecutorService() {
         return new ThreadPoolExecutor(
                 properties.getThreadNumber(),
@@ -129,6 +132,24 @@ public DataProviderRepository dataProviderRepository() {
     @Bean(name = "aggregateQueryInterceptorChain")
     @ConditionalOnMissingBean(AggregateQueryInterceptorChain.class)
     public AggregateQueryInterceptorChain aggregateQueryInterceptorChain() {
-        return new AggregateQueryInterceptorChainImpl();
+        Map<String, AggregateQueryInterceptor> interceptorMap = applicationContext.getBeansOfType(AggregateQueryInterceptor.class);
+        AggregateQueryInterceptorChainImpl interceptorChain = new AggregateQueryInterceptorChainImpl();
+        if(interceptorMap != null  && ! interceptorMap.isEmpty()) {
+            List<AggregateQueryInterceptor> interceptors = new ArrayList<>(interceptorMap.values());
+            interceptors.sort(new Comparator<AggregateQueryInterceptor>() {
+                @Override
+                public int compare(AggregateQueryInterceptor o1, AggregateQueryInterceptor o2) {
+                    Order order1 = o1.getClass().getAnnotation(Order.class);
+                    Order order2 = o2.getClass().getAnnotation(Order.class);
+                    int oi1 = order1 == null ? Ordered.LOWEST_PRECEDENCE :  order1.value();
+                    int oi2 = order2 == null ?  Ordered.LOWEST_PRECEDENCE : order2.value();
+                    return oi1 - oi2;
+                }
+            });
+            for (AggregateQueryInterceptor interceptor : interceptors) {
+                interceptorChain.addInterceptor(interceptor);
+            }
+        }
+        return interceptorChain;
     }
 }

spring-boot-data-aggregator-example/src/main/java/io/github/lvyahui8/spring/example/configuration/AggregatorCustomConfiguration.java

@@ -1,9 +1,5 @@
 package io.github.lvyahui8.spring.example.configuration;
 
-import io.github.lvyahui8.spring.aggregate.interceptor.AggregateQueryInterceptorChain;
-import io.github.lvyahui8.spring.aggregate.interceptor.impl.AggregateQueryInterceptorChainImpl;
-import io.github.lvyahui8.spring.example.interceptor.SampleAggregateQueryInterceptor;
-import org.springframework.beans.factory.annotation.Qualifier;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.scheduling.concurrent.CustomizableThreadFactory;
@@ -30,18 +26,4 @@ public ExecutorService aggregateExecutorService() {
                 new LinkedBlockingDeque<>(1024),
                 new CustomizableThreadFactory("example-async"));
     }
-
-    /**
-     * 自定义拦截器处理链
-     *
-     * @param sampleAggregateQueryInterceptor
-     * @return
-     */
-    @Bean(name = "aggregateQueryInterceptorChain")
-    public AggregateQueryInterceptorChain aggregateQueryInterceptorChain(
-            @Qualifier("sampleAggregateQueryInterceptor") SampleAggregateQueryInterceptor sampleAggregateQueryInterceptor) {
-        AggregateQueryInterceptorChainImpl aggregateQueryInterceptorChain = new AggregateQueryInterceptorChainImpl();
-        aggregateQueryInterceptorChain.addInterceptor(sampleAggregateQueryInterceptor);
-        return aggregateQueryInterceptorChain;
-    }
 }

spring-boot-data-aggregator-example/src/main/java/io/github/lvyahui8/spring/example/interceptor/PerSetupAggregateQueryInterceptor.java

@@ -0,0 +1,22 @@
+package io.github.lvyahui8.spring.example.interceptor;
+
+import io.github.lvyahui8.spring.aggregate.context.AggregationContext;
+import io.github.lvyahui8.spring.aggregate.interceptor.impl.AggregateQueryInterceptorAdapter;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.core.annotation.Order;
+import org.springframework.stereotype.Component;
+
+/**
+ * @author lvyahui (lvyahui8@gmail.com,lvyahui8@126.com)
+ * @since 2019/9/7 23:02
+ */
+@Component
+@Order(1)
+@Slf4j
+public class PerSetupAggregateQueryInterceptor extends AggregateQueryInterceptorAdapter {
+    @Override
+    public boolean querySubmitted(AggregationContext aggregationContext) {
+        log.info("current thread {}", Thread.currentThread().getName());
+        return super.querySubmitted(aggregationContext);
+    }
+}

spring-boot-data-aggregator-example/src/main/java/io/github/lvyahui8/spring/example/interceptor/SampleAggregateQueryInterceptor.java

@@ -4,13 +4,15 @@
 import io.github.lvyahui8.spring.aggregate.interceptor.AggregateQueryInterceptor;
 import io.github.lvyahui8.spring.aggregate.model.DataProvideDefinition;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.core.annotation.Order;
 import org.springframework.stereotype.Component;
 
 /**
  * @author lvyahui (lvyahui8@gmail.com,lvyahui8@126.com)
  * @since 2019/8/4 17:24
  */
 @Component
+@Order(2)
 @Slf4j
 public class SampleAggregateQueryInterceptor implements AggregateQueryInterceptor {
     @Override