data-fast support spring framework advisors

Author: xlorne

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/DataFastConfiguration.java

@@ -3,6 +3,7 @@
 import com.codingapi.springboot.fast.executor.JpaExecutor;
 import com.codingapi.springboot.fast.mapping.MvcEndpointMapping;
 import com.codingapi.springboot.fast.registrar.MvcMappingRegistrar;
+import org.springframework.aop.Advisor;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
 import org.springframework.context.annotation.Bean;
@@ -11,6 +12,7 @@
 import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerMapping;
 
 import javax.persistence.EntityManager;
+import java.util.List;
 
 @Configuration
 @ConditionalOnClass(WebMvcConfigurer.class)
@@ -25,8 +27,10 @@ public MvcEndpointMapping mvcEndpointMapping(RequestMappingHandlerMapping handle
 
     @Bean(initMethod = "registerMvcMapping")
     @ConditionalOnMissingBean
-    public MvcMappingRegistrar mappingRegistrar(MvcEndpointMapping mvcEndpointMapping, JpaExecutor jpaExecutor) {
-        return new MvcMappingRegistrar(mvcEndpointMapping, jpaExecutor);
+    public MvcMappingRegistrar mappingRegistrar(MvcEndpointMapping mvcEndpointMapping,
+                                                JpaExecutor jpaExecutor,
+                                                List<Advisor> advisors) {
+        return new MvcMappingRegistrar(mvcEndpointMapping, jpaExecutor,advisors);
     }
 
     @Bean

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/executor/MvcMethodProxy.java renamed to springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/executor/MvcMethodInterceptor.java

@@ -2,32 +2,38 @@
 
 import com.codingapi.springboot.fast.annotation.FastMapping;
 import lombok.AllArgsConstructor;
+import org.aopalliance.intercept.MethodInterceptor;
+import org.aopalliance.intercept.MethodInvocation;
 
-import java.lang.reflect.InvocationHandler;
 import java.lang.reflect.Method;
 
 @AllArgsConstructor
-public class MvcMethodProxy implements InvocationHandler {
+public class MvcMethodInterceptor implements MethodInterceptor {
 
     private final JpaExecutor jpaExecutor;
 
     @Override
-    public Object invoke(Object proxy, Method method, Object[] args)
+    public Object invoke(MethodInvocation invocation)
             throws Throwable {
+        Method method = invocation.getMethod();
+        Object[] args = invocation.getArguments();
 
         if (method.equals(Object.class.getMethod("equals", Object.class))) {
             return false;
         }
+
         if (method.equals(Object.class.getMethod("hashCode"))) {
             return hashCode();
         }
+
         FastMapping fastMapping = method.getAnnotation(FastMapping.class);
         if (fastMapping != null) {
             Class<?> returnType = method.getReturnType();
             return jpaExecutor.execute(fastMapping.value(), fastMapping.countQuery(), args, returnType);
         }
-
         // mvc mapping proxy can't execute return null.
         return null;
     }
+
+
 }

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/registrar/MvcMappingRegistrar.java

@@ -3,17 +3,21 @@
 import com.codingapi.springboot.fast.annotation.FastMapping;
 import com.codingapi.springboot.fast.exception.FastMappingErrorException;
 import com.codingapi.springboot.fast.executor.JpaExecutor;
-import com.codingapi.springboot.fast.executor.MvcMethodProxy;
+import com.codingapi.springboot.fast.executor.MvcMethodInterceptor;
 import com.codingapi.springboot.fast.mapping.MvcEndpointMapping;
 import lombok.AllArgsConstructor;
 import lombok.SneakyThrows;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.aop.Advisor;
+import org.springframework.aop.framework.AdvisedSupport;
+import org.springframework.aop.framework.AopProxy;
+import org.springframework.aop.framework.DefaultAopProxyFactory;
 import org.springframework.data.domain.Pageable;
 import org.springframework.util.StringUtils;
 
 import java.lang.reflect.Method;
-import java.lang.reflect.Proxy;
 import java.util.HashSet;
+import java.util.List;
 import java.util.Set;
 
 @Slf4j
@@ -23,43 +27,59 @@ public class MvcMappingRegistrar {
     private final MvcEndpointMapping mvcEndpointMapping;
     private final JpaExecutor jpaExecutor;
 
+    private final DefaultAopProxyFactory proxyFactory = new DefaultAopProxyFactory();
+
+    private final List<Advisor> advisors;
+
     @SneakyThrows
     public void registerMvcMapping() {
         for (Class<?> clazz : classSet) {
             Method[] methods = clazz.getDeclaredMethods();
             for (Method method : methods) {
                 FastMapping fastMapping = method.getAnnotation(FastMapping.class);
                 if (verify(fastMapping, method)) {
-                    MvcMethodProxy handler = new MvcMethodProxy(jpaExecutor);
-                    Object methodProxy = Proxy.newProxyInstance(clazz.getClassLoader(), new Class[]{clazz}, handler);
-                    mvcEndpointMapping.addMapping(fastMapping.mapping(), fastMapping.method(), methodProxy, method);
+                    AdvisedSupport advisedSupport = createAdvisedSupport(clazz);
+                    AopProxy proxy = proxyFactory.createAopProxy(advisedSupport);
+                    mvcEndpointMapping.addMapping(fastMapping.mapping(), fastMapping.method(),
+                            proxy.getProxy(), method);
                 }
             }
         }
     }
 
+    private AdvisedSupport createAdvisedSupport(Class<?> clazz){
+        AdvisedSupport advisedSupport = new AdvisedSupport(clazz);
+        MvcMethodInterceptor interceptor = new MvcMethodInterceptor(jpaExecutor);
+        advisedSupport.setTarget(interceptor);
+        advisedSupport.addAdvisors(advisors);
+        advisedSupport.addAdvice(interceptor);
+        return advisedSupport;
+    }
+
     private boolean verify(FastMapping fastMapping, Method method) throws FastMappingErrorException {
         if (fastMapping == null) {
             return false;
         }
 
         if (!StringUtils.hasText(fastMapping.mapping())) {
-            throw new FastMappingErrorException(String.format("fast method %s missing mapping .", method.getName()));
+            throw new FastMappingErrorException(String.format("fast method %s missing mapping .",
+                    method.getName()));
         }
 
         if (!StringUtils.hasText(fastMapping.value())) {
-            throw new FastMappingErrorException(String.format("fast mapping %s missing value .", fastMapping.mapping()));
+            throw new FastMappingErrorException(String.format("fast mapping %s missing value .",
+                    fastMapping.mapping()));
         }
 
         Class<?>[] parameterTypes = method.getParameterTypes();
         for (Class<?> parameter : parameterTypes) {
             if (Pageable.class.isAssignableFrom(parameter)) {
                 if (!StringUtils.hasText(fastMapping.countQuery())) {
-                    throw new FastMappingErrorException(String.format("fast mapping %s missing countQuery .", fastMapping.mapping()));
+                    throw new FastMappingErrorException(String.format("fast mapping %s missing countQuery .",
+                            fastMapping.mapping()));
                 }
             }
         }
-
         return true;
     }