Spring boot 创建一个只放一个线程的的线程池

原创

mob64ca12ee2ba5 2024-10-11 07:08:41 ©著作权

文章标签 线程池 spring 单线程 文章分类 spring boot 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12ee2ba5的原创作品，请联系作者获取转载授权，否则将追究法律责任

Spring Boot 创建一个只放一个线程的线程池

在现代软件开发中，线程池是一种重要技术，它允许我们有效地管理和使用线程。Spring Boot作为一种流行的Java开发框架，也提供了非常便捷的线程池管理功能。本文将介绍如何在Spring Boot中创建一个只放一个线程的线程池，并通过代码示例进行演示。

什么是线程池？

线程池是预先创建的线程集合，能够提高多线程应用程序的效率。在需要进行大量短期任务时，创建和销毁线程会消耗资源，线程池避免了这种开销。线程池的基本思想是重用现有线程，而不是频繁创建和销毁线程。

Spring Boot 中的线程池

Spring Boot 提供了一些内置的线程池实现，如 ThreadPoolTaskExecutor，我们可以使用这个类来创建我们的线程池。接下来，我们将创建一个只包含一个线程的线程池，这对于处理一些单线程任务是很有用的。

创建单线程线程池的步骤

下面将通过几个步骤来实现一个只放一个线程的线程池：

依赖配置
如果使用Spring Boot，确保你的pom.xml中包含了Spring Boot的相关依赖。以下是基本的依赖项：
```
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter</artifactId>
</dependency>
```

配置线程池
在你的Spring Boot应用程序中，可以通过配置类来定义线程池。以下是一个简单的配置示例：

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

@Configuration
public class ThreadPoolConfig {

    @Bean
    public ThreadPoolTaskExecutor threadPoolTaskExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(1); // 设置核心线程数为1
        executor.setMaxPoolSize(1);  // 设置最大线程数为1
        executor.setQueueCapacity(100); // 设置队列容量
        executor.setThreadNamePrefix("single-thread-"); // 设置线程名前缀
        executor.initialize(); // 初始化线程池
        return executor;
    }
}

在这个配置中，我们设置了核心线程数和最大线程数均为1，这样线程池中只会有一个线程在工作。

使用线程池
接下来，我们将演示如何使用上面创建的线程池来执行任务：

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

@Service
public class TaskService {

    private final ThreadPoolTaskExecutor taskExecutor;

    @Autowired
    public TaskService(ThreadPoolTaskExecutor taskExecutor) {
        this.taskExecutor = taskExecutor;
    }

    @Async
    public void executeTask(String taskName) {
        System.out.println("Task " + taskName + " is starting.");
        try {
            Thread.sleep(2000); // 模拟长时间运行的任务
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
        System.out.println("Task " + taskName + " is completed.");
    }
}

在这个示例中，当调用executeTask方法时，任务将会被提交到我们的线程池中。由于线程池只允许一个线程处理，因此任务会一个接一个地执行。

调用任务
最后，我们可以在控制器中调用这个任务来进行测试：

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 {

    private final TaskService taskService;

    @Autowired
    public TaskController(TaskService taskService) {
        this.taskService = taskService;
    }

    @GetMapping("/execute")
    public String execute() {
        taskService.executeTask("Task 1");
        taskService.executeTask("Task 2");
        return "Tasks are executing...";
    }
}

通过访问 /execute 端点，可以启动两个任务，验证它们是一个接一个地执行的。