#ifndef PRODUCERCONSUMERBUFFER_H
#define PRODUCERCONSUMERBUFFER_H

#include <mutex>
#include <condition_variable>
#include <queue>
#include <thread>

using namespace std;


template<class T>
class ProducerConsumerBuffer {
 private:
  queue<T>           queue_;
  mutex              mutex_;
  condition_variable get_cv_;
  bool               done_;

 public:
  ProducerConsumerBuffer() {
    done_ = false;
  }

  ~ProducerConsumerBuffer() {
    done();
  }

  void done() {
    lock_guard<mutex> lock(mutex_);
    done_ = true;
    get_cv_.notify_all();
  }

  T get() {
    unique_lock<mutex> lock(mutex_);
    while (!done_ && queue_.empty()) {
      cout << "\tgettor waiting" << endl;
      get_cv_.wait(lock);
      cout << "\tgettor waking up" << endl;
    }
    if ( done_ ) return T();
    T result = queue_.front();
    queue_.pop();
    return result;
  }

  void put(const T &item) {
    unique_lock<mutex> lock(mutex_);
    queue_.push(item);
    lock.unlock();
    get_cv_.notify_one();
  }
};


#endif // PRODUCERCONSUMERBUFFER_H