initialize all of the statements which is marked datasourceId when configuration set supportDynamicRoutingDataSource.

The databaseIdProvider provide current databaseid for statement match.

Author: Terrdi

src/main/java/org/apache/ibatis/session/Configuration.java

@@ -15,6 +15,7 @@
  */
 package org.apache.ibatis.session;
 
+import java.sql.SQLException;
 import java.util.Arrays;
 import java.util.Collection;
 import java.util.HashMap;
@@ -23,6 +24,7 @@
 import java.util.LinkedList;
 import java.util.List;
 import java.util.Map;
+import java.util.Objects;
 import java.util.Properties;
 import java.util.Set;
 import java.util.concurrent.ConcurrentHashMap;
@@ -831,7 +833,11 @@ public boolean hasParameterMap(String id) {
   }
 
   public void addMappedStatement(MappedStatement ms) {
-    mappedStatements.put(ms.getId(), ms);
+    String id = ms.getId();
+    if (this.getSupportDynamicRoutingDataSource() && Objects.nonNull(ms.getDatabaseId())) {
+      id = id + "#" + ms.getDatabaseId();
+    }
+    mappedStatements.put(id, ms);
   }
 
   public Collection<String> getMappedStatementNames() {
@@ -920,7 +926,22 @@ public MappedStatement getMappedStatement(String id, boolean validateIncompleteS
     if (validateIncompleteStatements) {
       buildAllStatements();
     }
-    return mappedStatements.get(id);
+    MappedStatement statement = mappedStatements.get(this.getMappedStatementId(id));
+    if (this.getSupportDynamicRoutingDataSource() && Objects.isNull(statement)) {
+      statement = mappedStatements.get(id);
+    }
+    return statement;
+  }
+
+  protected String getMappedStatementId(String id) {
+    try {
+      String databaseId = this.getCurrentDatabaseId();
+      if (this.getSupportDynamicRoutingDataSource() && Objects.nonNull(databaseId)) {
+        return id + "#" + databaseId;
+      }
+    } catch (SQLException ignore) {
+    }
+    return id;
   }
 
   public Map<String, XNode> getSqlFragments() {
@@ -959,7 +980,8 @@ public boolean hasStatement(String statementName, boolean validateIncompleteStat
     if (validateIncompleteStatements) {
       buildAllStatements();
     }
-    return mappedStatements.containsKey(statementName);
+    return mappedStatements.containsKey(this.getMappedStatementId(statementName))
+        || this.getSupportDynamicRoutingDataSource() && mappedStatements.containsKey(statementName);
   }
 
   public void addCacheRef(String namespace, String referencedNamespace) {