diff --git a/liteflow-spring-boot-starter/src/main/java/com/yomahub/liteflow/springboot/LiteflowExecutorInit.java b/liteflow-spring-boot-starter/src/main/java/com/yomahub/liteflow/springboot/LiteflowExecutorInit.java index 2b341265a..7a7fc57ee 100644 --- a/liteflow-spring-boot-starter/src/main/java/com/yomahub/liteflow/springboot/LiteflowExecutorInit.java +++ b/liteflow-spring-boot-starter/src/main/java/com/yomahub/liteflow/springboot/LiteflowExecutorInit.java @@ -2,24 +2,22 @@ package com.yomahub.liteflow.springboot; import com.yomahub.liteflow.core.FlowExecutor; import com.yomahub.liteflow.flow.FlowBus; -import org.springframework.beans.factory.InitializingBean; -import org.springframework.boot.CommandLineRunner; +import org.springframework.beans.factory.SmartInitializingSingleton; /** * 执行器初始化类 主要用于在启动时执行执行器的初始化方法,避免在运行执行器时第一次初始化而耗费时间 * * @author Bryan.Zhang */ -public class LiteflowExecutorInit implements CommandLineRunner { +public class LiteflowExecutorInit implements SmartInitializingSingleton { private final FlowExecutor flowExecutor; public LiteflowExecutorInit(FlowExecutor flowExecutor) { this.flowExecutor = flowExecutor; } - @Override - public void run(String... args) throws Exception { + public void afterSingletonsInstantiated() { flowExecutor.init(true); FlowBus.needInit(); }