From 483eb2f56657e8e7f419ab1a4fab8dce9ade8609 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sat, 27 Apr 2024 20:24:20 +0200 Subject: Adding upstream version 14.2.21. Signed-off-by: Daniel Baumann --- src/common/QueueRing.h | 64 ++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 64 insertions(+) create mode 100644 src/common/QueueRing.h (limited to 'src/common/QueueRing.h') diff --git a/src/common/QueueRing.h b/src/common/QueueRing.h new file mode 100644 index 00000000..af5c47be --- /dev/null +++ b/src/common/QueueRing.h @@ -0,0 +1,64 @@ +// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- +// vim: ts=8 sw=2 smarttab + +#ifndef QUEUE_RING_H +#define QUEUE_RING_H + +#include "common/ceph_mutex.h" + +#include +#include +#include + +template +class QueueRing { + struct QueueBucket { + ceph::mutex lock = ceph::make_mutex("QueueRing::QueueBucket::lock"); + ceph::condition_variable cond; + typename std::list entries; + + QueueBucket() {} + QueueBucket(const QueueBucket& rhs) { + entries = rhs.entries; + } + + void enqueue(const T& entry) { + lock.lock(); + if (entries.empty()) { + cond.notify_all(); + } + entries.push_back(entry); + lock.unlock(); + } + + void dequeue(T *entry) { + std::unique_lock l(lock); + while (entries.empty()) { + cond.wait(l); + }; + ceph_assert(!entries.empty()); + *entry = entries.front(); + entries.pop_front(); + }; + }; + + 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 -- cgit v1.2.3