基础知识可见我的上一篇帖子
准备工作:
依赖:
<!-- 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方法即可