wangwei 5 vuotta sitten
vanhempi
commit
55d8d67b10

+ 4 - 4
src/main/java/cn/com/qmth/examcloud/ws/core/MessageHandlerHolder.java

@@ -27,8 +27,8 @@ public class MessageHandlerHolder implements ApplicationRunner {
 
 	private static Map<String, Method> map = Maps.newConcurrentMap();
 
-	public static Method getMethod(String path) {
-		return map.get(path);
+	public static Method getMethod(String mapping) {
+		return map.get(mapping);
 	}
 
 	@Override
@@ -51,8 +51,8 @@ public class MessageHandlerHolder implements ApplicationRunner {
 				}
 				String valueOfMethod = messageHandlerOfMethod.value();
 
-				String path = join(valueOfClass, valueOfMethod);
-				map.put(path, method);
+				String mapping = join(valueOfClass, valueOfMethod);
+				map.put(mapping, method);
 			}
 
 		}

+ 6 - 5
src/main/java/cn/com/qmth/examcloud/ws/core/WebSocketServerEndpoint.java

@@ -7,18 +7,19 @@ import javax.websocket.OnError;
 import javax.websocket.OnMessage;
 import javax.websocket.OnOpen;
 import javax.websocket.Session;
+import javax.websocket.server.PathParam;
 import javax.websocket.server.ServerEndpoint;
 
 import org.springframework.stereotype.Component;
 
 import cn.com.qmth.examcloud.commons.util.DateUtil;
 
-@ServerEndpoint("/ws")
+@ServerEndpoint("{mapping}")
 @Component
 public class WebSocketServerEndpoint {
 
 	@OnOpen
-	public void onOpen(Session session) {
+	public void onOpen(Session session, @PathParam("mapping") String mapping) {
 		System.out.println("onOpen; session=" + session);
 		try {
 			session.getBasicRemote().sendText(DateUtil.chinaNow() + "| fuck");
@@ -28,12 +29,12 @@ public class WebSocketServerEndpoint {
 	}
 
 	@OnClose
-	public void onClose(Session session) {
+	public void onClose(Session session, @PathParam("mapping") String mapping) {
 		System.out.println("onClose; session=" + session);
 	}
 
 	@OnMessage
-	public void onMessage(Session session, String message) {
+	public void onMessage(Session session, @PathParam("mapping") String mapping, String message) {
 		System.out.println("onMessage; session=" + session);
 		System.out.println("onMessage; message=" + message);
 		try {
@@ -44,7 +45,7 @@ public class WebSocketServerEndpoint {
 	}
 
 	@OnError
-	public void onError(Session session, Throwable t) {
+	public void onError(Session session, @PathParam("mapping") String mapping, Throwable t) {
 		System.out.println("onError; session=" + session);
 		t.printStackTrace();
 	}