浏览代码

优化数据源日志提示

zhouhao 7 年之前
父节点
当前提交
16dc3511ec

+ 8 - 3
hsweb-datasource/hsweb-datasource-api/src/main/java/org/hswebframework/web/datasource/DefaultJdbcExecutor.java

@@ -16,14 +16,19 @@ import java.util.Objects;
  */
 @Transactional(rollbackFor = Throwable.class)
 public class DefaultJdbcExecutor extends AbstractJdbcSqlExecutor {
-    
+
+    protected String getDatasourceId() {
+        String id = DataSourceHolder.switcher().currentDataSourceId();
+        return id == null ? "default" : id;
+    }
+
     @Override
     public Connection getConnection() {
         DataSource dataSource = DataSourceHolder.currentDataSource().getNative();
         Connection connection = DataSourceUtils.getConnection(dataSource);
         boolean isConnectionTransactional = DataSourceUtils.isConnectionTransactional(connection, dataSource);
         if (logger.isDebugEnabled()) {
-            logger.debug("DataSource ({}) JDBC Connection [{}] will {}be managed by Spring", DataSourceHolder.switcher().currentDataSourceId(), connection, (isConnectionTransactional ? "" : "not "));
+            logger.debug("DataSource ({}) JDBC Connection [{}] will {}be managed by Spring", getDatasourceId(), connection, (isConnectionTransactional ? "" : "not "));
         }
         return connection;
     }
@@ -31,7 +36,7 @@ public class DefaultJdbcExecutor extends AbstractJdbcSqlExecutor {
     @Override
     public void releaseConnection(Connection connection) throws SQLException {
         if (logger.isDebugEnabled()) {
-            logger.debug("Releasing DataSource ({}) JDBC Connection [{}]", DataSourceHolder.switcher().currentDataSourceId(), connection);
+            logger.debug("Releasing DataSource ({}) JDBC Connection [{}]", getDatasourceId(), connection);
         }
         DataSourceUtils.releaseConnection(connection, DataSourceHolder.currentDataSource().getNative());
     }