Przeglądaj źródła

feature #IAJD9H 期望liteflow 有自己的生命周期扩展

everywhere.z 7 miesięcy temu
rodzic
commit
566ebcf6e4

+ 8 - 8
liteflow-solon-plugin/src/main/java/com/yomahub/liteflow/solon/integration/XPluginImpl.java

@@ -49,16 +49,16 @@ public class XPluginImpl implements Plugin {
 		context.beanMake(LiteflowMainAutoConfiguration.class);
 
 		// 订阅生命周期实现类
-		context.subWrapsOfType(LifeCycle.class, bw -> {
-			LifeCycle lifeCycle = bw.raw();
-			LifeCycleHolder.addLifeCycle(lifeCycle);
-		});
+		context.subBeansOfType(LifeCycle.class, LifeCycleHolder::addLifeCycle);
 
 		// 订阅 NodeComponent 组件
-		context.subWrapsOfType(NodeComponent.class, bw -> {
-			NodeComponent node1 = bw.raw();
-			node1.setNodeId(bw.name());
-			FlowBus.addManagedNode(bw.name(), bw.raw());
+		context.lifecycle(()-> {
+			//扫描完成后,收集组件
+			context.beanForeach(bw -> {
+				if (bw.raw() instanceof NodeComponent) {
+					FlowBus.addManagedNode(bw.name(), bw.raw());
+				}
+			});
 		});
 
 		Set<Class<?>> liteflowMethodClassSet = new HashSet<>();