From 91f930d4c4d7aeb0f41f93f292abbc90cf8c6f77 Mon Sep 17 00:00:00 2001 From: Teddy Date: Tue, 30 Apr 2024 14:17:06 +0200 Subject: [PATCH] FIX #15956 - Send web socket notification on incident assignment/ack (#16086) * fix: send websocket notification on incident assignment * style: ran java linting --- .../jdbi3/TestCaseResolutionStatusRepository.java | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/TestCaseResolutionStatusRepository.java b/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/TestCaseResolutionStatusRepository.java index 176cddcf2ad..14f9bcb3dca 100644 --- a/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/TestCaseResolutionStatusRepository.java +++ b/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/TestCaseResolutionStatusRepository.java @@ -40,6 +40,7 @@ import org.openmetadata.service.util.EntityUtil; import org.openmetadata.service.util.JsonUtils; import org.openmetadata.service.util.RestUtil; import org.openmetadata.service.util.ResultList; +import org.openmetadata.service.util.WebsocketNotificationHandler; import org.openmetadata.service.util.incidentSeverityClassifier.IncidentSeverityClassifierInterface; public class TestCaseResolutionStatusRepository @@ -324,6 +325,9 @@ public class TestCaseResolutionStatusRepository .withUpdatedAt(System.currentTimeMillis()); FeedRepository feedRepository = Entity.getFeedRepository(); feedRepository.create(thread); + + // Send WebSocket Notification + WebsocketNotificationHandler.handleTaskNotification(thread); } private void patchTaskAssignee(Thread originalTask, EntityReference newAssignee, String user) { @@ -334,7 +338,11 @@ public class TestCaseResolutionStatusRepository JsonPatch patch = JsonUtils.getJsonPatch(originalTask, updatedTask); FeedRepository feedRepository = Entity.getFeedRepository(); - feedRepository.patchThread(null, originalTask.getId(), user, patch); + RestUtil.PatchResponse thread = + feedRepository.patchThread(null, originalTask.getId(), user, patch); + + // Send WebSocket Notification + WebsocketNotificationHandler.handleTaskNotification(thread.entity()); } public void inferIncidentSeverity(TestCaseResolutionStatus incident) {