diff options
Diffstat (limited to 'backend')
| -rw-r--r-- | backend/api/api/Services/ChatHub.cs | 6 | ||||
| -rw-r--r-- | backend/api/api/Services/MlConnectionService.cs | 3 |
2 files changed, 8 insertions, 1 deletions
diff --git a/backend/api/api/Services/ChatHub.cs b/backend/api/api/Services/ChatHub.cs index 3dd6ab3e..efa13a3b 100644 --- a/backend/api/api/Services/ChatHub.cs +++ b/backend/api/api/Services/ChatHub.cs @@ -43,6 +43,12 @@ namespace api.Services { await Clients.All.SendAsync("Notify",message); } + public static bool CheckUser(string id) + { + if (Users[id] == null) + return false; + return true; + } } diff --git a/backend/api/api/Services/MlConnectionService.cs b/backend/api/api/Services/MlConnectionService.cs index 2ac977d9..1b001973 100644 --- a/backend/api/api/Services/MlConnectionService.cs +++ b/backend/api/api/Services/MlConnectionService.cs @@ -57,7 +57,8 @@ namespace api.Services Dataset newDataset = JsonConvert.DeserializeObject<Dataset>(result.Content); newDataset.isPreProcess = true; _datasetService.Update(newDataset); - await _ichat.Clients.Client(ChatHub.Users[id]).SendAsync("Notify", "Preprocessed dataset with name "+newDataset.name); + if(ChatHub.CheckUser(id)) + await _ichat.Clients.Client(ChatHub.Users[id]).SendAsync("Notify", "Preprocessed dataset with name "+newDataset.name); return; } |