动态数据源实现[优化]: 使用aop加注解 优化代码

Author: 曾梓健

springboot-dataaccess/pom.xml

@@ -50,6 +50,10 @@
 			<groupId>org.springframework.boot</groupId>
 			<artifactId>spring-boot-starter-web</artifactId>
 		</dependency>
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-starter-aop</artifactId>
+		</dependency>
 	</dependencies>
 
 	<build>

springboot-dataaccess/src/main/java/com/hand/springbootdataaccess/aop/RoutingAspect.java

@@ -0,0 +1,23 @@
+package com.hand.springbootdataaccess.aop;
+
+import com.hand.springbootdataaccess.config.RoutingDataSourceContext;
+import com.hand.springbootdataaccess.config.RoutingWith;
+import org.aspectj.lang.ProceedingJoinPoint;
+import org.aspectj.lang.annotation.Around;
+import org.aspectj.lang.annotation.Aspect;
+import org.springframework.stereotype.Component;
+
+/**
+ * @author zijian.zeng@hand-china.com
+ * @since 2023-01-30
+ */
+@Aspect
+@Component
+public class RoutingAspect {
+
+	@Around("@annotation(routingWith)")
+	public Object routingWithDataSource(ProceedingJoinPoint joinPoint, RoutingWith routingWith) throws Throwable {
+		new RoutingDataSourceContext(routingWith.key());
+		return joinPoint.proceed();
+	}
+}

springboot-dataaccess/src/main/java/com/hand/springbootdataaccess/config/RoutingWith.java

@@ -0,0 +1,17 @@
+package com.hand.springbootdataaccess.config;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+/**
+ * @author zijian.zeng@hand-china.com
+ * @since 2023-01-30
+ */
+@Target(ElementType.METHOD)
+@Retention(RetentionPolicy.RUNTIME)
+public @interface RoutingWith {
+
+	String key() default "master";
+}

springboot-dataaccess/src/main/java/com/hand/springbootdataaccess/controller/ProductController.java

@@ -2,6 +2,7 @@
 
 import com.hand.springbootdataaccess.config.MyDataSourceAutoConfiguration;
 import com.hand.springbootdataaccess.config.RoutingDataSourceContext;
+import com.hand.springbootdataaccess.config.RoutingWith;
 import com.hand.springbootdataaccess.service.ProductService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.RequestMapping;
@@ -19,15 +20,15 @@ public class ProductController {
 
 
 	@RequestMapping("/findAllM")
+	@RoutingWith
 	public String findAllM() {
-		new RoutingDataSourceContext(MyDataSourceAutoConfiguration.MASTER);
 		productService.findAllProductM();
 		return "m";
 	}
 
 	@RequestMapping("/findAllS")
+	@RoutingWith(key = "slave")
 	public String findAllS() {
-		new RoutingDataSourceContext(MyDataSourceAutoConfiguration.SLAVE);
 		productService.findAllProductS();
 		return "s";
 	}