SERVER-67905 Created PriorityTicketHolder that queues by 3 priority levels

1 files changed, 19 insertions, 0 deletions

diff --git a/src/mongo/util/concurrency/ticketholder.h b/src/mongo/util/concurrency/ticketholder.h
index 7512e3908f3..66769f1c015 100644
--- a/src/mongo/util/concurrency/ticketholder.h
+++ b/src/mongo/util/concurrency/ticketholder.h
@@ -367,6 +367,8 @@ public:
 
     int available() const override final;
 
+    int queued() const override final;
+
 private:
     bool _tryAcquireTicket();
 
@@ -400,6 +402,7 @@ private:
 
     QueueMutex _queueMutex;
     AtomicWord<int> _ticketsAvailable;
+    AtomicWord<int> _enqueuedElements;
     ServiceContext* _serviceContext;
 };
 
@@ -421,6 +424,22 @@ private:
     std::uint32_t _totalWeight;
 };
 
+class PriorityTicketHolder final : public SchedulingTicketHolder {
+public:
+    explicit PriorityTicketHolder(int numTickets, ServiceContext* serviceContext);
+
+private:
+    enum class QueueType : unsigned int {
+        LowPriorityQueue = 0,
+        NormalPriorityQueue = 1,
+        QueueTypeSize = 2
+    };
+
+    void _dequeueWaitingThread() override final;
+
+    Queue& _getQueueToUse(OperationContext* opCtx, const AdmissionContext* admCtx) override final;
+};
+
 /**
  * RAII-style movable token that gets generated when a ticket is acquired and is automatically
  * released when going out of scope.