C/C++ 中实现类似 Go Channels 功能的方案

// 任务对象
struct Task {
    int client_socket;
    std::string data;
};

// 任务队列 (可以使用线程安全的队列)
std::queue<Task> task_queue;
std::mutex task_queue_mutex;
std::condition_variable task_queue_cv;

// 工作线程函数
void worker_thread() {
    while (true) {
        std::unique_lock<std::mutex> lock(task_queue_mutex);
        task_queue_cv.wait(lock, []{ return !task_queue.empty(); }); // 等待任务

        Task task = task_queue.front();
        task_queue.pop();
        lock.unlock();

        // 处理任务
        process_data(task.client_socket, task.data);
    }
}

// 主线程函数
void main_thread() {
    // 创建线程池
    std::vector<std::thread> thread_pool;
    for (int i = 0; i < NUM_THREADS; ++i) {
        thread_pool.emplace_back(worker_thread);
    }

    // 监听端口，接收连接
    while (true) {
        int client_socket = accept_connection();
        std::string data = receive_data(client_socket);

        // 创建任务对象
        Task task = {client_socket, data};

        // 将任务放入任务队列
        std::lock_guard<std::mutex> lock(task_queue_mutex);
        task_queue.push(task);
        task_queue_cv.notify_one(); // 唤醒一个工作线程
    }

    // 等待所有线程结束 (实际应用中需要更优雅的关闭方式)
    for (auto& thread : thread_pool) {
        thread.join();
    }
}

#include "Poco/ThreadPool.h"
#include "Poco/Runnable.h"
#include "Poco/Thread.h"
#include <iostream>

class MyTask : public Poco::Runnable {
public:
    MyTask(int id) : _id(id) {}

    void run() {
        std::cout << "Task " << _id << " running in thread " << Poco::Thread::current()->name() << std::endl;
        Poco::Thread::sleep(1000); // 模拟耗时操作
        std::cout << "Task " << _id << " finished" << std::endl;
    }

private:
    int _id;
};

int main() {
    Poco::ThreadPool& pool = Poco::ThreadPool::defaultPool();
    pool.setCapacity(4); // 设置最大线程数
    pool.setMaxQueueSize(10); // 设置最大任务队列长度

    for (int i = 0; i < 10; ++i) {
        pool.start(new MyTask(i)); // 提交任务
    }

    pool.joinAll(); // 等待所有任务完成

    return 0;
}