WebSocket

背景

在实现聊天室项目的时候,我遇到了一个问题,那就是消息是如何从服务器端自己推送给客户端的呢?一开始我的想法是每隔一段时间,就让客户端向服务器端请求数据,也就是所谓的"轮询"机制,但这种方式显然是太耗费资源的,我在查阅资料后,发现有一个应用层协议专门服务于这种情况的:WebSocket。 (依旧是基于传输层的TCP实现的)

报文格式

websocket握手过程

代码示例

先来编写服务器端的代码:

1)创建一个类,作为WebSocketHandler(处理websocket中的各个通信流程)

具体的业务场景:

java 复制代码
@Component
public class WebSocketComponent extends TextWebSocketHandler {

    @Autowired
    OnlineUserManager onlineUserManager;
    @Autowired
    private ObjectMapper objectMapper;

    @Autowired
    MessageMapper messageMapper;

    @Autowired
    MessageSessionMapper messageSessionMapper;

    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        UserInfo userInfo = (UserInfo) session.getAttributes().get("SESSION_USER");
        if(userInfo == null){
            return;
        }
        System.out.println("get websocket session -> "+userInfo.getId());
        onlineUserManager.online(userInfo.getId(),session);
    }

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        UserInfo userInfo = (UserInfo) session.getAttributes().get("SESSION_USER");
        if(userInfo == null){
            return;
        }

        MessageRequest request = objectMapper.readValue(message.getPayload(),MessageRequest.class);
        if(request.getType().equals("message")){
            transferMessage(userInfo,request);
        }else{
            System.out.println("ERROR : MessageRequest:Type is error!");
        }
    }

    private void transferMessage(UserInfo userInfo, MessageRequest request) throws IOException {
        //先构造一个response 并且转成String类型
        MessageResponse messageResponse = new MessageResponse();
        messageResponse.setContent(request.getContent());
        messageResponse.setFromid(userInfo.getId());
        messageResponse.setFromname(userInfo.getUsername());
        messageResponse.setSessionid(request.getSessionId());
        String resJson = objectMapper.writeValueAsString(messageResponse);
        //通过sessionId 来获取好友列表 再遍历这个链表 进行传输信息
        List<Friend> list = messageSessionMapper.getFriendsBySID(request.getSessionId(),userInfo.getId());
        Friend mySelf = new Friend();
        mySelf.setFriendname(userInfo.getUsername());
        mySelf.setFriendId(userInfo.getId());
        list.add(mySelf);
        for(Friend friend : list){
            WebSocketSession webSocketSession = onlineUserManager.getSession(friend.getFriendId());
            if(webSocketSession == null){
                // 这种情况是未登录的状态 先不发送
                continue;
            }else{
                webSocketSession.sendMessage(new TextMessage(resJson));
            }
        }
        // 还要把这个消息 发给 数据库 进行永久保存操作
        Message message = new Message();
        message.setFromid(userInfo.getId());
        message.setSessionid(request.getSessionId());
        message.setContent(request.getContent());
        messageMapper.add(message);
    }

    @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
        UserInfo userInfo = (UserInfo) session.getAttributes().get("SESSION_USER");
        if(userInfo == null){
            return;
        }
        onlineUserManager.offline(userInfo.getId(),session);
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        UserInfo userInfo = (UserInfo) session.getAttributes().get("SESSION_USER");
        if(userInfo == null){
            return;
        }
        onlineUserManager.offline(userInfo.getId(),session);
    }
}

2)把上述类的实例,给注册到Spring里面,配置路由。(关联上哪个路径对应到上述的handler)

java 复制代码
@Configuration
@EnableWebSocket    //启动websocket
public class WebSocketConfig implements WebSocketConfigurer {
    @Autowired
    WebSocketComponent webSocketComponent;

    //通过这个方法 进行路由注册
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(webSocketComponent,"websocketmessage")
                .addInterceptors(new HttpSessionHandshakeInterceptor());
    }
}

写客户端的代码:

相关推荐
cui_win16 分钟前
【网络】Linux 内核优化实战 - net.ipv4.tcp_congestion_control
linux·网络·tcp/ip
yqcoder34 分钟前
12. 说一下 https 的加密过程
网络协议·http·https
roboko_1 小时前
TCP详解——流量控制、滑动窗口
服务器·网络·tcp/ip
长弓三石2 小时前
鸿蒙网络编程系列57-仓颉版固定包头可变包体解决TCP粘包问题
网络·tcp/ip·harmonyos
FileLink跨网文件交换3 小时前
跨网文件交换?内外网文件交换十大方法构建安全合规的数据传输通道
运维·服务器·网络
️️(^~^)3 小时前
静态路由综合配置实验报告
服务器·网络·计算机网络·智能路由器
Andy杨4 小时前
20250710-2-Kubernetes 集群部署、配置和验证-网络组件存在的意义?_笔记
网络·笔记·kubernetes
沫儿笙4 小时前
弧焊机器人气体全方位节能指南
网络·人工智能·机器人
车载测试工程师5 小时前
汽车功能安全-嵌入式软件测试(软件合格性测试)【目的、验证输入、集成&验证要求】11
功能测试·网络协议·测试工具·安全·车载系统·汽车·测试覆盖率
IPdodo全球网络服务6 小时前
什么是IP关联?跨境卖家如何有效避免IP关联?
网络·网络协议·tcp/ip