Added setBucket and recordQueueBeginTime to ServerRequestData

Summary: - This diff introduces `setBucket` and `recordQueueBeginTime` to `ServerRequestData` Reviewed By: sazonovkirill Differential Revision: D70208422 fbshipit-source-id: 20926c85c7d8b10136b928f4e89eabcdd072770d
facebook · Feb 28, 2025 · 05f03a9 · 05f03a9
1 parent 992a40f
commit 05f03a9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/thrift/lib/cpp2/async/ServerRequestData.cpp b/thrift/lib/cpp2/async/ServerRequestData.cpp
@@ -26,6 +26,14 @@ void ServerRequestData::setRequestExecutionEnd(TimePoint now) {
   requestExecutionEnd = now;
 }
 
+void ServerRequestData::recordQueueBeginTime(TimePoint now) {
+  queueBegin = now;
+}
+
+void ServerRequestData::setBucket(size_t priority, size_t bucketNumber) {
+  bucket = {priority, bucketNumber};
+}
+
 ServerRequestData::Duration ServerRequestData::queuedDuration() const {
   return requestExecutionBegin - queueBegin;
 }

diff --git a/thrift/lib/cpp2/async/ServerRequestData.h b/thrift/lib/cpp2/async/ServerRequestData.h
@@ -36,6 +36,8 @@ struct ServerRequestData {
 
   void setRequestExecutionBegin(TimePoint now = Clock::now());
   void setRequestExecutionEnd(TimePoint now = Clock::now());
+  void recordQueueBeginTime(TimePoint now = Clock::now());
+  void setBucket(size_t priority, size_t bucketNumber);
 
   Duration queuedDuration() const;
   Duration requestExecutionDuration() const;