From 475bb5db4993c3be127ec061229f32edab4b7bb5 Mon Sep 17 00:00:00 2001
From: IsolatedSushi <simen.vanherpt@gmail.com>
Date: Sat, 27 Feb 2021 18:26:14 +0100
Subject: [PATCH] Update webSocketGateway.js

---
 backend/webSocketGateway/src/webSocketGateway.js | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/backend/webSocketGateway/src/webSocketGateway.js b/backend/webSocketGateway/src/webSocketGateway.js
index 3b2a253..1150c9c 100644
--- a/backend/webSocketGateway/src/webSocketGateway.js
+++ b/backend/webSocketGateway/src/webSocketGateway.js
@@ -103,6 +103,9 @@ function getProjectorConnection(ws,client){
 
 function getKNNConnection(connection){
   connection.knnConn.client.getIDfromServer({},function(error,response){
+    if(!response){
+      return;
+    }
     console.log(error);
     connection.knnConn.id = response["id"];
     const meta = new grpc.Metadata();
@@ -185,6 +188,7 @@ function SetupKProjectors(amount,connection){
   //Create that amount of grpc connections
   for(var i = 0; i< amount; i+=1){
     var grpcConnection = getProjectorConnection(connection.ws,connection.projectorConn.client);
+    console.log("test")
     allCalls.push(grpcConnection)
   }
   connection.projectorConn.calls = allCalls;
@@ -222,7 +226,7 @@ function sendRowToServer(row,lineIndex,connection) {
 }
 
 function sendRowToKNN(trainingSetRow,connection){
-  if(!connection.knnConn){
+  if(!connection.knnConn.calls || connection.knnConn.calls.length == 0){
     return;
   }
 
-- 
GitLab