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

(mq) Fix bug in notice handling where they were registered on the wrong name

Browse Source
This commit is contained in:
Viktor Lofgren 2023-08-05 14:44:16 +02:00
parent bf37a3eb25
commit 715d61dfea
1 changed files with 19 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
code/common/service/src/main/java/nu/marginalia/service/server/mq/ServiceMqSubscription.java
View File

@ -18,29 +18,40 @@ public class ServiceMqSubscription implements MqSubscription {
private final Map<String, Method> notifications = new HashMap<>();
private final Service service;
public ServiceMqSubscription(Service service) {
this.service = service;
/* Wire up all methods annotated with @MqRequest and @MqNotification
* to receive corresponding messages from this subscription */
for (var method : service.getClass().getMethods()) {
var annotation = method.getAnnotation(MqRequest.class);
if (annotation != null) {
requests.put(annotation.endpoint(), method);
}
if (method.getAnnotation(MqNotification.class) != null) {
notifications.put(method.getName(), method);
}
for (var method : service.getClass().getMethods()) {
var annotation = method.getAnnotation(MqNotification.class);
if (annotation != null) {
notifications.put(annotation.endpoint(), method);
}
}
}
@Override
public boolean filter(MqMessage rawMessage) {
boolean isInteresting = requests.containsKey(rawMessage.function())
|| notifications.containsKey(rawMessage.function());
if (!isInteresting) {
logger.warn("Received message for unknown function " + rawMessage.function());
if (requests.containsKey(rawMessage.function())) {
return true;
}
if (notifications.containsKey(rawMessage.function())) {
return true;
}
return isInteresting;
logger.warn("Received message for unknown function " + rawMessage.function());
return false;
}
@Override