Spring分库代理实现

src/main/java/top/fjy8018/designpattern/pattern/structural/proxy/db/DataSourceContextHolder.java

@@ -0,0 +1,34 @@
+package top.fjy8018.designpattern.pattern.structural.proxy.db;
+
+/**
+ * 数据源上下文
+ *
+ * @author F嘉阳
+ * @date 2020/2/29 16:27
+ */
+public class DataSourceContextHolder {
+    /**
+     * 存放数据源Bean Name
+     */
+    private static final ThreadLocal<String> CONTEXT_HOLDER = new ThreadLocal<>();
+
+    /**
+     * 配置哪个数据库，则动态数据源线程就会获取到哪个数据库
+     *
+     * @param dbType
+     */
+    public static void setDBType(String dbType) {
+        CONTEXT_HOLDER.set(dbType);
+    }
+
+    public static String getDBType() {
+        return CONTEXT_HOLDER.get();
+    }
+
+    /**
+     * 清除，防止内存泄漏
+     */
+    public static void clear() {
+        CONTEXT_HOLDER.remove();
+    }
+}

src/main/java/top/fjy8018/designpattern/pattern/structural/proxy/db/DynamicDataSource.java

@@ -0,0 +1,106 @@
+package top.fjy8018.designpattern.pattern.structural.proxy.db;
+
+import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
+
+/**
+ * Spring分库实现
+ *
+ * @author F嘉阳
+ * @date 2020/2/29 16:26
+ */
+public class DynamicDataSource extends AbstractRoutingDataSource {
+
+    /**
+     * 动态获取数据库
+     *
+     * @return 当前线程所属数据源
+     */
+    @Override
+    protected Object determineCurrentLookupKey() {
+        return DataSourceContextHolder.getDBType();
+    }
+
+
+//    <bean id="db0" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close">
+//        <property name="driverClassName" value="${db.driverClassName}"/>
+//        <property name="url" value="${db.url}"/>
+//        <property name="username" value="${db.username}"/>
+//        <property name="password" value="${db.password}"/>
+//        <!-- 连接池启动时的初始值 -->
+//        <property name="initialSize" value="${db.initialSize}"/>
+//        <!-- 连接池的最大值 -->
+//        <property name="maxActive" value="${db.maxActive}"/>
+//        <!-- 最大空闲值.当经过一个高峰时间后，连接池可以慢慢将已经用不到的连接慢慢释放一部分，一直减少到maxIdle为止 -->
+//        <property name="maxIdle" value="${db.maxIdle}"/>
+//        <!-- 最小空闲值.当空闲的连接数少于阀值时，连接池就会预申请去一些连接，以免洪峰来时来不及申请 -->
+//        <property name="minIdle" value="${db.minIdle}"/>
+//        <!-- 最大建立连接等待时间。如果超过此时间将接到异常。设为－1表示无限制 -->
+//        <property name="maxWait" value="${db.maxWait}"/>
+//        <!--#给出一条简单的sql语句进行验证 -->
+//         <!--<property name="validationQuery" value="select getdate()" />-->
+//        <property name="defaultAutoCommit" value="${db.defaultAutoCommit}"/>
+//        <!-- 回收被遗弃的（一般是忘了释放的）数据库连接到连接池中 -->
+//         <!--<property name="removeAbandoned" value="true" />-->
+//        <!-- 数据库连接过多长时间不用将被视为被遗弃而收回连接池中 -->
+//         <!--<property name="removeAbandonedTimeout" value="120" />-->
+//        <!-- #连接的超时时间，默认为半小时。 -->
+//        <property name="minEvictableIdleTimeMillis" value="${db.minEvictableIdleTimeMillis}"/>
+//
+//        <!--# 失效检查线程运行时间间隔，要小于MySQL默认-->
+//        <property name="timeBetweenEvictionRunsMillis" value="40000"/>
+//        <!--# 检查连接是否有效-->
+//        <property name="testWhileIdle" value="true"/>
+//        <!--# 检查连接有效性的SQL语句-->
+//        <property name="validationQuery" value="SELECT 1 FROM dual"/>
+//    </bean>
+
+
+//    <bean id="db1" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close">
+//        <property name="driverClassName" value="${db.driverClassName}"/>
+//        <property name="url" value="${db.url}"/>
+//        <property name="username" value="${db.username}"/>
+//        <property name="password" value="${db.password}"/>
+//        <!-- 连接池启动时的初始值 -->
+//        <property name="initialSize" value="${db.initialSize}"/>
+//        <!-- 连接池的最大值 -->
+//        <property name="maxActive" value="${db.maxActive}"/>
+//        <!-- 最大空闲值.当经过一个高峰时间后，连接池可以慢慢将已经用不到的连接慢慢释放一部分，一直减少到maxIdle为止 -->
+//        <property name="maxIdle" value="${db.maxIdle}"/>
+//        <!-- 最小空闲值.当空闲的连接数少于阀值时，连接池就会预申请去一些连接，以免洪峰来时来不及申请 -->
+//        <property name="minIdle" value="${db.minIdle}"/>
+//        <!-- 最大建立连接等待时间。如果超过此时间将接到异常。设为－1表示无限制 -->
+//        <property name="maxWait" value="${db.maxWait}"/>
+//        <!--#给出一条简单的sql语句进行验证 -->
+//         <!--<property name="validationQuery" value="select getdate()" />-->
+//        <property name="defaultAutoCommit" value="${db.defaultAutoCommit}"/>
+//        <!-- 回收被遗弃的（一般是忘了释放的）数据库连接到连接池中 -->
+//         <!--<property name="removeAbandoned" value="true" />-->
+//        <!-- 数据库连接过多长时间不用将被视为被遗弃而收回连接池中 -->
+//         <!--<property name="removeAbandonedTimeout" value="120" />-->
+//        <!-- #连接的超时时间，默认为半小时。 -->
+//        <property name="minEvictableIdleTimeMillis" value="${db.minEvictableIdleTimeMillis}"/>
+//
+//        <!--# 失效检查线程运行时间间隔，要小于MySQL默认-->
+//        <property name="timeBetweenEvictionRunsMillis" value="40000"/>
+//        <!--# 检查连接是否有效-->
+//        <property name="testWhileIdle" value="true"/>
+//        <!--# 检查连接有效性的SQL语句-->
+//        <property name="validationQuery" value="SELECT 1 FROM dual"/>
+//    </bean>
+
+
+//	<bean id="dataSource" class="com.geely.design.pattern.structural.proxy.db.DynamicDataSource">
+//		<property name="targetDataSources">
+//			<map key-type="java.lang.String">
+//				<entry value-ref="db0" key="db0"></entry>
+//				<entry value-ref="db1" key="db1"></entry>
+//			</map>
+//		</property>
+//		<property name="defaultTargetDataSource" ref="db0"></property>
+//	</bean>
+
+
+//	<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
+//		<property name="dataSource" ref="dataSource" />
+//	</bean>
+}