实现一个简单的webSocket聊天demo

基础知识可见我的上一篇帖子

准备工作：

依赖：

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

消息实体数据库表

DROP TABLE IF EXISTS `message`;
CREATE TABLE `message`  (
  `id` int(11) UNSIGNED ZEROFILL NOT NULL AUTO_INCREMENT COMMENT '主键',
  `from` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '记录谁发送的消息',
  `to` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '记录消息发送给谁',
  `msg` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '消息主体',
  `date` datetime NULL DEFAULT NULL COMMENT '消息发送的时间',
  `type` int(11) NULL DEFAULT NULL COMMENT '消息发送的类型，0系统群发，1用户私聊',
  `is_read` int(11) NULL DEFAULT NULL COMMENT '消息是否已读，0未读，1已读',
  PRIMARY KEY (`id`) USING BTREE
) ENGINE = InnoDB AUTO_INCREMENT = 1 CHARACTER SET = utf8 COLLATE = utf8_general_ci ROW_FORMAT = Dynamic;

SET FOREIGN_KEY_CHECKS = 1;

开始构建项目

1. 新建一个springboot项目chat_demo,并使用mybatisx插件快速构建项目

 2. 创建webSocket配置类，即注册ServerEndpointExporter，该bean用于扫描被@ServerEndpoint注解的类，并将其作为服务端

3. 创建被@ServerEndpoint注解的类用于作为webSocket服务器并指定连接的uri

4.使用注解标记对应的方法,，并启动程序测试是否可用（注意此处的map应该改为static类变量）

 5. webSocket在线测试网站Websocket在线测试 (stackoverflow.org.cn)

 

 

成功完成基础的websocket连接，可以开始编辑业务逻辑

实现基础的在线聊天（不包括消息持久化）

用来发送消息的方法，参数分别为接收消息的用户的session，和对应的消息
private void sendMessage(Session toSession,String msg){
    try {
        toSession.getBasicRemote().sendText(msg);
    } catch (IOException e) {
        throw new RuntimeException(e);
    }
}

改造方法

    /**]
     * 用户连接成功后将用户名和其对应的session存入map中
     * @param session
     * @param username
     */
    @OnOpen
    public void onOpen(Session session, @PathParam(value = "username") String username) {
        log.info("用户{}连接已创建",username);
        sessionMap.put(username,session);
    }
    /**
     * 所有用户发送的消息都会来到此处，我们需要将消息中转，发送给对应的用户
     * @param msg
     * @param username
     */
    @OnMessage
    public void onMessage(String msg, @PathParam("username") String username) {
        log.info("收到用户{}发来的消息：{}",username,msg);
        Message message = JSON.parseObject(msg, Message.class);
        //根据message中的to属性获取接收消息的用户的session，利用其session将消息转发过去
        Session toSession = sessionMap.get(message.getTo());
            sendMessage(toSession, message.getMsg());
    }
    /**
     * 用户断开连接，将其从sessionMap中移除
     * @param session
     * @param username
     */
    @OnClose
    public void onClose(Session session,@PathParam("username") String username) {
        log.info("用户{}已经关闭连接",username);
        sessionMap.remove(username);
    }

新建三个websocket在线测试的窗口创建三个链连接 

构建后台message实体类的json格式数据并发送

 只有李四接收到消息

 一个简单的无持久化的在线聊天后台就写好了，，其余的群发，多人发送，之类的都是同一个道理，获取session调用send方法即可