基于AbstractRoutingDataSource自动切换数据源

nononi · nononi · commit ca0240d737d5 · 2022-04-17T13:16:11.000+08:00
diff --git a/06db/work9ofweek7/src/main/java/io/nononi/dynamicdb/annotation/ReadOnly.java b/06db/work9ofweek7/src/main/java/io/nononi/dynamicdb/annotation/ReadOnly.java
@@ -0,0 +1,15 @@
+package io.nononi.dynamicdb.annotation;
+
+import org.springframework.stereotype.Component;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+@Target(ElementType.METHOD)
+@Retention(RetentionPolicy.RUNTIME)
+@Component
+public @interface ReadOnly {
+
+}
diff --git a/06db/work9ofweek7/src/main/java/io/nononi/dynamicdb/aspect/ReadOnlyAspect.java b/06db/work9ofweek7/src/main/java/io/nononi/dynamicdb/aspect/ReadOnlyAspect.java
@@ -0,0 +1,44 @@
+package io.nononi.dynamicdb.aspect;
+
+import io.nononi.dynamicdb.annotation.ReadOnly;
+import io.nononi.dynamicdb.datasource.DynamicDataSource;
+import org.aspectj.lang.ProceedingJoinPoint;
+import org.aspectj.lang.annotation.Around;
+import org.aspectj.lang.annotation.Aspect;
+import org.aspectj.lang.annotation.Pointcut;
+import org.aspectj.lang.reflect.MethodSignature;
+import org.springframework.stereotype.Component;
+
+import java.lang.reflect.Method;
+
+@Aspect
+@Component
+public class ReadOnlyAspect {
+
+    @Pointcut("@annotation(io.nononi.dynamicdb.annotation.ReadOnly)")
+    public void readOnly(){};
+
+
+    @Around("readOnly()")
+    public Object around(ProceedingJoinPoint point) throws Throwable{
+        MethodSignature signature = (MethodSignature) point.getSignature();
+        Method method = signature.getMethod();
+
+        ReadOnly readOnly = method.getAnnotation(ReadOnly.class);
+        if(readOnly == null){
+            DynamicDataSource.setDataSource("master");
+            System.out.println("datasource is :" + "master");
+        } else {
+            DynamicDataSource.setDataSource("slave");
+            System.out.println("datasource is :" + "slave");
+        }
+
+        try {
+            return point.proceed();
+        }finally {
+            DynamicDataSource.clearDataSource();
+            System.out.println("clean datasource");
+        }
+    }
+
+}
diff --git a/06db/work9ofweek7/src/main/java/io/nononi/dynamicdb/datasource/DynamicDataSource.java b/06db/work9ofweek7/src/main/java/io/nononi/dynamicdb/datasource/DynamicDataSource.java
@@ -0,0 +1,39 @@
+package io.nononi.dynamicdb.datasource;
+
+import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
+
+import javax.sql.DataSource;
+import java.util.Map;
+
+public class  DynamicDataSource extends AbstractRoutingDataSource {
+
+
+    /**
+     * threadLocal &#35774;&#32622;&#25968;&#25454;&#28304;
+     */
+    private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>();
+
+    public DynamicDataSource(DataSource defaultTargetDataSource, Map<Object, Object> targetDataSources){
+        super.setDefaultTargetDataSource(defaultTargetDataSource);
+        super.setTargetDataSources(targetDataSources);
+        super.afterPropertiesSet();
+    }
+
+    @Override
+    protected Object determineCurrentLookupKey() {
+        return getDataSource();
+    }
+
+    public static void setDataSource(String dataSource) {
+        CONTEXT_HOLDER.set(dataSource);
+    }
+
+    public static String getDataSource() {
+        return CONTEXT_HOLDER.get();
+    }
+
+    public static void clearDataSource() {
+        CONTEXT_HOLDER.remove();
+    }
+
+}
diff --git a/06db/work9ofweek7/src/main/java/io/nononi/dynamicdb/datasource/DynamicDataSourceConfig.java b/06db/work9ofweek7/src/main/java/io/nononi/dynamicdb/datasource/DynamicDataSourceConfig.java
@@ -0,0 +1,44 @@
+package io.nononi.dynamicdb.datasource;
+
+import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.Primary;
+import org.springframework.jdbc.datasource.DriverManagerDataSource;
+
+import javax.sql.DataSource;
+import java.util.HashMap;
+import java.util.Map;
+
+@Configuration
+public class DynamicDataSourceConfig {
+
+    @Bean(name = "master")
+    public DataSource masterDatasource(){
+        DriverManagerDataSource dataSource = new DriverManagerDataSource();
+        dataSource.setDriverClassName("com.mysql.jdbc.Driver");
+        dataSource.setUrl("jdbc:mysql://localhost:3316/db");
+        dataSource.setUsername("root");
+        dataSource.setPassword("");
+        return dataSource;
+    }
+
+    @Bean(name = "slave")
+    public DataSource slaveDatasource(){
+        DriverManagerDataSource dataSource = new DriverManagerDataSource();
+        dataSource.setDriverClassName("com.mysql.jdbc.Driver");
+        dataSource.setUrl("jdbc:mysql://localhost:3326/db");
+        dataSource.setUsername("root");
+        dataSource.setPassword("");
+        return dataSource;
+    }
+
+    @Bean
+    @Primary
+    public DynamicDataSource dataSource(@Qualifier("master") DataSource masterDatasource, @Qualifier("slave") DataSource slaveDatasource){
+        Map<Object, Object> targetDataSources = new HashMap<>();
+        targetDataSources.put("master", masterDatasource);
+        targetDataSources.put("slave",slaveDatasource);
+        return new DynamicDataSource(masterDatasource,targetDataSources);
+    }
+}
diff --git a/06db/work9ofweek7/src/main/java/io/nononi/dynamicdb/service/OrderService.java b/06db/work9ofweek7/src/main/java/io/nononi/dynamicdb/service/OrderService.java
@@ -0,0 +1,6 @@
+package io.nononi.dynamicdb.service;
+
+public interface OrderService {
+    void insert(int id ,String name);
+    void query();
+}
diff --git a/06db/work9ofweek7/src/main/java/io/nononi/dynamicdb/service/impl/OrderServiceImpl.java b/06db/work9ofweek7/src/main/java/io/nononi/dynamicdb/service/impl/OrderServiceImpl.java
@@ -0,0 +1,55 @@
+package io.nononi.dynamicdb.service.impl;
+
+import io.nononi.dynamicdb.annotation.ReadOnly;
+import io.nononi.dynamicdb.datasource.DynamicDataSource;
+import io.nononi.dynamicdb.service.OrderService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+
+@Component
+public class OrderServiceImpl implements OrderService {
+
+    @Autowired
+    private DynamicDataSource dataSource;
+
+    @Override
+    public void insert(int id, String name) {
+        try {
+            Connection conn = dataSource.getConnection();
+            System.out.println(conn.getMetaData().getURL());
+            String sql = "INSERT INTO `order` (id,name) VALUES (?,?);";
+            PreparedStatement stmt = conn.prepareStatement(sql);
+            stmt.setInt(1,id);
+            stmt.setString(2,name);
+            stmt.executeUpdate();
+
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+
+    }
+
+    @Override
+    @ReadOnly
+    public void query() {
+        String sql = "SELECT * FROM `order`;";
+        try {
+            Connection conn = dataSource.getConnection();
+            System.out.println(conn.getMetaData().getURL());
+            PreparedStatement stmt = conn.prepareStatement(sql);
+            ResultSet resultSet = stmt.executeQuery();
+            while (resultSet.next()) {
+                System.out.print(resultSet.getInt("id") + " ");
+                System.out.println(resultSet.getString("name") + " ");
+            }
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+
+    }
+}
diff --git a/06db/work9ofweek7/src/test/java/io/nononi/dynamicdb/DynamicdbApplicationTests.java b/06db/work9ofweek7/src/test/java/io/nononi/dynamicdb/DynamicdbApplicationTests.java
@@ -0,0 +1,22 @@
+package io.nononi.dynamicdb;
+
+import io.nononi.dynamicdb.service.impl.OrderServiceImpl;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.junit.jupiter.SpringExtension;
+
+@SpringBootTest
+@ExtendWith(SpringExtension.class)
+class DynamicdbApplicationTests {
+    @Autowired
+    private OrderServiceImpl orderService;
+
+    @Test
+    public void testDynamicDb(){
+        orderService.insert(4,"ohhhhhh");
+        orderService.query();
+    }
+
+}
