【Springboot入门- Actuator监控-线程池】

要在Spring Boot中监控线程池，可以利用Spring Boot Actuator的监控功能。

1. 添加Actuator依赖

• 在项目的pom.xml文件中添加spring-boot-starter-actuator依赖

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-actuator</artifactId>
</dependency>

2. 配置Actuator

• 在application.properties或application.yml中配置Actuator端点。

management:
  endpoints:
    web:
      exposure:
        include: "*"
      base-path: /actuator

3. 如果有安全框架需要匿名访问，这里我用的是shiro

• 在ShiroConfig.java中

@Bean("shiroFilter")
	public ShiroFilterFactoryBean shiroFilter(SecurityManager securityManager) {
		ShiroFilterFactoryBean shiroFilterFactoryBean = new ShiroFilterFactoryBean();
		shiroFilterFactoryBean.setSecurityManager(securityManager);
		// 拦截器
		Map<String, String> filterChainDefinitionMap = new LinkedHashMap<String, String>();
		filterChainDefinitionMap.put("/actuator/**", "anon");
		shiroFilterFactoryBean.setFilterChainDefinitionMap(filterChainDefinitionMap);
		return shiroFilterFactoryBean;
	}

4. 创建自定义线程池

@Configuration
public class ApiThreadPool {

    @Bean
    public ThreadPoolTaskExecutor apiThreadPoolTaskExecutor(){
        ThreadPoolTaskExecutor threadPoolTaskExecutor = new ThreadPoolTaskExecutor();
        // 线程池名字
        threadPoolTaskExecutor.setThreadNamePrefix("api线程池");
        // 核心线程数
        threadPoolTaskExecutor.setCorePoolSize(6);
        // 最大线程数
        threadPoolTaskExecutor.setMaxPoolSize(12);
        // 队列为20
        threadPoolTaskExecutor.setQueueCapacity(20);
        // 拒绝策略为 放到主线程去
        threadPoolTaskExecutor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
        return threadPoolTaskExecutor;

    }
}

5. 实现自定义端点

• 创建一个类并使用@Endpoint注解标记，实现自定义Actuator端点。

@Component
@Endpoint(id = "threadPool")
public class ThreadPoolEndpoint {
    @Autowired
    private ThreadPoolTaskExecutor apiThreadPoolTaskExecutor;

    @ReadOperation
    public Map<String, Object> threadPoolStatus() {
        ThreadPoolExecutor executor = apiThreadPoolTaskExecutor.getThreadPoolExecutor();
        Map<String, Object> status = new HashMap<>();
        status.put("corePoolSize", executor.getCorePoolSize());
        status.put("maximumPoolSize", executor.getMaximumPoolSize());
        status.put("activeCount", executor.getActiveCount());
        status.put("queueSize", executor.getQueue().size());
        status.put("completedTaskCount", executor.getCompletedTaskCount());
        return status;
    }
}

6. 业务代码

@Service
public class FiStepOneServiceImpl {

    @Autowired
    private List<FiBizOneService> fiBizOneServices;

    /**
     * 抽取数据逻辑
     * @param jsonObject
     */
    @Override
    public void execute(SyncBuilder<FiDto> jsonObject) {
        //对于单据要加锁
 
        try{
 
            //并发线程池
            CountDownLatch countDownLatch = new CountDownLatch(fiBizOneServices.size());
            for (FiBizOneService fiBizOneService : fiBizOneServices) {
                // 线程池跑任务
                apiThreadPoolTaskExecutor.execute(() -> {
                    fiBizOneService.execute(jsonObject.getData());
                    countDownLatch.countDown();
                });
            }
            // 等待所有线程执行完毕, 超时1小时
            countDownLatch.await(1, TimeUnit.HOURS);
        } catch (InterruptedException e) {
            e.printStackTrace();
        } finally {
            
        }

    }


}

9. 测试端点

• 启动应用程序并访问/actuator/threadPool端点来获取线程池的状态(端口和项目名根据实际情况)。

http://localhost:10023/actuator/threadPool

10. 结果截图

• 初始状态
  
• 进行中
  
  
• 结束