Refactor UnregisterStorageUnitBackendHandler

apache · Sep 29, 2023 · 8482fa8 · 8482fa8
1 parent 629b0bb
commit 8482fa8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...e/proxy/backend/handler/distsql/rdl/storage/unit/UnregisterStorageUnitBackendHandler.java b/...e/proxy/backend/handler/distsql/rdl/storage/unit/UnregisterStorageUnitBackendHandler.java
@@ -71,8 +71,8 @@ public void checkSQLStatement(final String databaseName, final UnregisterStorage
     }
 
     private void checkExisted(final String databaseName, final Collection<String> storageUnitNames) {
-        Map<String, NewStorageUnitMetaData> storageUnits = ProxyContext.getInstance().getDatabase(databaseName).getResourceMetaData().getStorageUnitMetaData().getMetaDataMap();
-        Collection<String> notExistedStorageUnits = storageUnitNames.stream().filter(each -> !storageUnits.containsKey(each)).collect(Collectors.toList());
+        Map<String, NewStorageUnitMetaData> metaDataMap = ProxyContext.getInstance().getDatabase(databaseName).getResourceMetaData().getStorageUnitMetaData().getMetaDataMap();
+        Collection<String> notExistedStorageUnits = storageUnitNames.stream().filter(each -> !metaDataMap.containsKey(each)).collect(Collectors.toList());
         ShardingSpherePreconditions.checkState(notExistedStorageUnits.isEmpty(), () -> new MissingRequiredStorageUnitsException(databaseName, notExistedStorageUnits));
     }