当前位置：首页 > news >正文

JAVA-Spring Boot多线程

news 来源：原创 2025/5/8 23:48:37

方式	特点	适用场景
`ExecutorService`	Java 原生 API，灵活性高，适合需要完全控制线程池的场景。	通用多线程任务处理
`TaskExecutor`	Spring 提供的线程池封装，与 Spring 容器集成良好，推荐在 Spring Boot 项目中使用。	Spring 项目中的多线程任务处理
`@Async` + 自定义线程池	结合 Spring 的异步功能和线程池，适合需要异步执行的任务。	异步任务处理

在 Spring Boot 中，虽然没有直接等同于 C# 的 ThreadPool 类的功能，但可以通过以下几种方式实现类似线程池的功能：

1. 使用 Java 自带的 `ExecutorService`

Java 提供了 ExecutorService 接口和 Executors 工具类，可以创建和管理线程池。这种方式类似于 C# 的 ThreadPool。

示例代码：

Java

深色版本

import org.springframework.stereotype.Service;

import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

@Service
public class ThreadPoolService {

    // 创建固定大小的线程池
    private final ExecutorService executorService = Executors.newFixedThreadPool(5);

    public void executeTask(int taskId) {
        executorService.submit(() -> {
            System.out.println("Executing task " + taskId + " in thread: " + Thread.currentThread().getName());
            try {
                Thread.sleep(1000); // 模拟耗时操作
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            System.out.println("Finished task " + taskId);
        });
    }
}

调用代码：

Java

深色版本

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class TaskController {

    @Autowired
    private ThreadPoolService threadPoolService;

    @GetMapping("/run-tasks")
    public String runTasks() {
        for (int i = 1; i <= 10; i++) {
            threadPoolService.executeTask(i);
        }
        return "Tasks submitted!";
    }
}

运行结果：

访问 /run-tasks 后，您会看到类似以下输出：

深色版本

Executing task 1 in thread: pool-1-thread-1
Executing task 2 in thread: pool-1-thread-2
Finished task 1
...

2. 使用 Spring 的 `TaskExecutor`

Spring 提供了 TaskExecutor 接口，它是对 Java 线程池的封装，更符合 Spring 的编程风格。

配置线程池：

Java

深色版本

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

import java.util.concurrent.Executor;

@Configuration
public class ThreadPoolConfig {

    @Bean(name = "taskExecutor")
    public Executor taskExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(5); // 核心线程数
        executor.setMaxPoolSize(10); // 最大线程数
        executor.setQueueCapacity(25); // 队列容量
        executor.setThreadNamePrefix("SpringThreadPool-"); // 线程名称前缀
        executor.initialize();
        return executor;
    }
}

使用线程池：

Java

深色版本

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import org.springframework.stereotype.Service;

@Service
public class TaskService {

    @Autowired
    private ThreadPoolTaskExecutor taskExecutor;

    public void executeTasks() {
        for (int i = 1; i <= 10; i++) {
            final int taskId = i;
            taskExecutor.execute(() -> {
                System.out.println("Executing task " + taskId + " in thread: " + Thread.currentThread().getName());
                try {
                    Thread.sleep(1000); // 模拟耗时操作
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                System.out.println("Finished task " + taskId);
            });
        }
    }
}

调用代码：

Java

深色版本

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class TaskController {

    @Autowired
    private TaskService taskService;

    @GetMapping("/run-tasks")
    public String runTasks() {
        taskService.executeTasks();
        return "Tasks submitted!";
    }
}

运行结果：

访问 /run-tasks 后，输出类似于：

深色版本

Executing task 1 in thread: SpringThreadPool-1
Executing task 2 in thread: SpringThreadPool-2
Finished task 1
...

3. 使用 `@Async` 和自定义线程池

如果您希望结合 Spring 的异步功能和线程池，可以为 @Async 方法指定一个自定义线程池。

配置线程池：

Java

深色版本

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

import java.util.concurrent.Executor;

@Configuration
@EnableAsync
public class AsyncConfig {

    @Bean(name = "asyncTaskExecutor")
    public Executor asyncTaskExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(5);
        executor.setMaxPoolSize(10);
        executor.setQueueCapacity(25);
        executor.setThreadNamePrefix("AsyncThreadPool-");
        executor.initialize();
        return executor;
    }
}

定义异步方法：

Java

深色版本

import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;

@Service
public class AsyncTaskService {

    @Async("asyncTaskExecutor") // 使用指定的线程池
    public void executeTask(int taskId) {
        System.out.println("Executing task " + taskId + " in thread: " + Thread.currentThread().getName());
        try {
            Thread.sleep(1000); // 模拟耗时操作
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        System.out.println("Finished task " + taskId);
    }
}

调用代码：

Java

深色版本

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class TaskController {

    @Autowired
    private AsyncTaskService asyncTaskService;

    @GetMapping("/run-tasks")
    public String runTasks() {
        for (int i = 1; i <= 10; i++) {
            asyncTaskService.executeTask(i);
        }
        return "Tasks submitted!";
    }
}

运行结果：

访问 /run-tasks 后，输出类似于：

深色版本

Executing task 1 in thread: AsyncThreadPool-1
Executing task 2 in thread: AsyncThreadPool-2
Finished task 1
...

4. 总结与选择

方式	特点	适用场景
`ExecutorService`	Java 原生 API，灵活性高，适合需要完全控制线程池的场景。	通用多线程任务处理
`TaskExecutor`	Spring 提供的线程池封装，与 Spring 容器集成良好，推荐在 Spring Boot 项目中使用。	Spring 项目中的多线程任务处理
`@Async` + 自定义线程池	结合 Spring 的异步功能和线程池，适合需要异步执行的任务。	异步任务处理