You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
983 B
30 lines
983 B
package com.mortal.bigmodelr.config;
|
|
|
|
import org.springframework.web.socket.TextMessage;
|
|
import org.springframework.web.socket.WebSocketSession;
|
|
import org.springframework.web.socket.handler.TextWebSocketHandler;
|
|
import java.io.IOException;
|
|
import java.util.List;
|
|
import java.util.concurrent.CopyOnWriteArrayList;
|
|
|
|
public class MyWebSocketHandler extends TextWebSocketHandler {
|
|
private static final List<WebSocketSession> sessions = new CopyOnWriteArrayList<>();
|
|
|
|
@Override
|
|
public void afterConnectionEstablished(WebSocketSession session) {
|
|
sessions.add(session);
|
|
}
|
|
|
|
// 向所有客户端推送数据
|
|
public static void broadcast(String message) {
|
|
for (WebSocketSession session : sessions) {
|
|
try {
|
|
if (session.isOpen()) {
|
|
session.sendMessage(new TextMessage(message));
|
|
}
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
}
|
|
}
|