Fix index out of bounds exception for migration without unique keys (#…

…34127)
apache · Dec 23, 2024 · 44c49a1 · 44c49a1
1 parent 80e148e
commit 44c49a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...pipeline/core/consistencycheck/table/calculator/RecordSingleTableInventoryCalculator.java b/...pipeline/core/consistencycheck/table/calculator/RecordSingleTableInventoryCalculator.java
@@ -148,7 +148,7 @@ private void fulfillCalculationContext(final CalculationContext calculationConte
     }
 
     private String getQuerySQL(final SingleTableInventoryCalculateParameter param) {
-        ShardingSpherePreconditions.checkNotNull(param.getFirstUniqueKey(),
+        ShardingSpherePreconditions.checkState(param.getUniqueKeys() != null && !param.getUniqueKeys().isEmpty() && null != param.getFirstUniqueKey(),
                 () -> new UnsupportedOperationException("Record inventory calculator does not support table without unique key and primary key now."));
         PipelineDataConsistencyCalculateSQLBuilder pipelineSQLBuilder = new PipelineDataConsistencyCalculateSQLBuilder(param.getDatabaseType());
         Collection<String> columnNames = param.getColumnNames().isEmpty() ? Collections.singleton("*") : param.getColumnNames();