当前位置： 首页 > news >正文

服务器主动发送响应？聊天模块如何实现？

news 2025/4/30 7:49:06

一、背景知识

当我们在某聊天界面中发送一个消息时，如A给B发送了一个消息，而B马上就收到，仔细思考会发现以下问题   ：

1. A给服务器发送请求，但服务器却给B发送了响应

2.B没有向服务器发送请求，却收到了响应

很明显，单纯的使用HTTP协议很难做到这一点，因此需要使用另一种应用层协议——WebSocket

二、了解WebSocket报文格式

2.1 简单了解websocket报文格式

 

1.FIN（ 结束标志位）：表示是否关闭连接；

2.RSV1~RSV2（保留位） ：为后续可能的功能扩展保留 

3.opcode（操作码）：有多个，用于判断如何处理数据（如1表示文本数据） 

4.MASK（掩码）：与数据安全相关

5. 1)Payload len（载荷）：7个bit，表示0-127字节

    2)Extended payload length（扩展载荷）：16个bit，

    3)Extend payload length continued（扩展载荷）：64个bit

（如果payload len小于126 ,模式1生效，如果等于126，模式2生效，等于127，模式3生效）

6.payload data（载荷）

2.2 了解WebSocket握手过程

WebSocket是由HTTP升级而成：

 在这之后就建立好了WebSocket连接，后续就通过WebSocket进行数据传输了

三、基于WebSocket编写代码

使用WebSocket，需要引入相关依赖：

<dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-websocket</artifactId>
</dependency>

3.1 编写服务器示例代码

编写服务器代码主要有以下几步：

1. 创建一个类继承TextWebSocketHandler（处理通信流程）；

2. 重写afterConnectionEstablished、handleTextMessage、handleTransportError、afterConnectionClosed四个方法；

3.实现WebSocketConfigurer接口，重写方法并配置路由（关联对应Handler的路径）；

一、继承TextWebSockeHandler并从写上述4个方法：

@Component
public class TestWebSocketController extends TextWebSocketHandler {@Overridepublic void afterConnectionEstablished(WebSocketSession session) throws Exception {//这个方法会在websocket连接建立之后，自动调用System.out.println("TestSocket 连接成功");}@Overrideprotected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {//websocket收到消息时自动调用System.out.println("TestSocket 收到消息" + message.toString());//session是个会话，记录了通信双方是谁session.sendMessage(message);}@Overridepublic void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {//连接异常时自动调用System.out.println("TestSocket 连接异常");}@Overridepublic void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {//连接正常关闭后，自动调用System.out.println("Test Socket 连接正常关闭");}
}

其中，handleTextMessage是这几个方法的重点，主要的代码逻辑都是在这里实现，形参message表示载荷中的数据，需要做的就是将这里的message发送给通信双方的另一方

二、实现WebSocketConfigurer接口，配置路由：
 

@Configuration
@EnableWebSocket//启动websocket
public class WebSocketConfig implements WebSocketConfigurer {@Autowiredprivate TestWebSocketController testWebSocketController;@Autowiredprivate WebSocketController webSocketController;@Overridepublic void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
//        通过这个方法，创建好的handler类注册到具体的路径
//        当浏览器，websocket请求路径时“/test”时，就会调用到TestWebSocketController这个类里的方法registry.addHandler(testWebSocketController,"/test");}
}

3.2 编写前端示例代码

 

<!DOCTYPE html>
<html lang="en"><head><meta charset="UTF-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><title>websocket 测试</title>
</head><body><input type="text" id="message"><button id="send-button">发送</button><script>//编写js使用websocket的代码//创建一个websocket实例let websocket = new WebSocket("ws://127.0.0.1:8080/test");//给websocket注册回调函数websocket.onopen = function () {//连接建立完成后，自动执行console.log("websocket 连接成功");}websocket.onclose = function () {//连接断开后，自动执行console.log("websocket 连接断开");}websocket.onerror = function () {//连接异常时，自动执行console.log("websocket 连接异常");}websocket.onmessage = function (e) {//收到消息时，自动执行console.log("websocket 收到消息" + e.data);}let messageInput = document.querySelector('#message');let sendButton = document.querySelector('#send-button');sendButton.onclick = function() {console.log("websocket 发送消息：" + messageInput.value);websocket.send(messageInput.value); }</script>
</body></html>

3.3 效果展示

四、基于WebSocket实现网页聊天项目的消息转发

这里的网页聊天项目和网页聊天相似，如：

具体实现分为以下几步：

一、约定前后端交互接口

二、实现TextWebSocketHandler接口

1> 服务器消息转发逻辑分析

上图了解到，每个会话的好友可能是一个，也可能是多个，并且未上线的好友无法接收消息，而服务器转发消息，需要知道对端的WebSocketSession,因此，可以使用map存储上线用户的userId 和 webSocketSession

1.用户上线，map插入值；

2.用户下线，map 删除值；

@Component
@Slf4j
public class OnLineUserManager {//此处的 哈希表 要考虑线程安全问题private ConcurrentHashMap<Integer, WebSocketSession> sessions = new ConcurrentHashMap<>();// 1) 用户上线，哈希表插入值public void online(Integer userId,WebSocketSession session){if(sessions.get(userId) != null){//说明用户已经上线，登录失败log.info("[" + userId + "] 登录失败，已在其它地方登录！");return;}sessions.put(userId,session);log.info("[" + userId + "] 上线！");}// 2）用户下线，哈希表删除值public void offline(Integer userId,WebSocketSession session){ WebSocketSession exitSession = sessions.get(userId);if(exitSession == session){//两个session是同一个，才下线sessions.remove(userId);log.info("[" + userId + "] 下线！");}}// 3）根据userId 获取 WebSocketSessionpublic WebSocketSession getSession(Integer userId){return sessions.get(userId);}
}

2>实现 TextWebSocketHandler 接口，重写4个方法

@Slf4j
@Component
public class WebSocketController extends TextWebSocketHandler {@Autowiredprivate OnLineUserManager onLineUserManager;private ObjectMapper objectMapper = new ObjectMapper();@Autowiredprivate MessageMapper messageMapper;@Autowiredprivate SessionMapper sessionMapper;@Overridepublic void afterConnectionEstablished(WebSocketSession session) throws Exception {log.info("websocket 连接成功");Integer userId = (Integer) session.getAttributes().get("session_userId");if(userId == null){return;}onLineUserManager.online(userId,session);}@Overrideprotected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {log.info("websocket 收到消息" + message.toString());//1.获取用户信息Integer userId = (Integer) session.getAttributes().get("session_userId");String username = (String) session.getAttributes().get("session_username");if(userId == null || username == null){log.info("用户未登录,无法转发");return;}//2.解析请求,注意把json字符串转化为java对象MessageRequest req = objectMapper.readValue(message.getPayload(), MessageRequest.class);if(req.getType().equals("message")){//转发消息transferMessage(userId,username,req);}else {log.info("req.type 有误:" + message.getPayload());}}//通过这个方法进行转发private void transferMessage(Integer fromId, String fromName,MessageRequest req) throws IOException {//1.构造待转发的响应对象MessageResponse messageResponse = new MessageResponse("message",fromId,fromName, req.getSessionId(), req.getContent());//转成字符串String responseJson = objectMapper.writeValueAsString(messageResponse);log.info("转发消息：responseJson" + responseJson);//2.根据请求中的sessionId ,获取到message_session中有那些用户（查询数据库）List<Friend> friends = sessionMapper.selectFriendBySessionId(req.getSessionId(),fromId);//3.遍历users，通过OnLineUserManager获取到websocketsession发送响应,// ！！！也要给自己发送一个消息,因此需要将自己也加入friends,让自己可以看到自己发送的消息Friend myself = new Friend();myself.setFriendId(fromId);myself.setFriendName(fromName);friends.add(myself);for (Friend friend:friends) {WebSocketSession webSocketSession = onLineUserManager.getSession(friend.getFriendId());if(webSocketSession == null){//用户不在线，不发送continue;}webSocketSession.sendMessage(new TextMessage(responseJson));}//4.转发的消息需要存储在数据库,这样下线之后，重新上线可以获取历史消息//向message表写入记录Message message = new Message();message.setFromId(fromId);message.setSessionId(req.getSessionId());message.setContent(req.getContent());messageMapper.insertMessage(message);}@Overridepublic void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {log.info("websocket 连接异常" + exception.toString());Integer userId = (Integer) session.getAttributes().get("session_userId");if(userId == null){return;}onLineUserManager.offline(userId,session);}@Overridepublic void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {log.info("websocket 连接正常关闭" + status.toString());Integer userId = (Integer) session.getAttributes().get("session_userId");if(userId == null){   return;}onLineUserManager.offline(userId,session);}
}

3> 实现 WebSocketConfigurer 接口，配置路由

@Configuration
@EnableWebSocket//启动websocket
public class WebSocketConfig implements WebSocketConfigurer {@Autowiredprivate TestWebSocketController testWebSocketController;@Autowiredprivate WebSocketController webSocketController;@Overridepublic void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {registry.addHandler(webSocketController,"/WebSocketMessage")//！！！由于userId 和 username 都是保存在httpSession中的，无法直接在webSocketSession中使用，需要添加以下带代码.addInterceptors(new HttpSessionHandshakeInterceptor());}
}

4> 测试结果

http://www.xdnf.cn/news/216073.html

相关文章：

  • 【Vue3/Typescript】合并多个pdf并预览打印，兼容低版本浏览器
  • CentOS NFS共享目录
  • 【GESP】C++三级练习 luogu-B2118 验证子串
  • 后验概率最大化（MAP）估计算法原理以及相具体的应用实例附C++代码示例
  • 源码编译安装LAMP
  • Python 3.12数据结构与算法革命
  • 实现使用Lucene对某个信息内容进行高频词提取并输出
  • 2025年04月29日Github流行趋势
  • TA学习之路——2.4 图形传统光照模型详解
  • HCIE证书失效？续证流程与影响全解析
  • Java 高级技术之Gradle
  • Ubuntu实现远程文件传输
  • C 语言 static 与 extern 详解
  • 海思SD3403边缘计算AI核心设备概述
  • 2025年欧洲西南部大停电
  • H3C ER3208G3路由实现内网机器通过公网固定IP访问内网服务器
  • 电流探头的消磁与直流偏置校准
  • 深入了解僵尸网络 IP：威胁与防范
  • Redis核心与底层实现场景题深度解析
  • 生物化学笔记：神经生物学概论04 视觉通路简介视网膜视网膜神经细胞大小神经节细胞(视错觉)
  • 故障诊断——复现github代码ClassBD-CNN（BDCNN）
  • BT136-ASEMI无人机专用功率器件BT136
  • 超详细复现—平抑风电波动的电-氢混合储能容量优化配置
  • python入门:找出字典中key和value不相同的部分,并替换成新的value
  • Makefile 在 ARM MCU 开发中的编译与链接参数详解与实践
  • rsync命令详解与实用案例
  • 【Unity】使用Socket建立客户端和服务端并进行通信的例子
  • 学习笔记：Qlib 量化投资平台框架 — MAIN COMPONENTS Part Ⅱ
  • Spring Boot 缓存机制：从原理到实践
  • HTML 表单