Nacos 實現動態化線程池，真香！

在後臺開發中，會經常用到線程池技術，對於線程池核心參數的配置很大程度上依靠經驗。然而，由於系統運行過程中存在的不確定性，我們很難一勞永逸地規劃一個合理的線程池參數。在對線程池配置參數進行調整時，一般需要對服務進行重啟，這樣修改的成本就會偏高。一種解決辦法就是，將線程池的配置放到平臺側，運行開發同學 ...

在後臺開發中，會經常用到線程池技術，對於線程池核心參數的配置很大程度上依靠經驗。然而，由於系統運行過程中存在的不確定性，我們很難一勞永逸地規劃一個合理的線程池參數。

在對線程池配置參數進行調整時，一般需要對服務進行重啟，這樣修改的成本就會偏高。一種解決辦法就是，將線程池的配置放到平臺側，運行開發同學根據系統運行情況對核心參數進行動態配置。

本文以Nacos作為服務配置中心，以修改線程池核心線程數、最大線程數為例，實現一個簡單的動態化線程池。

代碼實現

推薦一個開源免費的 Spring Boot 實戰項目：

https://github.com/javastacks/spring-boot-best-practice

1.依賴

<dependency>
    <groupId>com.alibaba.cloud</groupId>
    <artifactId>spring-cloud-starter-alibaba-nacos-discovery</artifactId>
    <version>2021.1</version>
</dependency>
<dependency>
    <groupId>com.alibaba.cloud</groupId>
    <artifactId>spring-cloud-starter-alibaba-nacos-config</artifactId>
    <version>2021.1</version>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter</artifactId>
</dependency>

2.配置yml文件

Spring Boot 基礎就不介紹了，推薦看這個實戰項目：

https://github.com/javastacks/spring-boot-best-practice

bootstrap.yml：

server:
  port: 8010
  # 應用名稱（nacos會將該名稱當做服務名稱）
spring:
  application:
    name: order-service
  cloud:
    nacos:
      discovery:
        namespace: public
        server-addr: 192.168.174.129:8848
      config:
        server-addr: 192.168.174.129:8848
        file-extension: yml

application.yml：

spring:
  profiles:
    active: dev

為什麼要配置兩個yml文件？

springboot中配置文件的載入是存在優先順序順序的，bootstrap優先順序高於application。

nacos在項目初始化時，要保證先從配置中心進行配置拉取，拉取配置之後才能保證項目的正常啟動。

3.nacos配置

登錄到nacos管理頁面，新建配置，如下圖所示：

註意Data ID的命名格式為，${spring.application.name}-${spring.profile.active}.${spring.cloud.nacos.config.file-extension} ，在本文中，Data ID的名字就是order-service-dev.yml。

這裡我們只配置了兩個參數，核心線程數量和最大線程數。

4.線程池配置和nacos配置變更監聽

@RefreshScope
@Configuration
public class DynamicThreadPool implements InitializingBean {
    @Value("${core.size}")
    private String coreSize;

    @Value("${max.size}")
    private String maxSize;

    private static ThreadPoolExecutor threadPoolExecutor;

    @Autowired
    private NacosConfigManager nacosConfigManager;

    @Autowired
    private NacosConfigProperties nacosConfigProperties;

    @Override
    public void afterPropertiesSet() throws Exception {
        //按照nacos配置初始化線程池
        threadPoolExecutor = new ThreadPoolExecutor(Integer.parseInt(coreSize), Integer.parseInt(maxSize), 10L, TimeUnit.SECONDS,
                new LinkedBlockingQueue<>(10),
                new ThreadFactoryBuilder().setNameFormat("c_t_%d").build(),
                new RejectedExecutionHandler() {
                    @Override
                    public void rejectedExecution(Runnable r, ThreadPoolExecutor executor) {
                        System.out.println("rejected!");
                    }
                });

        //nacos配置變更監聽
        nacosConfigManager.getConfigService().addListener("order-service-dev.yml", nacosConfigProperties.getGroup(),
                new Listener() {
                    @Override
                    public Executor getExecutor() {
                        return null;
                    }

                    @Override
                    public void receiveConfigInfo(String configInfo) {
                        //配置變更，修改線程池配置
                        System.out.println(configInfo);
                        changeThreadPoolConfig(Integer.parseInt(coreSize), Integer.parseInt(maxSize));
                    }
                });
    }

    /**
     * 列印當前線程池的狀態
     */
    public String printThreadPoolStatus() {
        return String.format("core_size:%s,thread_current_size:%s;" +
                        "thread_max_size:%s;queue_current_size:%s,total_task_count:%s", threadPoolExecutor.getCorePoolSize(),
                threadPoolExecutor.getActiveCount(), threadPoolExecutor.getMaximumPoolSize(), threadPoolExecutor.getQueue().size(),
                threadPoolExecutor.getTaskCount());
    }

    /**
     * 給線程池增加任務
     *
     * @param count
     */
    public void dynamicThreadPoolAddTask(int count) {
        for (int i = 0; i < count; i++) {
            int finalI = i;
            threadPoolExecutor.execute(new Runnable() {
                @Override
                public void run() {
                    try {
                        System.out.println(finalI);
                        Thread.sleep(10000);
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }
                }
            });
        }
    }

    /**
     * 修改線程池核心參數
     *
     * @param coreSize
     * @param maxSize
     */
    private void changeThreadPoolConfig(int coreSize, int maxSize) {
        threadPoolExecutor.setCorePoolSize(coreSize);
        threadPoolExecutor.setMaximumPoolSize(maxSize);
    }
}

這個代碼就是實現動態線程池和核心了，需要說明的是：

@RefreshScope：這個註解用來支持nacos的動態刷新功能；
@Value("${max.size}")，@Value("${core.size}")：這兩個註解用來讀取我們上一步在nacos配置的具體信息；同時，nacos配置變更時，能夠實時讀取到變更後的內容
nacosConfigManager.getConfigService().addListener：配置監聽，nacos配置變更時實時修改線程池的配置。

5.controller

為了觀察線程池動態變更的效果，增加Controller類。

@RestController
@RequestMapping("/threadpool")
public class ThreadPoolController {

    @Autowired
    private DynamicThreadPool dynamicThreadPool;

    /**
     * 列印當前線程池的狀態
     */
    @GetMapping("/print")
    public String printThreadPoolStatus() {
        return dynamicThreadPool.printThreadPoolStatus();
    }

    /**
     * 給線程池增加任務
     *
     * @param count
     */
    @GetMapping("/add")
    public String dynamicThreadPoolAddTask(int count) {
        dynamicThreadPool.dynamicThreadPoolAddTask(count);
        return String.valueOf(count);
    }
}