Add original size and chunk logging for BigValueRoute set requests

Summary: This is a follow up to D76844605 which was reverted due to causing a build failure in Ceres MQB

Reviewed By: disylh

Differential Revision: D77383006

fbshipit-source-id: e6217ceb1bb8d0ced9c3d5e7a7549b7f73457b73

Author: IvanTopolcic

third-party/mcrouter/src/mcrouter/McrouterFiberContext.h

@@ -68,6 +68,11 @@ struct AxonContext {
   std::string poolFilter;
 };
 
+struct BigValueContext {
+  uint64_t originalItemSize{0};
+  uint32_t numChunks{0};
+};
+
 template <class RouterInfo>
 class fiber_local {
  private:
@@ -94,6 +99,7 @@ class fiber_local {
     std::shared_ptr<AxonContext> axonCtx{nullptr};
     int64_t accumulatedBeforeReqInjectedLatencyUs{0};
     int64_t accumulatedAfterReqInjectedLatencyUs{0};
+    std::optional<BigValueContext> bigValueContext{std::nullopt};
   };
 
   static auto makeGuardHelperBase(McrouterFiberContext&& tmp) {
@@ -313,6 +319,15 @@ class fiber_local {
     return folly::fibers::local<McrouterFiberContext>().load;
   }
 
+  static std::optional<BigValueContext>& getBigValueContext() {
+    return folly::fibers::local<McrouterFiberContext>().bigValueContext;
+  }
+
+  static void setBigValueContext(BigValueContext bigValueContext) {
+    folly::fibers::local<McrouterFiberContext>().bigValueContext =
+        bigValueContext;
+  }
+
   static void incNetworkTransportTimeBy(int64_t duration_us) {
     folly::fibers::local<McrouterFiberContext>().networkTransportTimeUs +=
         duration_us;

third-party/mcrouter/src/mcrouter/ProxyRequestContextTyped.h

@@ -207,7 +207,8 @@ class ProxyRequestContextWithInfo : public ProxyRequestContext {
         getProductId(request),
         getRegionalizationEntity(request),
         getUsecaseId(request),
-        getReplySourceBitMask(reply));
+        getReplySourceBitMask(reply),
+        fiber_local<RouterInfo>::getBigValueContext());
     assert(logger_.hasValue());
     logger_->template log<Request>(loggerContext);
     assert(additionalLogger_.hasValue());

third-party/mcrouter/src/mcrouter/lib/RequestLoggerContext.h

@@ -120,7 +120,8 @@ struct RequestLoggerContext {
       std::optional<int64_t> productId_ = std::nullopt,
       std::optional<int32_t> regionalizationEntity_ = std::nullopt,
       std::optional<int64_t> usecaseId_ = std::nullopt,
-      uint32_t replySourceBitMask_ = 0)
+      uint32_t replySourceBitMask_ = 0,
+      std::optional<BigValueContext> bigValueContext_ = std::nullopt)
       : strippedRoutingPrefix(strippedRoutingPrefix_),
         requestClass(requestClass_),
         poolName(poolName_),
@@ -140,7 +141,8 @@ struct RequestLoggerContext {
         productId(productId_),
         regionalizationEntity(regionalizationEntity_),
         usecaseId(usecaseId_),
-        replySourceBitMask(replySourceBitMask_) {}
+        replySourceBitMask(replySourceBitMask_),
+        bigValueContext(bigValueContext_) {}
 
   RequestLoggerContext(const RequestLoggerContext&) = delete;
   RequestLoggerContext& operator=(const RequestLoggerContext&) = delete;
@@ -165,6 +167,7 @@ struct RequestLoggerContext {
   std::optional<int32_t> regionalizationEntity;
   std::optional<uint64_t> usecaseId;
   const uint32_t replySourceBitMask;
+  std::optional<BigValueContext> bigValueContext;
 };
 
 } // namespace mcrouter

third-party/mcrouter/src/mcrouter/routes/BigValueRoute-inl.h

@@ -234,8 +234,11 @@ typename std::enable_if_t<
         typename BigValueRoute<RouterInfo>::ChunksInfo>>
 BigValueRoute<RouterInfo>::chunkUpdateRequests(const Request& req) const {
   const folly::IOBuf& value = *req.value_ref();
-  int numChunks = (value.computeChainDataLength() + options_.threshold - 1) /
-      options_.threshold;
+  uint64_t itemSize = value.computeChainDataLength();
+  int numChunks = (itemSize + options_.threshold - 1) / options_.threshold;
+  fiber_local<RouterInfo>::setBigValueContext(BigValueContext{
+      .originalItemSize = itemSize,
+      .numChunks = folly::to<uint32_t>(numChunks)});
   ChunksInfo info(numChunks, detail::hashBigValue(value));
 
   std::vector<McSetRequest> chunkReqs;