FIX #15956 - Send web socket notification on incident assignment/ack (#16086)

* fix: send websocket notification on incident assignment

* style: ran java linting
This commit is contained in:
Teddy 2024-04-30 14:17:06 +02:00 committed by GitHub
parent 89b083b6f2
commit 91f930d4c4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -40,6 +40,7 @@ import org.openmetadata.service.util.EntityUtil;
import org.openmetadata.service.util.JsonUtils; import org.openmetadata.service.util.JsonUtils;
import org.openmetadata.service.util.RestUtil; import org.openmetadata.service.util.RestUtil;
import org.openmetadata.service.util.ResultList; import org.openmetadata.service.util.ResultList;
import org.openmetadata.service.util.WebsocketNotificationHandler;
import org.openmetadata.service.util.incidentSeverityClassifier.IncidentSeverityClassifierInterface; import org.openmetadata.service.util.incidentSeverityClassifier.IncidentSeverityClassifierInterface;
public class TestCaseResolutionStatusRepository public class TestCaseResolutionStatusRepository
@ -324,6 +325,9 @@ public class TestCaseResolutionStatusRepository
.withUpdatedAt(System.currentTimeMillis()); .withUpdatedAt(System.currentTimeMillis());
FeedRepository feedRepository = Entity.getFeedRepository(); FeedRepository feedRepository = Entity.getFeedRepository();
feedRepository.create(thread); feedRepository.create(thread);
// Send WebSocket Notification
WebsocketNotificationHandler.handleTaskNotification(thread);
} }
private void patchTaskAssignee(Thread originalTask, EntityReference newAssignee, String user) { private void patchTaskAssignee(Thread originalTask, EntityReference newAssignee, String user) {
@ -334,7 +338,11 @@ public class TestCaseResolutionStatusRepository
JsonPatch patch = JsonUtils.getJsonPatch(originalTask, updatedTask); JsonPatch patch = JsonUtils.getJsonPatch(originalTask, updatedTask);
FeedRepository feedRepository = Entity.getFeedRepository(); FeedRepository feedRepository = Entity.getFeedRepository();
feedRepository.patchThread(null, originalTask.getId(), user, patch); RestUtil.PatchResponse<Thread> thread =
feedRepository.patchThread(null, originalTask.getId(), user, patch);
// Send WebSocket Notification
WebsocketNotificationHandler.handleTaskNotification(thread.entity());
} }
public void inferIncidentSeverity(TestCaseResolutionStatus incident) { public void inferIncidentSeverity(TestCaseResolutionStatus incident) {