支持多数据源切换功能

Author: xlorne

tx-plugins-db/src/main/java/com/codingapi/tx/datasource/bean/LCNDataSourceLocal.java

@@ -0,0 +1,29 @@
+package com.codingapi.tx.datasource.bean;
+
+/**
+ * create by lorne on 2017/12/7
+ */
+public class LCNDataSourceLocal {
+
+    private final static ThreadLocal<LCNDataSourceLocal> currentLocal = new ThreadLocal<LCNDataSourceLocal>();
+
+    public static LCNDataSourceLocal current() {
+        return currentLocal.get();
+    }
+
+    public static void setCurrent(LCNDataSourceLocal current) {
+        currentLocal.set(current);
+    }
+
+    private String key;
+
+
+    public String getKey() {
+        return key;
+    }
+
+    public void setKey(String key) {
+        this.key = key;
+    }
+
+}

tx-plugins-db/src/main/java/com/codingapi/tx/datasource/relational/LCNConnection.java

@@ -0,0 +1,11 @@
+package com.codingapi.tx.datasource.relational;
+
+import com.codingapi.tx.datasource.ILCNResource;
+
+import java.sql.Connection;
+
+/**
+ * create by lorne on 2017/12/7
+ */
+public interface LCNConnection extends  Connection,ILCNResource<Connection> {
+}

tx-plugins-db/src/main/java/com/codingapi/tx/datasource/relational/LCNDBConnection.java

@@ -22,7 +22,7 @@
  * create by lorne on 2017/7/29
  */
 
-public class LCNDBConnection extends AbstractTransactionThread implements Connection,ILCNResource<Connection> {
+public class LCNDBConnection extends AbstractTransactionThread implements LCNConnection {
 
 
     private Logger logger = LoggerFactory.getLogger(LCNDBConnection.class);
@@ -178,7 +178,9 @@ public TxTask getWaitTask() {
 
     @Override
     public void setAutoCommit(boolean autoCommit) throws SQLException {
-        connection.setAutoCommit(false);
+        if(connection!=null) {
+            connection.setAutoCommit(false);
+        }
     }
 
     @Override

tx-plugins-db/src/main/java/com/codingapi/tx/datasource/relational/LCNDynamicTransactionDataSource.java

@@ -0,0 +1,80 @@
+package com.codingapi.tx.datasource.relational;
+
+import com.codingapi.tx.aop.bean.TxTransactionLocal;
+import com.codingapi.tx.datasource.bean.LCNDataSourceLocal;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import javax.sql.DataSource;
+import java.sql.Connection;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
+
+/**
+ * 关系型数据库动态代理连接池对象
+ * create by lorne on 2017/7/29
+ */
+
+public class LCNDynamicTransactionDataSource extends LCNTransactionDataSource {
+
+
+    private Logger logger = LoggerFactory.getLogger(LCNDynamicTransactionDataSource.class);
+
+
+    private Map<String,DataSource> dataSourceMap = new ConcurrentHashMap<>();
+
+
+    private String getNowDataSourceKey(){
+        if(LCNDataSourceLocal.current()==null){
+            return "default";
+        }
+        return LCNDataSourceLocal.current().getKey();
+    }
+
+    @Override
+    public boolean hasGroup(String group) {
+        return super.hasGroup(getNowDataSourceKey()+group);
+    }
+
+    @Override
+    protected Connection createLcnConnection(Connection connection, TxTransactionLocal txTransactionLocal) {
+        nowCount++;
+        if(txTransactionLocal.isHasStart()){
+            LCNStartConnection lcnStartConnection = new LCNStartConnection(connection,subNowCount);
+            logger.info("get new start connection - > "+txTransactionLocal.getGroupId());
+            pools.put(getNowDataSourceKey()+txTransactionLocal.getGroupId(), lcnStartConnection);
+            txTransactionLocal.setHasConnection(true);
+            return lcnStartConnection;
+        }else {
+            LCNDBConnection lcn = new LCNDBConnection(connection, dataSourceService, subNowCount);
+            logger.info("get new connection ->" + txTransactionLocal.getGroupId());
+            pools.put(getNowDataSourceKey()+txTransactionLocal.getGroupId(), lcn);
+            txTransactionLocal.setHasConnection(true);
+            return lcn;
+        }
+    }
+
+    public void addDataSource(String key, DataSource dataSource){
+        dataSourceMap.put(key,dataSource);
+        logger.info("add datasource of "+key);
+    }
+
+    public void setDataSource(DataSource dataSource) {
+        super.setDataSource(dataSource);
+        addDataSource("default",dataSource);
+        logger.info("load default datasource.");
+    }
+
+    @Override
+    protected DataSource getDataSource() {
+        logger.info("getDataSource--->");
+        if(LCNDataSourceLocal.current()==null){
+            return super.getDataSource();
+        }else{
+            String key = LCNDataSourceLocal.current().getKey();
+            logger.info("get datasource of "+key);
+            return dataSourceMap.get(key);
+        }
+    }
+}

tx-plugins-db/src/main/java/com/codingapi/tx/datasource/relational/LCNStartConnection.java

@@ -17,7 +17,7 @@
 /**
  * create by lorne on 2017/12/1
  */
-public class LCNStartConnection extends AbstractTransactionThread implements Connection,ILCNResource<Connection>{
+public class LCNStartConnection extends AbstractTransactionThread implements LCNConnection{
 
     private Logger logger = LoggerFactory.getLogger(LCNStartConnection.class);
 
@@ -151,7 +151,9 @@ protected void closeConnection() throws SQLException{
 
     @Override
     public void setAutoCommit(boolean autoCommit) throws SQLException {
-        connection.setAutoCommit(false);
+        if(connection!=null) {
+            connection.setAutoCommit(false);
+        }
     }
 
 

tx-plugins-db/src/main/java/com/codingapi/tx/datasource/relational/LCNTransactionDataSource.java

@@ -23,9 +23,12 @@ public class LCNTransactionDataSource extends AbstractResourceProxy<Connection,L
     private org.slf4j.Logger logger = LoggerFactory.getLogger(LCNTransactionDataSource.class);
 
 
-    private DataSource dataSource;
+    protected DataSource dataSource;
 
 
+    protected DataSource getDataSource() {
+        return dataSource;
+    }
 
     public void setDataSource(DataSource dataSource) {
         this.dataSource = dataSource;
@@ -70,7 +73,7 @@ public Connection getConnection() throws SQLException {
 
         Connection connection =(Connection)loadConnection();
         if(connection==null) {
-             connection = initLCNConnection(dataSource.getConnection());
+             connection = initLCNConnection(getDataSource().getConnection());
             if(connection==null){
                 throw new SQLException("connection was overload");
             }
@@ -87,7 +90,7 @@ public Connection getConnection(String username, String password) throws SQLExce
 
         Connection connection = (Connection)loadConnection();
         if(connection==null) {
-            connection =  initLCNConnection(dataSource.getConnection(username, password));
+            connection =  initLCNConnection(getDataSource().getConnection(username, password));
             if(connection==null){
                 throw new SQLException("connection was overload");
             }
@@ -102,36 +105,36 @@ public Connection getConnection(String username, String password) throws SQLExce
 
     @Override
     public PrintWriter getLogWriter() throws SQLException {
-        return dataSource.getLogWriter();
+        return getDataSource().getLogWriter();
     }
 
     @Override
     public void setLogWriter(PrintWriter out) throws SQLException {
-        dataSource.setLogWriter(out);
+        getDataSource().setLogWriter(out);
     }
 
     @Override
     public void setLoginTimeout(int seconds) throws SQLException {
-        dataSource.setLoginTimeout(seconds);
+        getDataSource().setLoginTimeout(seconds);
     }
 
     @Override
     public int getLoginTimeout() throws SQLException {
-        return dataSource.getLoginTimeout();
+        return getDataSource().getLoginTimeout();
     }
 
     @Override
     public Logger getParentLogger() throws SQLFeatureNotSupportedException {
-        return dataSource.getParentLogger();
+        return getDataSource().getParentLogger();
     }
 
     @Override
     public <T> T unwrap(Class<T> iface) throws SQLException {
-        return dataSource.unwrap(iface);
+        return getDataSource().unwrap(iface);
     }
 
     @Override
     public boolean isWrapperFor(Class<?> iface) throws SQLException {
-        return dataSource.isWrapperFor(iface);
+        return getDataSource().isWrapperFor(iface);
     }
 }