PowerToys/src/common/interop/async_message_queue.h

#pragma once
#include <queue>
#include <thread>
#include <mutex>
#include <condition_variable>
#include <string>

class AsyncMessageQueue
{
private:
    std::mutex queue_mutex;
    std::queue<std::wstring> message_queue;
    std::condition_variable message_ready;
    bool interrupted = false;

    //Disable copy
    AsyncMessageQueue(const AsyncMessageQueue&);
    AsyncMessageQueue& operator=(const AsyncMessageQueue&);

public:
    AsyncMessageQueue()
    {
    }
    void queue_message(std::wstring message)
    {
        this->queue_mutex.lock();
        this->message_queue.push(message);
        this->queue_mutex.unlock();
        this->message_ready.notify_one();
    }
    std::wstring pop_message()
    {
        std::unique_lock<std::mutex> lock(this->queue_mutex);
        while (message_queue.empty() && !this->interrupted)
        {
            this->message_ready.wait(lock);
        }
        if (this->interrupted)
        {
            //Just returns a empty string if the queue was interrupted.
            return std::wstring(L"");
        }
        std::wstring message = this->message_queue.front();
        this->message_queue.pop();
        return message;
    }
    void interrupt()
    {
        this->queue_mutex.lock();
        this->interrupted = true;
        this->queue_mutex.unlock();
        this->message_ready.notify_all();
    }
};