Class SocketService

src/main/java/com/chen/LeoBlog/service/SocketService.java:14–46

@Service
@Slf4j
public class SocketService {

    /**
     * 给指定用户发送信息
     *
     * @param session session
     * @param msg     发送的消息
     */
    public void sendMessage(Session session, String msg) {
        synchronized (session) {
            final RemoteEndpoint.Basic basic = session.getBasicRemote();
            if (basic == null)
                return;
            try {
                basic.sendText(msg);
            } catch (IOException e) {
                log.error("消息发送异常，异常情况: {}", e.getMessage());
            }
        }
    }

    public void sendMessageAll(String message, Long userId) {
        log.info("广播：群发消息");
        // 遍历map
        SocketPool.getSessionMap().forEach((keyId, session) -> {
            if (!userId.equals(keyId)) {
                sendMessage(session, message);
            }
        });
    }
}