diff --git a/liteflow-spring-boot-starter/src/main/java/com/yomahub/liteflow/springboot/config/LiteflowExecutorAutoConfiguration.java b/liteflow-spring-boot-starter/src/main/java/com/yomahub/liteflow/springboot/config/LiteflowExecutorAutoConfiguration.java index 75879aa0e..f87967171 100644 --- a/liteflow-spring-boot-starter/src/main/java/com/yomahub/liteflow/springboot/config/LiteflowExecutorAutoConfiguration.java +++ b/liteflow-spring-boot-starter/src/main/java/com/yomahub/liteflow/springboot/config/LiteflowExecutorAutoConfiguration.java @@ -3,11 +3,13 @@ package com.yomahub.liteflow.springboot.config; import com.yomahub.liteflow.property.LiteflowConfig; import com.yomahub.liteflow.util.ExecutorHelper; import com.yomahub.liteflow.util.LiteFlowExecutorPoolShutdown; +import com.yomahub.liteflow.util.SpringAware; import org.springframework.boot.autoconfigure.AutoConfigureAfter; import org.springframework.boot.autoconfigure.condition.ConditionalOnBean; import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; +import org.springframework.context.annotation.Import; import java.util.concurrent.ExecutorService; @@ -20,6 +22,7 @@ import java.util.concurrent.ExecutorService; @AutoConfigureAfter({LiteflowPropertyAutoConfiguration.class}) @ConditionalOnProperty(prefix = "liteflow", name = "enable", havingValue = "true") @ConditionalOnBean(LiteflowConfig.class) +@Import(SpringAware.class) public class LiteflowExecutorAutoConfiguration { @Bean("whenExecutors") diff --git a/liteflow-testcase-springboot/src/test/java/com/yomahub/liteflow/test/config/LiteflowConfigSpringbootTest.java b/liteflow-testcase-springboot/src/test/java/com/yomahub/liteflow/test/config/LiteflowConfigSpringbootTest.java index bd80d2f7c..f79c99382 100644 --- a/liteflow-testcase-springboot/src/test/java/com/yomahub/liteflow/test/config/LiteflowConfigSpringbootTest.java +++ b/liteflow-testcase-springboot/src/test/java/com/yomahub/liteflow/test/config/LiteflowConfigSpringbootTest.java @@ -46,7 +46,7 @@ public class LiteflowConfigSpringbootTest extends BaseTest { Assert.assertEquals(300000L, config.getDelay().longValue()); Assert.assertEquals(300000L, config.getPeriod().longValue()); Assert.assertFalse(config.getEnableLog()); - Assert.assertEquals(4, config.getWhenMaxWorkers().longValue()); + Assert.assertEquals(16, config.getWhenMaxWorkers().longValue()); Assert.assertEquals(512, config.getWhenQueueLimit().longValue()); Assert.assertEquals(true, config.isParseOnStart()); }