[server] Use background mode for delete partition assignment in zk

Author: LiebingYu

fluss-server/src/main/java/org/apache/fluss/server/zk/ZooKeeperClient.java

@@ -222,8 +222,27 @@ public void deleteTableAssignment(long tableId) throws Exception {
 
     public void deletePartitionAssignment(long partitionId) throws Exception {
         String path = PartitionIdZNode.path(partitionId);
-        zkClient.delete().deletingChildrenIfNeeded().forPath(path);
-        LOG.info("Deleted table assignment for partition id {}.", partitionId);
+        // delete partition assignment ZNode will recursively delete all the children which may
+        // block
+        // for a long time, so we use background mode here.
+        zkClient.delete()
+                .deletingChildrenIfNeeded()
+                .inBackground(
+                        (client, event) -> {
+                            if (event.getResultCode() == KeeperException.Code.OK.intValue()) {
+                                LOG.info(
+                                        "Deleted table assignment for partition id {}.",
+                                        partitionId);
+                            } else {
+                                KeeperException.Code code =
+                                        KeeperException.Code.get(event.getResultCode());
+                                LOG.error(
+                                        "Failed to delete partition assignment for partition id {}. Error message: {}",
+                                        partitionId,
+                                        code.toString());
+                            }
+                        })
+                .forPath(path);
     }
 
     // --------------------------------------------------------------------------------------------