1 files changed, 14 insertions, 12 deletions
diff --git a/src/main/java/me/imsonmia/epqapi/controller/ChatMessageController.java b/src/main/java/me/imsonmia/epqapi/controller/ChatMessageController.java
index bf94991..7bfcba6 100644
--- a/src/main/java/me/imsonmia/epqapi/controller/ChatMessageController.java
+++ b/src/main/java/me/imsonmia/epqapi/controller/ChatMessageController.java
@@ -1,19 +1,21 @@
 package me.imsonmia.epqapi.controller;
 
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.PathVariable;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.messaging.handler.annotation.MessageMapping;
+import org.springframework.messaging.handler.annotation.SendTo;
+import org.springframework.stereotype.Controller;
 
-import me.imsonmia.epqapi.model.ChatMessage;
-import me.imsonmia.epqapi.repository.ChatMessageRepository;
+import me.imsonmia.epqapi.model.Message;
 
-@RestController
-@RequestMapping("/api/v1")
+@Controller
+// @RequestMapping("/api/v1")
 public class ChatMessageController {
-    private ChatMessageRepository chatMessageRepository;
-    @GetMapping("/msg/{id}")
-    public ChatMessage getMessageById(@PathVariable(value = "id") Long id) {
-        return chatMessageRepository.findById(id).get();
+    @MessageMapping("/chat")
+    @SendTo("/sub/chat")
+    public Message messageHandler(Message message) throws Exception {
+        return message;
     }
+    // @GetMapping("/msg/{id}")
+    // public ChatMessage getMessageById(@PathVariable(value = "id") Long id) {
+    //     return chatMessageRepository.findById(id).get();
+    // }
 }