X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=src%2Fceph%2Fsrc%2Fcommon%2FQueueRing.h;fp=src%2Fceph%2Fsrc%2Fcommon%2FQueueRing.h;h=42582d24f4462c508ae357e8f9256aeda2d0540f;hb=812ff6ca9fcd3e629e49d4328905f33eee8ca3f5;hp=0000000000000000000000000000000000000000;hpb=15280273faafb77777eab341909a3f495cf248d9;p=stor4nfv.git diff --git a/src/ceph/src/common/QueueRing.h b/src/ceph/src/common/QueueRing.h new file mode 100644 index 0000000..42582d2 --- /dev/null +++ b/src/ceph/src/common/QueueRing.h @@ -0,0 +1,63 @@ +#ifndef QUEUE_RING_H +#define QUEUE_RING_H + +#include "common/Mutex.h" +#include "common/Cond.h" + +#include +#include +#include + +template +class QueueRing { + struct QueueBucket { + Mutex lock; + Cond cond; + typename std::list entries; + + QueueBucket() : lock("QueueRing::QueueBucket::lock") {} + QueueBucket(const QueueBucket& rhs) : lock("QueueRing::QueueBucket::lock") { + entries = rhs.entries; + } + + void enqueue(const T& entry) { + lock.Lock(); + if (entries.empty()) { + cond.Signal(); + } + entries.push_back(entry); + lock.Unlock(); + } + + void dequeue(T *entry) { + lock.Lock(); + if (entries.empty()) { + cond.Wait(lock); + }; + assert(!entries.empty()); + *entry = entries.front(); + entries.pop_front(); + lock.Unlock(); + }; + }; + + std::vector buckets; + int num_buckets; + + std::atomic cur_read_bucket = { 0 }; + std::atomic cur_write_bucket = { 0 }; + +public: + QueueRing(int n) : buckets(n), num_buckets(n) { + } + + void enqueue(const T& entry) { + buckets[++cur_write_bucket % num_buckets].enqueue(entry); + }; + + void dequeue(T *entry) { + buckets[++cur_read_bucket % num_buckets].dequeue(entry); + } +}; + +#endif