Process high pri request with mutex in rate limiter

Signed-off-by: Yang Zhang <yang.zhang@pingcap.com>
tikv · Sep 23, 2024 · 4b56a99 · 4b56a99
1 parent ca2cf28
commit 4b56a99
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utilities/rate_limiters/write_amp_based_rate_limiter.cc b/utilities/rate_limiters/write_amp_based_rate_limiter.cc
@@ -142,6 +142,7 @@ void WriteAmpBasedRateLimiter::Request(int64_t bytes, const Env::IOPriority pri,
   TEST_SYNC_POINT("WriteAmpBasedRateLimiter::Request");
   TEST_SYNC_POINT_CALLBACK("WriteAmpBasedRateLimiter::Request:1",
                            &rate_bytes_per_sec_);
+  MutexLock g(&request_mutex_);
   if (auto_tuned_.load(std::memory_order_acquire) &&
       (pri == Env::IO_HIGH || pri == Env::IO_USER) &&
       duration_highpri_bytes_through_ + duration_bytes_through_ + bytes <=
@@ -154,7 +155,6 @@ void WriteAmpBasedRateLimiter::Request(int64_t bytes, const Env::IOPriority pri,
     return;
   }
   assert(bytes <= refill_bytes_per_period_.load(std::memory_order_relaxed));
-  MutexLock g(&request_mutex_);
 
   if (auto_tuned_.load(std::memory_order_acquire)) {
     std::chrono::microseconds now(NowMicrosMonotonic(env_));