fix: reserve extra size for high priority request in priority queue

oceanbase · Aug 11, 2023 · 14e357b · 14e357b
1 parent 019cb49
commit 14e357b
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/deps/oblib/src/lib/queue/ob_priority_queue.h b/deps/oblib/src/lib/queue/ob_priority_queue.h
@@ -129,7 +129,15 @@ class ObPriorityQueue2
   int push(ObLink* data, int priority)
   {
     int ret = OB_SUCCESS;
-    if (ATOMIC_FAA(&size_, 1) > limit_) {
+    int64_t extra;
+    if (priority < HIGH_PRIOS) {
+      extra = 2048;
+    } else if (priority < NORMAL_PRIOS + HIGH_PRIOS) {
+      extra = 1024;
+    } else {
+      extra = 0;
+    }
+    if (ATOMIC_FAA(&size_, 1) > limit_ + extra) {
       ret = OB_SIZE_OVERFLOW;
     } else if (OB_UNLIKELY(NULL == data) || OB_UNLIKELY(priority < 0) || OB_UNLIKELY(priority >= PRIO_CNT)) {
       ret = OB_INVALID_ARGUMENT;