add message handling

asyncapi · Feb 7, 2024 · 6723e88 · 6723e88
1 parent a73c1ef
commit 6723e88
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/template/src/main/java/com/asyncapi/service/MessageHandlerService.java b/template/src/main/java/com/asyncapi/service/MessageHandlerService.java
@@ -15,6 +15,9 @@
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.messaging.Message;
 import org.springframework.stereotype.Service;
+{%- if asyncapi | isProtocol('ws') and hasPublish %}
+import org.springframework.messaging.handler.annotation.MessageMapping;
+{%- endif %}
 {%- if asyncapi | isProtocol('kafka') and hasPublish %}
 import org.springframework.kafka.annotation.KafkaListener;
 import org.springframework.kafka.support.KafkaHeaders;
@@ -115,6 +118,11 @@ public class MessageHandlerService {
         // parametrized listener
     }
         {%- endif %}
+    {% elif asyncapi | isProtocol('ws') %}
+    @MessageMapping("/{{channelName}}")
+    public void handle{{methodName | upperFirst}}({{typeName}} payload) {
+        LOGGER.info("Message received from {{- channelName -}} : " + payload);
+    }
     {%- else %}
         {%- if hasParameters %}
     @Value("${mqtt.topic.{{-methodName-}}}")