HarmonyOS taskpool线程数量控制

我们在一些业务场景中需要使用线程池，并且想让线程池中的线程按一定数量排队执行。

比如总共有100个任务，我需要构建一个大小为5的线程池，这样的每次是5个线程同时在执行。每5个一组，直到100个任务全部执行完。

HarmonyOS

put_get

2024-08-28 09:36:13

浏览

回答 2

待解决

回答 2

按赞同

按时间

鱼弦CTO

在 HarmonyOS 中，使用线程池可以有效管理并发任务的执行。要实现一个固定大小的线程池，并让线程按一定数量排队执行，我们可以使用 ThreadPoolExecutor 类来创建和管理我们的线程池。

以下是一个示例，展示如何构建一个大小为 5 的线程池来处理 100 个任务：

### 示例代码

#### 创建线程池及提交任务

import ohos.app.Context;
import ohos.eventhandler.EventHandler;
import ohos.eventhandler.EventRunner;

import java.util.concurrent.Executors;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

public class TaskPoolExample {
    private static final int POOL_SIZE = 5; // 线程池大小
    private static final int TOTAL_TASKS = 100; // 总任务数

    public static void main(String[] args) {
        // 创建一个固定大小为5的线程池
        ThreadPoolExecutor executor = (ThreadPoolExecutor) Executors.newFixedThreadPool(POOL_SIZE);

        // 提交100个任务到线程池中
        for (int i = 0; i < TOTAL_TASKS; i++) {
            int taskId = i + 1;
            executor.submit(new Task(taskId));
        }
        
        // 关闭线程池
        executor.shutdown();
        
        try {
            // 等待所有任务完成
            boolean allTasksCompleted = executor.awaitTermination(Long.MAX_VALUE, TimeUnit.NANOSECONDS);
            if (allTasksCompleted) {
                System.out.println("所有任务已完成");
            } else {
                System.out.println("等待任务完成超时");
            }
        } catch (InterruptedException e) {
            System.err.println("任务等待被中断: " + e.getMessage());
        }
    }

    // 定义一个任务类，实现Runnable接口
    static class Task implements Runnable {
        private final int taskId;

        public Task(int taskId) {
            this.taskId = taskId;
        }

        @Override
        public void run() {
            System.out.println("任务 " + taskId + " 正在执行");
            // 模拟任务执行时间
            try {
                Thread.sleep(1000); // 这里的睡眠时间可以根据实际任务调整
            } catch (InterruptedException e) {
                Thread.currentThread().interrupt();
                System.err.println("任务 " + taskId + " 被中断");
            }
            System.out.println("任务 " + taskId + " 完成");
        }
    }
}

#### 解释

初始化线程池：我们使用Executors.newFixedThreadPool(POOL_SIZE) 方法创建一个固定大小为 5 的线程池。
提交任务：通过executor.submit(new Task(taskId)) 将 100 个任务提交到线程池中。每个任务实现了Runnable 接口，在run 方法中定义任务的具体执行逻辑。
关闭线程池：通过executor.shutdown() 方法关闭线程池，表示不再接受新任务。
等待任务完成：使用executor.awaitTermination(Long.MAX_VALUE, TimeUnit.NANOSECONDS) 方法阻塞当前线程，直到所有任务完成或超时。

此示例会确保每次只有 5 个线程同时在执行任务，当一个任务完成后，下一个任务会被调度执行，直到 100 个任务全部完成。

### 适应 HarmonyOS 环境

如果你是在 HarmonyOS 的某些特定模块（如 UI 线程）中使用线程池，需要注意 HarmonyOS 的应用架构。通常，除非需要进行复杂的后台任务管理，否则建议将这种繁重任务放在服务组件中执行，以避免影响主线程的响应性。

### 小结

通过以上代码，我们可以控制线程池中的线程按一定数量排队执行任务。这样能够保证系统资源的有效利用，同时避免因过多并发任务而导致的性能问题。