Record QUIC connect and MoQ handshake time in MoQClientBase

Summary: See title

Reviewed By: sandarsh

Differential Revision: D95980873

fbshipit-source-id: f770db5d6ce2f4a51daac8f5d0a459b5a669775a

Author: Aman Sharma

moxygen/MoQClientBase.cpp

@@ -36,6 +36,9 @@ folly::coro::Task<void> MoQClientBase::setupMoQSession(
   std::vector<std::string> alpn =
       alpns.empty() ? getDefaultMoqtProtocols(false) : alpns;
   XLOG(DBG1) << "MoQClientBase: QUIC ALPNs: " << folly::join(", ", alpn);
+
+  auto quicConnectStart = std::chrono::steady_clock::now();
+
   // Establish QUIC connection with multiple ALPN options
   auto quicClient = co_await connectQuic(
       folly::SocketAddress(
@@ -45,6 +48,9 @@ folly::coro::Task<void> MoQClientBase::setupMoQSession(
       alpn,
       transportSettings);
 
+  quicConnectTime_ = std::chrono::duration_cast<std::chrono::milliseconds>(
+      std::chrono::steady_clock::now() - quicConnectStart);
+
   if (logger_) {
     if (auto scid = quicClient->getClientConnectionId()) {
       logger_->setSrcCid(*scid);
@@ -69,12 +75,17 @@ folly::coro::Task<void> MoQClientBase::setupMoQSession(
   quicWebTransport_->setHandler(this);
   wt = quicWebTransport_.get();
 
+  auto moqHandshakeStart = std::chrono::steady_clock::now();
+
   auto result = co_await folly::coro::co_awaitTry(completeSetupMoQSession(
       wt,
       url_.getPath(),
       std::move(publishHandler),
       std::move(subscribeHandler)));
 
+  moqHandshakeTime_ = std::chrono::duration_cast<std::chrono::milliseconds>(
+      std::chrono::steady_clock::now() - moqHandshakeStart);
+
   if (result.hasException()) {
     co_yield folly::coro::co_error(result.exception());
   }

moxygen/MoQClientBase.h

@@ -63,6 +63,14 @@ class MoQClientBase : public proxygen::WebTransportHandler {
     return exec_;
   }
 
+  std::chrono::milliseconds getQuicConnectTime() {
+    return quicConnectTime_;
+  }
+
+  std::chrono::milliseconds getMoQHandshakeTime() {
+    return moqHandshakeTime_;
+  }
+
   std::shared_ptr<MoQSession> moqSession_;
   virtual folly::coro::Task<void> setupMoQSession(
       std::chrono::milliseconds connect_timeout,
@@ -114,6 +122,8 @@ class MoQClientBase : public proxygen::WebTransportHandler {
   std::shared_ptr<proxygen::QuicWebTransport> quicWebTransport_;
   std::optional<std::string> negotiatedProtocol_;
   std::shared_ptr<fizz::CertificateVerifier> verifier_;
+  std::chrono::milliseconds quicConnectTime_{0};
+  std::chrono::milliseconds moqHandshakeTime_{0};
 };
 
 } // namespace moxygen