Add state annotations in SonarWebSocketImpl

Summary: These will be displayed in the sonar diagnostics screen, for troubleshooting connection issues.

Reviewed By: danielbuechele

Differential Revision: D9150552

fbshipit-source-id: c6d65fba86e7564fbb004aaa7b0303a1d5952e5d

xplat/Sonar/SonarWebSocketImpl.cpp

@@ -99,10 +99,11 @@ SonarWebSocketImpl::~SonarWebSocketImpl() {
 }
 
 void SonarWebSocketImpl::start() {
+  auto step = sonarState_->start("Start connection thread");
   folly::makeFuture()
       .via(sonarEventBase_->getEventBase())
       .delayed(std::chrono::milliseconds(0))
-      .thenValue([this](auto&&){ startSync(); });
+      .thenValue([this, step](auto&&){ step->complete(); startSync(); });
 }
 
 void SonarWebSocketImpl::startSync() {
@@ -137,6 +138,7 @@ void SonarWebSocketImpl::doCertificateExchange() {
           "device", deviceData_.device)("app", deviceData_.app)));
   address.setFromHostPort(deviceData_.host, insecurePort);
 
+  auto connectingInsecurely = sonarState_->start("Connect insecurely");
   connectionIsTrusted_ = false;
   client_ =
       rsocket::RSocket::createConnectedClient(
@@ -148,6 +150,7 @@ void SonarWebSocketImpl::doCertificateExchange() {
           nullptr, // stats
           std::make_shared<ConnectionEvents>(this))
           .get();
+  connectingInsecurely->complete();
 
   ensureSonarDirExists();
   requestSignedCertFromSonar();
@@ -172,6 +175,7 @@ void SonarWebSocketImpl::connectSecurely() {
       absoluteFilePath(PRIVATE_KEY_FILE).c_str());
   sslContext->authenticate(true, false);
 
+  auto connectingSecurely = sonarState_->start("Connect securely");
   connectionIsTrusted_ = true;
   client_ =
       rsocket::RSocket::createConnectedClient(
@@ -185,6 +189,7 @@ void SonarWebSocketImpl::connectSecurely() {
           nullptr, // stats
           std::make_shared<ConnectionEvents>(this))
           .get();
+  connectingSecurely->complete();
   failedConnectionAttempts_ = 0;
 }
 
@@ -219,6 +224,7 @@ void SonarWebSocketImpl::sendMessage(const folly::dynamic& message) {
 }
 
 bool SonarWebSocketImpl::isCertificateExchangeNeeded() {
+  auto step = sonarState_->start("Check required certificates are present");
   if (failedConnectionAttempts_ >= 2) {
     return true;
   }
@@ -232,23 +238,29 @@ bool SonarWebSocketImpl::isCertificateExchangeNeeded() {
   if (caCert == "" || clientCert == "" || privateKey == "") {
     return true;
   }
-
+  step->complete();
   return false;
 }
 
 void SonarWebSocketImpl::requestSignedCertFromSonar() {
+  auto generatingCSR = sonarState_->start("Generate CSR");
   generateCertSigningRequest(
       deviceData_.appId.c_str(),
       absoluteFilePath(CSR_FILE_NAME).c_str(),
       absoluteFilePath(PRIVATE_KEY_FILE).c_str());
+  generatingCSR->complete();
+  auto loadingCSR = sonarState_->start("Load CSR");
   std::string csr = loadStringFromFile(absoluteFilePath(CSR_FILE_NAME));
+  loadingCSR->complete();
 
   folly::dynamic message = folly::dynamic::object("method", "signCertificate")(
       "csr", csr.c_str())("destination", absoluteFilePath("").c_str());
-  sonarEventBase_->add([this, message]() {
+  auto sendingCSR = sonarState_->start("Send CSR to desktop");
+  sonarEventBase_->add([this, message, sendingCSR]() {
     client_->getRequester()
         ->fireAndForget(rsocket::Payload(folly::toJson(message)))
-        ->subscribe([this]() {
+        ->subscribe([this, sendingCSR]() {
+          sendingCSR->complete();
           // Disconnect after message sending is complete.
           // This will trigger a reconnect which should use the secure channel.
           client_ = nullptr;