osci-render/Source/concurrency/ConsumerManager.h

#pragma once

#include <JuceHeader.h>
#include "../shape/Point.h"
#include "BufferConsumer.h"


class ConsumerManager {
public:
    ConsumerManager() {}
    ~ConsumerManager() {}

    std::shared_ptr<BufferConsumer> consumerRegister(std::vector<Point>& buffer) {
        std::shared_ptr<BufferConsumer> consumer = std::make_shared<BufferConsumer>(buffer);
        juce::SpinLock::ScopedLockType scope(consumerLock);
        consumers.push_back(consumer);

        return consumer;
    }
    
    void consumerRead(std::shared_ptr<BufferConsumer> consumer) {
        consumer->waitUntilFull();
        juce::SpinLock::ScopedLockType scope(consumerLock);
        consumers.erase(std::remove(consumers.begin(), consumers.end(), consumer), consumers.end());
    }

    void consumerStop(std::shared_ptr<BufferConsumer> consumer) {
        if (consumer != nullptr) {
            juce::SpinLock::ScopedLockType scope(consumerLock);
            consumer->forceNotify();
        }
    }

protected:
    juce::SpinLock consumerLock;
    std::vector<std::shared_ptr<BufferConsumer>> consumers;
};