FIFOWaitBlocker.hxx

Go to the documentation of this file.

//
// Created by 蒋志平 on 2024/10/17.
//
 
#ifndef FIFOWAITBLOCKER_HXX
#define FIFOWAITBLOCKER_HXX
 
#include <condition_variable>
#include <deque>
#include <thread>
#include <map>
 
class FIFOWaitBlocker {
public:
    static std::shared_ptr<FIFOWaitBlocker> getInstance(const std::string& instanceName) {
        std::lock_guard<std::mutex> lock(instanceMutex); // 加锁
        static std::map<std::string, std::shared_ptr<FIFOWaitBlocker>> instanceMap;
        if (!instanceMap.contains(instanceName)) {
            instanceMap[instanceName] = std::shared_ptr<FIFOWaitBlocker>(new FIFOWaitBlocker());
        }
 
        return instanceMap[instanceName];
    }
 
    void joinFIFOWaitQueue() {
        std::lock_guard guard(queueMutex);
        threadWaitQueue.emplace_back(std::this_thread::get_id());
    }
 
    [[nodiscard]] bool isThisThreadsTurn() {
        std::lock_guard guard(queueMutex);
        return !threadWaitQueue.empty() && threadWaitQueue.front() == std::this_thread::get_id();
    }
 
    void exit() {
        auto id = std::this_thread::get_id();
        std::lock_guard guard(queueMutex);
        std::erase_if(threadWaitQueue, [id](const std::thread::id& currentId) {
            return currentId == id;
        });
    }
 
private:
 
    std::mutex queueMutex;
    std::deque<std::thread::id> threadWaitQueue;
 
private:
    static std::mutex instanceMutex; // 全局锁，保护 instanceMap
    FIFOWaitBlocker() = default;
};
std::mutex FIFOWaitBlocker::instanceMutex;
#endif //FIFOWAITBLOCKER_HXX