Mirror/MarginaliaSearch
1
0
Fork 0
mirror of https://github.com/MarginaliaSearch/MarginaliaSearch.git synced 2025-02-24 05:18:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(mq single shot inbox) Flag messages as OK if there is no recipient

Browse Source
This commit is contained in:
Viktor Lofgren 2023-07-28 12:04:23 +02:00
parent 92cac52813
commit 27e781761d
1 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
code/common/message-queue/src/main/java/nu/marginalia/mq/inbox/MqSingleShotInbox.java
View File

@ -2,6 +2,7 @@ package nu.marginalia.mq.inbox;
import lombok.SneakyThrows; import lombok.SneakyThrows;
import nu.marginalia.mq.MqMessage; import nu.marginalia.mq.MqMessage;
import nu.marginalia.mq.MqMessageState;
import nu.marginalia.mq.persistence.MqPersistence; import nu.marginalia.mq.persistence.MqPersistence;
import java.sql.SQLException; import java.sql.SQLException;
@ -71,9 +72,24 @@ public class MqSingleShotInbox {
return Optional.empty(); return Optional.empty();
} }
/** Send a response to the specified message. If the original message has no response inbox,
* the original message will be marked as OK instead.
*
* @param originalMessage The original message
* @param response The response
*/
public void sendResponse(MqMessage originalMessage, MqInboxResponse response) { public void sendResponse(MqMessage originalMessage, MqInboxResponse response) {
try { try {
persistence.sendResponse(originalMessage.msgId(), response.state(), response.message()); if (!originalMessage.expectsResponse()) {
// If the original message doesn't expect a response, we can just mark it as OK,
// since the sendResponse method will fail explosively since it can't insert a response
// to a non-existent inbox.
persistence.updateMessageState(originalMessage.msgId(), MqMessageState.OK);
}
else {
persistence.sendResponse(originalMessage.msgId(), response.state(), response.message());
}
} catch (SQLException e) { } catch (SQLException e) {
throw new RuntimeException(e); throw new RuntimeException(e);
} }