ConditionalQueue.h

Go to the documentation of this file.

#ifndef SISCO_CONDITIONAL_QUEUE_H_
#define SISCO_CONDITIONAL_QUEUE_H_
 
#include <cassert>
#include <condition_variable>
#include <list>
 
namespace casacore::sisco {

template <typename T>
class ConditionalQueue {
 public:
  ConditionalQueue(size_t max_size) : max_size_(max_size) {}
 
  ~ConditionalQueue() = default;

  void Push(T&& value, std::unique_lock<std::mutex>& lock) {
    assert(lock.owns_lock());
    while (values_.size() == max_size_) {
      pop_condition_.wait(lock);
    }
    values_.emplace_back(std::move(value));
    // We need to notify all, because if only one thread is awoken that
    // is looking for a specific condition that is not met, the signal
    // is lost while another nother thread's condition may be met.
    push_condition_.notify_all();
  }

  template <typename... Args>
  void Emplace(std::unique_lock<std::mutex>& lock, Args&&... args) {
    assert(lock.owns_lock());
    while (values_.size() == max_size_) {
      pop_condition_.wait(lock);
    }
    values_.emplace_back(std::forward<Args>(args)...);
    // We need to notify all, because if only one thread is awoken that
    // is looking for a specific condition that is not met, the signal
    // is lost while another nother thread's condition may be met.
    push_condition_.notify_all();
  }

  template <typename Condition>
  bool PopIf(T& result, std::unique_lock<std::mutex>& lock,
             Condition condition) {
    assert(lock.owns_lock());
 
    typename std::list<T>::iterator iterator = GetNext(condition);
    while (iterator == values_.end()) {
      if (values_.empty() && is_finished_) return false;
      push_condition_.wait(lock);
      iterator = GetNext(condition);
    }
 
    result = std::move(*iterator);
    values_.erase(iterator);
    pop_condition_.notify_one();
 
    // If this thread emptied the finished queue while another thread is waiting
    // for a value, it needs to be notified to finish.
    if (values_.empty() && is_finished_) {
      push_condition_.notify_all();
    }
 
    return true;
  }

  void NotifyOneChange() { push_condition_.notify_all(); }

  void Finish([[maybe_unused]] std::unique_lock<std::mutex>& lock) {
    assert(lock.owns_lock());
    is_finished_ = true;
    push_condition_.notify_all();
  }
 
  size_t Size([[maybe_unused]] std::unique_lock<std::mutex>& lock) const {
    assert(lock.owns_lock());
    return values_.size();
  }
 
  size_t MaxSize() const { return max_size_; }
 
 private:
  template <typename Condition>
  std::list<T>::iterator GetNext(Condition&& condition) {
    for (typename std::list<T>::iterator i = values_.begin();
         i != values_.end(); ++i) {
      if (condition(*i)) return i;
    }
    return values_.end();
  }
 
  std::list<T> values_;
  bool is_finished_ = false;
  size_t max_size_;
  std::condition_variable pop_condition_;
  std::condition_variable push_condition_;
};
 
}  // namespace casacore::sisco
 
#endif