fix a bug for thrift connection

Arefsa78 · Arefsa78 · commit 9da59f57a9d8 · 2024-08-21T16:59:46.000-02:30
diff --git a/src/trainer/sample_trainer.cpp b/src/trainer/sample_trainer.cpp
@@ -138,8 +138,14 @@ SampleTrainer::actionImpl()
     while (M_rpc_client->isConnected() == false)
     {
         std::cout<<"Connecting to GRPC server..."<<std::endl;
-        connectedToGrpcServer = M_rpc_client->connectToGrpcServer();
-        if (connectedToGrpcServer == false) {
+        try{
+            connectedToGrpcServer = M_rpc_client->connectToGrpcServer();
+            if (connectedToGrpcServer == false) {
+                std::this_thread::sleep_for(std::chrono::seconds(1));
+            }
+        }
+        catch (const std::exception& e){
+            std::cerr << "Error: " << e.what() << std::endl;
             std::this_thread::sleep_for(std::chrono::seconds(1));
         }
     }

Original file line number	Diff line number	Diff line change
`@@ -138,8 +138,14 @@ SampleTrainer::actionImpl()`
`138`	`138`	`while (M_rpc_client->isConnected() == false)`
`139`	`139`	`{`
`140`	`140`	`std::cout<<"Connecting to GRPC server..."<<std::endl;`
`141`		`- connectedToGrpcServer = M_rpc_client->connectToGrpcServer();`
`142`		`- if (connectedToGrpcServer == false) {`
	`141`	`+ try{`
	`142`	`+ connectedToGrpcServer = M_rpc_client->connectToGrpcServer();`
	`143`	`+ if (connectedToGrpcServer == false) {`
	`144`	`+ std::this_thread::sleep_for(std::chrono::seconds(1));`
	`145`	`+ }`
	`146`	`+ }`
	`147`	`+ catch (const std::exception& e){`
	`148`	`+ std::cerr << "Error: " << e.what() << std::endl;`
`143`	`149`	`std::this_thread::sleep_for(std::chrono::seconds(1));`
`144`	`150`	`}`
`145`	`151`	`}`