Refactor ShadowDMLStatementDataSourceMappingsRetriever (#33556)

apache · Nov 5, 2024 · 0f6e2ef · 0f6e2ef
1 parent 121584e
commit 0f6e2ef
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...rdingsphere/shadow/route/retriever/dml/ShadowDMLStatementDataSourceMappingsRetriever.java b/...rdingsphere/shadow/route/retriever/dml/ShadowDMLStatementDataSourceMappingsRetriever.java
@@ -80,13 +80,13 @@ private ShadowColumnDataSourceMappingsRetriever createShadowDataSourceMappingsRe
         if (queryContext.getSqlStatementContext() instanceof SelectStatementContext) {
             return new ShadowSelectStatementDataSourceMappingsRetriever((SelectStatementContext) queryContext.getSqlStatementContext(), queryContext.getParameters(), tableAliasAndNameMappings);
         }
-        throw new UnsupportedOperationException(String.format("unsupported SQL statement context `%s`", queryContext.getSqlStatementContext().getClass().getName()));
+        return null;
     }
 
     @Override
     public Map<String, String> retrieve(final ShadowRule rule) {
         Collection<String> shadowTables = rule.filterShadowTables(tableAliasAndNameMappings.values());
         Map<String, String> result = tableHintDataSourceMappingsRetriever.retrieve(rule, shadowTables);
-        return result.isEmpty() ? shadowColumnDataSourceMappingsRetriever.retrieve(rule, shadowTables) : result;
+        return result.isEmpty() && null != shadowColumnDataSourceMappingsRetriever ? shadowColumnDataSourceMappingsRetriever.retrieve(rule, shadowTables) : result;
     }
 }