gray aop conponent

Author: bage2014

study-spring-boot/src/main/java/com/bage/study/springboot/aop/annotation/gray/GrayConfig.java

@@ -3,12 +3,15 @@
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Component;
 
+import java.util.Random;
+
 @Component
 @Slf4j
 public class GrayConfig {
 
     public boolean matchGraySwitch(String key) {
-        return false;
+        int nexted = new Random().nextInt(100);
+        return nexted < 50;
     }
 
 }

study-spring-boot/src/main/java/com/bage/study/springboot/aop/annotation/gray/GrayFlowApplication.java

@@ -0,0 +1,23 @@
+package com.bage.study.springboot.aop.annotation.gray;
+
+import com.bage.study.springboot.aop.annotation.rest.RestResult;
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+@SpringBootApplication
+@RestController
+public class GrayFlowApplication {
+
+    public static void main(String[] args) {
+        SpringApplication.run(GrayFlowApplication.class, args);
+    }
+
+    @RestResult
+    @RequestMapping("/hello")
+    public Object hello(){
+        return "hello";
+    }
+
+}

study-spring-boot/src/main/java/com/bage/study/springboot/aop/annotation/gray/OrderGrayFlow.java

@@ -11,6 +11,6 @@
 
     String keySpEL() default "";
 
-    String copyToMethod() default ""; // 到类对应方法，默认是相同方法
+    String toMethod() default ""; // 到类对应方法，默认是相同方法
 
 }

study-spring-boot/src/main/java/com/bage/study/springboot/aop/annotation/gray/OrderGrayFlowAspect.java

@@ -21,7 +21,7 @@ public class OrderGrayFlowAspect {
     @Autowired
     private OrderGrayFlowLogic orderGrayFlowLogic;
 
-    @Around(value = "@annotation(com.ctrip.train.trnorder.core.biz.temp.gray.OrderGrayFlow)")
+    @Around(value = "@annotation(com.bage.study.springboot.aop.annotation.gray.OrderGrayFlow)")
     public Object around(ProceedingJoinPoint pjp) throws Throwable {
         Object[] args = null;
         Object result = null;

study-spring-boot/src/main/java/com/bage/study/springboot/aop/annotation/gray/OrderGrayFlowLogic.java

@@ -45,7 +45,7 @@ public Object getBean(OrderGrayFlow annotation, Method method) {
     public Object invokeSync(Object bean, OrderGrayFlow annotation, Method method, Object[] args) {         // 同步，直接执行
         Object invoke = null;
         try {
-            String methodName = "".equals(annotation.copyToMethod()) ? method.getName() : annotation.copyToMethod();
+            String methodName = "".equals(annotation.toMethod()) ? method.getName() : annotation.toMethod();
             Class grayToClass = annotation.toClass();
             Method toMethod = grayToClass.getMethod(methodName, method.getParameterTypes());
             invoke = toMethod.invoke(bean, args);

study-spring-boot/src/test/java/com/bage/study/springboot/aop/annotation/gray/GrayFlowTest.java

@@ -0,0 +1,28 @@
+package com.bage.study.springboot.aop.annotation.gray;
+
+
+import com.bage.study.springboot.aop.annotation.flow.copy.FlowCopyApplication;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.junit4.SpringRunner;
+
+@RunWith(SpringRunner.class)
+@SpringBootTest(classes = GrayFlowApplication.class)
+public class GrayFlowTest {
+
+    @Autowired
+    MyHelloService helloService;
+
+    @Test
+    public void sayHi() {
+        helloService.sayHi("hihi");
+    }
+
+    @Test
+    public void sayHiHey() {
+        helloService.sayHi("hihi");
+        helloService.sayHey("heihie");
+    }
+}

study-spring-boot/src/test/java/com/bage/study/springboot/aop/annotation/gray/HelloService.java

@@ -0,0 +1,11 @@
+package com.bage.study.springboot.aop.annotation.gray;
+
+public interface HelloService {
+
+    String sayHi(String param);
+
+    String sayHey(String param);
+
+    String sayYo(String param);
+
+}

study-spring-boot/src/test/java/com/bage/study/springboot/aop/annotation/gray/MyHelloService.java

@@ -0,0 +1,28 @@
+package com.bage.study.springboot.aop.annotation.gray;
+
+import org.springframework.stereotype.Component;
+
+@Component
+public class MyHelloService implements HelloService {
+
+    @Override
+    @OrderGrayFlow(toClass = YouHelloService.class)
+    public String sayHi(String param) {
+        System.out.println("MyHelloService.sayHi:" + param);
+        return "MyHelloService.sayHi:" + param;
+    }
+
+    @Override
+    @OrderGrayFlow(toClass = YouHelloService.class)
+    public String sayHey(String param) {
+        System.out.println("MyHelloService.sayHey:" + param);
+        return "MyHelloService.sayHey:" + param;
+    }
+
+    @Override
+    public String sayYo(String param) {
+        System.out.println("MyHelloService.sayYo:" + param);
+        return "MyHelloService.sayYo:" + param;
+    }
+
+}

study-spring-boot/src/test/java/com/bage/study/springboot/aop/annotation/gray/TheirHelloService.java

@@ -0,0 +1,27 @@
+package com.bage.study.springboot.aop.annotation.gray;
+
+import org.springframework.stereotype.Component;
+
+@Component
+public class TheirHelloService implements HelloService {
+
+    @Override
+    @OrderGrayFlow(toClass = MyHelloService.class, toMethod = "sayHi")
+    public String sayHi(String param) {
+        System.out.println("TheirHelloService.sayHi:" + param);
+        return "TheirHelloService.sayHi:" + param;
+    }
+
+    @Override
+    public String sayHey(String param) {
+        System.out.println("TheirHelloService.sasayHeyyHi:" + param);
+        return "TheirHelloService.sayHey:" + param;
+    }
+
+    @Override
+    public String sayYo(String param) {
+        System.out.println("TheirHelloService.sayYo:" + param);
+        return "TheirHelloService.sayYo:" + param;
+    }
+
+}

study-spring-boot/src/test/java/com/bage/study/springboot/aop/annotation/gray/YouHelloService.java

@@ -0,0 +1,31 @@
+package com.bage.study.springboot.aop.annotation.gray;
+
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+@Component
+public class YouHelloService implements HelloService {
+
+    @Autowired
+    private TheirHelloService theirHelloService;
+
+    @Override
+    @OrderGrayFlow(toClass = MyHelloService.class, toMethod = "sayHi")
+    public String sayHi(String param) {
+        System.out.println("YouHelloService.sayHi:" + param);
+        return "YouHelloService.sayHi:" + param;
+    }
+
+    @Override
+    public String sayHey(String param) {
+        System.out.println("YouHelloService.sasayHeyyHi:" + param);
+        return "YouHelloService.sayHey:" + param;
+    }
+
+    @Override
+    public String sayYo(String param) {
+        System.out.println("YouHelloService.sayYo:" + param);
+        return "YouHelloService.sayYo:" + param;
+    }
+
+}