Fix ERROR connecting in train command

FYP-Auto-Scale-JasmineGraph · Aug 27, 2023 · 9ea52a1 · 9ea52a1
1 parent 2a07562
commit 9ea52a1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/frontend/JasmineGraphFrontEnd.cpp b/src/frontend/JasmineGraphFrontEnd.cpp
@@ -1224,6 +1224,8 @@ void *frontendservicesesion(std::string masterIP, int connFd, SQLiteDBInterface
                 jasminGraphTrainingInitiator->initiateTrainingLocally(graphID, trainData);
 
             }
+            write(connFd, DONE.c_str(), FRONTEND_COMMAND_LENGTH);
+            write(connFd, "\r\n", 2);
 
         } else if (line.compare(IN_DEGREE) == 0) {
             frontend_logger.log("Calculating In Degree Distribution", "info");

diff --git a/src/server/JasmineGraphServer.cpp b/src/server/JasmineGraphServer.cpp
@@ -3081,12 +3081,12 @@ void JasmineGraphServer::initiateCommunication(std::string graphID, std::string
 
         if (i==0) {
 
-            workerThreads[threadID] = std::thread(initiateServer,"localhost", serverPort,
+            workerThreads[threadID] = std::thread(initiateServer,workerInstance.hostname, serverPort,
                                                     serverDataPort,trainingArgs,fl_clients, to_string(i));
             threadID++;
         }
 
-        workerThreads[threadID] = std::thread(initiateClient,"localhost", serverPort, serverDataPort,trainingArgs +
+        workerThreads[threadID] = std::thread(initiateClient,workerInstance.hostname, serverPort, serverDataPort,trainingArgs +
                                                     " " + to_string(i), fl_clients, to_string(i));
         threadID++;