Add readWithPriority to PriorityMPMCQueue

[folly.git] / folly / PriorityMPMCQueue.h

diff --git a/folly/PriorityMPMCQueue.h b/folly/PriorityMPMCQueue.h
index fb4e4645228d8ff8985cca87380b1645f79e9922..b12cbf33942ccff671404f6a4373427408c368c1 100644 (file)
--- a/folly/PriorityMPMCQueue.h
+++ b/folly/PriorityMPMCQueue.h
@@ -66,6 +66,10 @@ class PriorityMPMCQueue {
     return false;
   }
 
+  bool readWithPriority(T& item, size_t priority) {
+    return queues_[priority].readIfNotEmpty(item);
+  }
+
   size_t size() const {
     size_t total_size = 0;
     for (auto& q : queues_) {