fix: add cpp retry

eesast · May 8, 2024 · ba6823e · ba6823e
1 parent 797a835
commit ba6823e
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/CAPI/cpp/API/src/Communication.cpp b/CAPI/cpp/API/src/Communication.cpp
@@ -227,14 +227,18 @@ bool Communication::Recycle(int32_t playerID, int32_t teamID)
 
 bool Communication::TryConnection(int32_t playerID, int32_t teamID)
 {
-    protobuf::BoolRes reply;
-    ClientContext context;
+    constexpr int maxRetryNum = 10;
     auto request = THUAI72Proto::THUAI72ProtobufIDMsg(playerID, teamID);
-    auto status = THUAI7Stub->TryConnection(&context, request, &reply);
-    if (status.ok())
-        return true;
-    else
-        return false;
+    for (int retryNum = 0; retryNum < maxRetryNum; retryNum++)
+    {
+        protobuf::BoolRes reply;
+        ClientContext context;
+        std::this_thread::sleep_for(std::chrono::milliseconds(2000));
+        auto status = THUAI7Stub->TryConnection(&context, request, &reply);
+        if (status.ok())
+            return true;
+    }
+    return false;
 }
 
 void Communication::AddPlayer(int32_t playerID, int32_t teamID, THUAI7::ShipType ShipType)
@@ -274,4 +278,4 @@ protobuf::MessageToClient Communication::GetMessage2Client()
                    { return haveNewMessage; });
     haveNewMessage = false;
     return message2Client;
-}
+}