From 4592427ddec559ff9edadcc7d16f9e136067b1e0 Mon Sep 17 00:00:00 2001
From: "everywhere.z" <weenyc31@163.com>
Date: Mon, 28 Nov 2022 23:36:43 +0800
Subject: [PATCH] =?UTF-8?q?enhancement=20#I633VH=20=E5=BB=BA=E8=AE=AEFlowB?=
 =?UTF-8?q?us=E6=8F=90=E4=BE=9B=E6=89=B9=E9=87=8F=E7=A7=BB=E9=99=A4?=
 =?UTF-8?q?=E5=AD=90=E9=93=BE=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../src/main/java/com/yomahub/liteflow/flow/FlowBus.java  | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/liteflow-core/src/main/java/com/yomahub/liteflow/flow/FlowBus.java b/liteflow-core/src/main/java/com/yomahub/liteflow/flow/FlowBus.java
index 920b45879..230a7ea35 100644
--- a/liteflow-core/src/main/java/com/yomahub/liteflow/flow/FlowBus.java
+++ b/liteflow-core/src/main/java/com/yomahub/liteflow/flow/FlowBus.java
@@ -35,8 +35,10 @@ import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.List;
 import java.util.Map;
+import java.util.function.Consumer;
 import java.util.stream.Collectors;
 
 /**
@@ -67,7 +69,7 @@ public class FlowBus {
 
     //这个方法主要用于第二阶段的替换chain
     public static void addChain(Chain chain) {
-        chainMap.put(chain.getChainName(), chain);
+        chainMap.put(chain.getChainId(), chain);
     }
 
     public static boolean containChain(String chainId) {
@@ -333,4 +335,8 @@ public class FlowBus {
             return false;
         }
     }
+
+    public static void removeChain(String... chainIds){
+        Arrays.stream(chainIds).forEach(FlowBus::removeChain);
+    }
 }