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

Spring Boot使用线程池创建多线程

Spring Boot 2 中,可以使用 @Autowired 注入 线程池ThreadPoolTaskExecutorExecutorService),从而管理线程的创建和执行。以下是使用 @Autowired 方式注入线程池的完整示例。


1. 通过 @Autowired 注入 ThreadPoolTaskExecutor

步骤 1:配置线程池

创建 ThreadPoolTaskExecutor@Bean 配置:

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 = "customTaskExecutor")
    public Executor taskExecutor() {
        ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
        executor.setCorePoolSize(5);  // 核心线程数
        executor.setMaxPoolSize(10);  // 最大线程数
        executor.setQueueCapacity(25); // 任务队列容量
        executor.setThreadNamePrefix("Async-Executor-"); // 线程名前缀
        executor.initialize();
        return executor;
    }
}

步骤 2:使用 @Autowired 注入线程池

Service 层,通过 @Autowired 注入 ThreadPoolTaskExecutor 并执行任务:

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import org.springframework.stereotype.Service;

import java.util.concurrent.Future;

@Service
public class AsyncTaskService {

    private static final Logger logger = LoggerFactory.getLogger(AsyncTaskService.class);

      @Autowired
    @Qualifier("customTaskExecutor") // 通过 @Qualifier 指定 Bean 名称
    private ThreadPoolTaskExecutor customTaskExecutor;

    // 提交异步任务
    public void runAsyncTask() {
        customTaskExecutor.execute(() -> {
            logger.info("异步任务执行,线程名:{}", Thread.currentThread().getName());
            try {
                Thread.sleep(2000); // 模拟耗时任务
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            logger.info("异步任务完成,线程名:{}", Thread.currentThread().getName());
        });
    }

    // 提交带返回值的异步任务
    public Future<String> runAsyncTaskWithResult() {
        return customTaskExecutor.submit(() -> {
            logger.info("执行带返回值的异步任务,线程名:{}", Thread.currentThread().getName());
            Thread.sleep(2000);
            return "任务完成";
        });
    }
}

步骤 3:在 Controller 中调用

Controller 层,通过 @Autowired 调用 AsyncTaskService

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.concurrent.Future;

@RestController
@RequestMapping("/task")
public class AsyncTaskController {

    @Autowired
    private AsyncTaskService asyncTaskService;

    @GetMapping("/run")
    public String runTask() {
        asyncTaskService.runAsyncTask();
        return "任务已提交";
    }

    @GetMapping("/runWithResult")
    public String runTaskWithResult() throws Exception {
        Future<String> result = asyncTaskService.runAsyncTaskWithResult();
        return "任务结果:" + result.get();
    }
}

2. 通过 @Autowired 注入 ThreadPoolTaskScheduler(适用于定时任务)

步骤 1:配置 ThreadPoolTaskScheduler

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

@Configuration
public class TaskSchedulerConfig {

    @Bean
    public ThreadPoolTaskScheduler taskScheduler() {
        ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();
        scheduler.setPoolSize(5);  // 线程池大小
        scheduler.setThreadNamePrefix("Scheduled-Task-");
        scheduler.initialize();
        return scheduler;
    }
}

步骤 2:在 Service 中使用 @Autowired 注入

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
import org.springframework.stereotype.Service;

import java.util.concurrent.ScheduledFuture;

@Service
public class ScheduledTaskService {

    private static final Logger logger = LoggerFactory.getLogger(ScheduledTaskService.class);

    @Autowired
    private ThreadPoolTaskScheduler taskScheduler;

    public void scheduleTask() {
        ScheduledFuture<?> future = taskScheduler.scheduleAtFixedRate(() -> {
            logger.info("执行定时任务,线程名:{}", Thread.currentThread().getName());
        }, 5000);
    }
}

步骤 3:在 Controller 中调用

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequestMapping("/schedule")
public class ScheduleTaskController {

    @Autowired
    private ScheduledTaskService scheduledTaskService;

    @GetMapping("/start")
    public String startScheduledTask() {
        scheduledTaskService.scheduleTask();
        return "定时任务已启动";
    }
}

3. 通过 @Autowired 注入 ExecutorService

如果你更喜欢 Java 原生的 ExecutorService,可以使用 @Bean 配置:

步骤 1:定义 ExecutorService 线程池

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

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

@Configuration
public class ExecutorServiceConfig {

    @Bean
    public ExecutorService fixedThreadPool() {
        return Executors.newFixedThreadPool(5);
    }
}

步骤 2:在 Service 中注入 ExecutorService

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Service;

import java.util.concurrent.ExecutorService;

@Service
public class ExecutorServiceTask {

    private static final Logger logger = LoggerFactory.getLogger(ExecutorServiceTask.class);

    @Autowired
    private ExecutorService executorService;

    public void executeTask() {
        executorService.execute(() -> {
            logger.info("执行任务,线程名:{}", Thread.currentThread().getName());
        });
    }
}

步骤 3:在 Controller 中调用

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequestMapping("/executor")
public class ExecutorServiceController {

    @Autowired
    private ExecutorServiceTask executorServiceTask;

    @GetMapping("/run")
    public String runTask() {
        executorServiceTask.executeTask();
        return "任务已提交";
    }
}

总结

方式适用场景配置方式
ThreadPoolTaskExecutor普通异步任务 (@Asyncexecute)@Autowired private ThreadPoolTaskExecutor
ThreadPoolTaskScheduler定时任务@Autowired private ThreadPoolTaskScheduler
ExecutorService原生 Java 线程池@Autowired private ExecutorService

推荐方式

  • 使用 ThreadPoolTaskExecutor 结合 @Autowired 来管理异步任务(推荐)。
  • 使用 ThreadPoolTaskScheduler 进行定时任务调度。
  • 避免直接使用 ExecutorService,因为它不受 Spring 管理,不能动态调整线程池参数。

这样可以 充分利用 Spring Boot 线程池管理,提高系统性能,减少资源消耗,并且代码更易维护! 🚀

相关文章:

  • 人工智能驱动数字孪生城市的实践探索
  • 《AI生成文章SEO 长尾关键词下拉词相关词抓取工具 SEO 裂变工具:高效驱动网站流量增长》
  • qq音乐 webpack 补环境
  • Unity3D仿星露谷物语开发31之设置地面属性方法探索
  • K8S学习之基础三十一:k8s中RBAC 的核心概念
  • MySQL的行级锁锁的到底是什么?
  • 深入探讨RAID 5的性能与容错能力:实验与分析(磁盘阵列)
  • 深入理解 IP、子网掩码、端口号和协议
  • 快速上手网络通信 -- Qt Network应用开发
  • vulhub/log4j2漏洞靶场----反弹shell
  • centos7安装时采用的默认分区(比如:/dev/sda3的对应挂载点是/),如何对系统扩容?
  • python的基本运用(六)(自定义函数def)
  • 特殊 IP 地址
  • 机器人交社保属于“无稽之谈”?
  • GAN生成对抗网络小记
  • 【leetcode100】括号生成
  • Linux内核,mmap_pgoff在mmap.c的实现
  • MyBatis SqlSessionFactory 是如何创建的?
  • Linux上位机开发实战(x86和arm自由切换)
  • java -jar 执行基于Maven构建的Java应用的方法总结
  • 人民空军:网上出现的“运-20向外方运送物资”为不实消息
  • “影像上海”中的自媒体影像特展:无论何时,影像都需要空间
  • 哈尔滨工业大学原副校长王魁业逝世,享年92岁
  • 卢正已任上海市司法局党委委员、副局长
  • 时代中国控股:前4个月销售额18.1亿元,境外债重组协议押后聆讯至5月底
  • 时隔14个月北京怀柔区重启供地,北京建工以3.59亿元摘得