溫馨提示×

C++線程池庫的異常處理機制是什么

c++
小樊
82
2024-09-06 10:35:11
欄目: 編程語言

在C++中，線程池庫通常使用std::exception_ptr來處理異常。當一個任務在線程池中執(zhí)行時，如果拋出了異常，線程池會捕獲這個異常并將其存儲為std::exception_ptr。然后，當調用者嘗試獲取任務的結果時，線程池會重新拋出這個異常。

以下是一個簡單的示例，展示了如何在C++線程池中處理異常：

#include<iostream>
#include<thread>
#include <future>
#include<vector>
#include <stdexcept>

class ThreadPool {
public:
    ThreadPool(size_t num_threads) {
        for (size_t i = 0; i < num_threads; ++i) {
            workers.emplace_back(&ThreadPool::work, this);
        }
    }

    ~ThreadPool() {
        for (auto& worker : workers) {
            if (worker.joinable()) {
                worker.join();
            }
        }
    }

    template<typename F, typename... Args>
    auto enqueue(F&& f, Args&&... args) -> std::future<typename std::result_of<F(Args...)>::type> {
        using return_type = typename std::result_of<F(Args...)>::type;
        auto task = std::make_shared<std::packaged_task<return_type()>>(
            std::bind(std::forward<F>(f), std::forward<Args>(args)...));
        std::future<return_type> res = task->get_future();
        tasks.push([task]() {
            try {
                (*task)();
            } catch (...) {
                task->set_exception(std::current_exception());
            }
        });
        return res;
    }

private:
    void work() {
        while (true) {
            std::function<void()> task;
            tasks.pop(task);
            if (!task) {
                return;
            }
            task();
        }
    }

    std::vector<std::thread> workers;
    SafeQueue<std::function<void()>> tasks;
};

int main() {
    ThreadPool pool(4);

    auto result = pool.enqueue([]() {
        throw std::runtime_error("An error occurred");
        return 42;
    });

    try {
        std::cout << "Result: "<< result.get()<< std::endl;
    } catch (const std::exception& e) {
        std::cerr << "Caught exception: " << e.what()<< std::endl;
    }

    return 0;
}

在這個示例中，我們創(chuàng)建了一個簡單的線程池，它可以接受任務并在工作線程中執(zhí)行它們。當任務拋出異常時，線程池會捕獲這個異常并將其存儲為std::exception_ptr。當調用者嘗試獲取任務的結果時，線程池會重新拋出這個異常。在main函數中，我們捕獲并處理了這個異常。

0 踩