websocket

.gitignore

@@ -33,3 +33,4 @@ build/
 .vscode/
 /images/
 /jdk/
+./*Test.java

src/main/java/com/scbfkj/uni/api/WebSocketServer.java

@@ -8,6 +8,7 @@ import jakarta.websocket.server.ServerEndpoint;
 import org.springframework.stereotype.Component;
 
 import java.io.IOException;
+import java.util.HashMap;
 import java.util.Map;
 
 @ServerEndpoint("/ws")
@@ -59,15 +60,21 @@ public class WebSocketServer {
     @OnMessage
     public void onMessage(String message) throws IOException {
         System.out.println(message);
-        String result;
+        Map<String, Object> result = new HashMap<>();
         try {
-            Map<String, Object> map = new DataProcessService().process((Map<String, Object>) DataFormatUtil.toMap(message));
-            result = DataFormatUtil.toString(map);
+            Map<?, ?> body = DataFormatUtil.toMap(message);
+            try {
+
+                result = new DataProcessService().process((Map<String, Object>) body);
+            } catch (Exception e) {
+                result = UniReturnUtil.fail(e);
+            }
+            result.put("request", body);
         } catch (Exception e) {
-            result = DataFormatUtil.toString(UniReturnUtil.fail(e));
+            result.put("request", message);
         }
 
-        session.getBasicRemote().sendText(result);
+        session.getBasicRemote().sendText(DataFormatUtil.toString(result));
 
     }