X-Git-Url: http://xvm.mit.edu/gitweb/invirt/third/libt4.git/blobdiff_plain/f2170465073de34adf89161d4287182b518352c4..c06ef44e7af1571710fd31dd0ab068dd77b1eb2d:/rpc/fifo.h diff --git a/rpc/fifo.h b/rpc/fifo.h index dde514d..dfb4d05 100644 --- a/rpc/fifo.h +++ b/rpc/fifo.h @@ -1,53 +1,47 @@ #ifndef fifo_h #define fifo_h -#include -#include "lock.h" +#include "types.h" // blocks enq() and deq() when queue is FULL or EMPTY template class fifo { - public: - fifo(size_t limit=0) : max_(limit) {} - bool enq(T, bool blocking=true); - void deq(T *); - bool size() { + public: + fifo(size_t limit=0) : max_(limit) {} + + bool enq(T e, bool blocking=true) { + lock ml(m_); + while (max_ && q_.size() >= max_) { + if (!blocking) + return false; + has_space_c_.wait(ml); + } + q_.push_back(e); + non_empty_c_.notify_one(); + return true; + } + + void deq(T * e) { + lock ml(m_); + while(q_.empty()) + non_empty_c_.wait(ml); + *e = q_.front(); + q_.pop_front(); + if (max_ && q_.size() < max_) + has_space_c_.notify_one(); + } + + bool size() { lock ml(m_); return q_.size(); } - private: - std::list q_; - mutex m_; + private: + std::list q_; + std::mutex m_; cond non_empty_c_; // q went non-empty - cond has_space_c_; // q is not longer overfull - size_t max_; // maximum capacity of the queue, block enq threads if exceeds this limit + cond has_space_c_; // q is not longer overfull + size_t max_; // maximum capacity of the queue, block enq threads if exceeds this limit }; -template bool -fifo::enq(T e, bool blocking) -{ - lock ml(m_); - while (max_ && q_.size() >= max_) { - if (!blocking) - return false; - has_space_c_.wait(ml); - } - q_.push_back(e); - non_empty_c_.notify_one(); - return true; -} - -template void -fifo::deq(T *e) -{ - lock ml(m_); - while(q_.empty()) - non_empty_c_.wait(ml); - *e = q_.front(); - q_.pop_front(); - if (max_ && q_.size() < max_) - has_space_c_.notify_one(); -} - #endif