SpringBoot+Websocket实现即时通讯
ztj100 2025-01-19 02:00 58 浏览 0 评论
1. 介绍
基于Spring Boot和websocket实现点对点在线聊天和简单的机器人自动回复功能，学生可选择在线空闲的老师咨询，无在线老师接入机器人根据学生提问的内容和问题编号从问题库中获取对应的答案来回复学生。
2.后台
2.1. 引入依赖
Bash
// https://mvnrepository.com/artifact/org.springframework.boot/spring-boot-starter-websocket
implementation group: 'org.springframework.boot', name: 'spring-boot-starter-websocket', version: '2.0.4.RELEASE'
2.2. websocket 配置类
WebsocketConfig.java
Bash
@Configuration
public class WebSocketConfig {
@Bean
public ServerEndpointExporter serverEndpointExporter() {
return new ServerEndpointExporter();
}
}
2.3. 封装显示消息对象
@Data
public class MessageVo {
@ApiModelProperty(value = "用户id")
private Integer userId;
@ApiModelProperty(value = "用户名")
private String username;
@ApiModelProperty(value = "消息内容")
private String msg;
@ApiModelProperty(value = "在线人数")
private int count;
}
2.4. 封装进行对话的对象
用来存放会话的用户信息
@Data
@ApiModel(description = "websocket会话对象")
public class SocketUserInfo extends BaseEntity {
//用户sessionId
private String sessionId;
//用户session
private Session session;
//目标用户sessionid
private String targetSessionId;
//用户角色
private String userRole;
//用户id
private Integer userId;
//用户名
private String userName;
//联系电话
private String tel;
2.5. websocket处理类
@Slf4j
@Component
@ServerEndpoint(value = "/groupChat/{targetSessionId}/{userId}",configurator = SpringContextUtil.class)
public class WebSocketServerController{
//用本地线程保存session
private static ThreadLocal<Session> sessions = new ThreadLocal<Session>();
//保存所有连接上的用户的session sessionId作为key
private static Map<String, SocketUserInfo> userSessionMap = new ConcurrentHashMap<>();
//保存老师的session
private static Map<String, SocketUserInfo> serverSessionMap = new ConcurrentHashMap<>();
//保存在线的学生
private static List<Map<String,Object>> studentUserMapList = new ArrayList<>();
//保存在线空闲的老师信息
private static List<Map<String,Object>> teacherUserMapList = new ArrayList<>();
@Autowired
private CacheProxy cacheProxy;
@Autowired
private BuiProblemAnswerServiceImpl buiProblemAnswerService;
/**
* 建立连接调用的方法，群成员加入
*
* @param session
* @param targetSessionId 学生咨询的对象的session id
*/
@OnOpen
public void onOpen(Session session,
@PathParam("targetSessionId") String targetSessionId,
@PathParam("userId") Integer userId) {
try {
MessageVo messageVo = new MessageVo();
//保证各个线程里的变量相对独立于其他线程内的变量
sessions.set(session);
//获取用户信息
SysLoginUserServiceImpl sysLoginUserService = SpringContextUtil.getBean(SysLoginUserServiceImpl.class);
Map<String, Object> map = sysLoginUserService.queryUserInfoById(userId);
//老师上线
if ("10000".equals(targetSessionId)){
//如果是老师和管理员角色,创建一个在线聊天信息
SocketUserInfo serverInfo = new SocketUserInfo();
serverInfo.setUserRole("老师");
serverInfo.setSession(session);
serverInfo.setUserId(userId);
serverInfo.setSessionId(session.getId());
serverInfo.setUserName(StrUtil.clearNull(map.get("userName")));
serverInfo.setTerminalCode(StrUtil.clearNull(map.get("tel")));
map.put("sessionId",session.getId());
//将老师信息存入到list中
teacherUserMapList.add(map);
//将在线老师信息写入缓存
cacheProxy.set("websocket",teacherUserMapList);
messageVo.setMsg(session.getId()+"您已经上线了");
messageVo.setUserId(userId);
sendMsg(session,messageVo);
//将老师的信息保存到map中
serverSessionMap.put(session.getId(),serverInfo);
}
if (!"10000".equals(targetSessionId)){//学生上线 创建一个在线学生信息
SocketUserInfo userInfo = new SocketUserInfo();
userInfo.setSessionId(session.getId());
userInfo.setSession(session);
userInfo.setUserRole("学生");
userInfo.setUserId(userId);
userInfo.setUserName(StrUtil.clearNull(map.get("userName")));
userInfo.setTerminalCode(StrUtil.clearNull(map.get("tel")));
//获取对应的老师信息
if (!targetSessionId.equals("kefu")){
SocketUserInfo serverInfo = serverSessionMap.get(targetSessionId);
//将该老师的连接设置为当前学生
serverInfo.setTargetSessionId(session.getId());
//将当前学生的连接对象设置为该老师
userInfo.setTargetSessionId(targetSessionId);
//将当前咨询的老师从空闲的列表中删除
for (int i = 0; i < teacherUserMapList.size(); i++) {
Map<String, Object> teacherMap = teacherUserMapList.get(i);
if (teacherMap.get("sessionId").equals(targetSessionId)){
teacherUserMapList.remove(i);
}
}
//更新可咨询的老师列表
cacheProxy.set("websocket",teacherUserMapList);
//给学生发送一条信息
messageVo.setMsg("老师"+serverInfo.getUserName()+"正在为你答疑！");
messageVo.setCount(2);
messageVo.setUserId(serverInfo.getUserId());
sendMsg(userInfo.getSession(), messageVo);
//给老师发送一条信息
messageVo.setMsg("正在为学生"+userInfo.getSessionId()+"答疑！");
messageVo.setUserId(userInfo.getUserId());
sendMsg(serverInfo.getSession(), messageVo);
}
if ("kefu".equals(targetSessionId)){
userInfo.setTargetSessionId("kefu");
messageVo.setMsg("请根据右侧的列表输入要您咨询的问题编号！！！");
sendMsg(session,messageVo);
}
//将在线用户信息保存到map中
userSessionMap.put(session.getId(), userInfo);
}
}catch (Exception e){
e.printStackTrace();
}
}
/**
* 收到消息调用的方法，群成员发送消息
*
* @param message
* @param session
*/
@OnMessage
public void onMessage(String message, Session session){
HashMap<String, String> result = new HashMap<>();
MessageVo messageVo = new MessageVo();
SocketUserInfo serverInfo = serverSessionMap.get(session.getId());
if (serverInfo != null) {//老师消息
log.info("老师"+ session.getId()+"发送消息：\""+ message +"\"给学生"+serverSessionMap.get(session.getId()).getTargetSessionId());
result.put("msg", "老师"+session.getId()+"："+message);
messageVo.setMsg(message);
messageVo.setUserId(serverInfo.getUserId());
messageVo.setCount(2);
//要判断是否绑定到的学生如果有就将消息传递到学生
if (null != serverInfo.getTargetSessionId()){
//将消息发送给绑定的学生
sendMsg(userSessionMap.get(serverSessionMap.get(session.getId()).getTargetSessionId()).getSession(), messageVo);
}
sendMsg(session,messageVo);
} else {//学生消息
System.out.println("学生"+ session.getId()+"发送消息：\""+ message +"\"给老师"+userSessionMap.get(session.getId()).getTargetSessionId());
result.put("msg", "学生"+session.getId()+"："+message);
messageVo.setCount(2);
messageVo.setMsg(message);
messageVo.setUserId(userSessionMap.get(session.getId()).getUserId());
//学生发出一条消息
sendMsg(session,messageVo);
//判断是否绑定了老师，如果有就发送消息
String targetSessionId = userSessionMap.get(session.getId()).getTargetSessionId();
if (null != targetSessionId && !targetSessionId.equals("kefu")){
//给当前学生咨询的老师发送消息内容
messageVo.setUserId(userSessionMap.get(session.getId()).getUserId());
sendMsg(serverSessionMap.get(userSessionMap.get(session.getId()).getTargetSessionId()).getSession(), messageVo);
}
if (null != targetSessionId && targetSessionId.equals("kefu")){
try {
Long problemNumber = Long.valueOf(StrUtil.clearNull(message));
String answer = buiProblemAnswerService.getAnswer(problemNumber);
messageVo.setMsg(answer);
if (answer == null){
messageVo.setMsg("请输入正确的问题编号！！！");
}
}catch (Exception e){
List<BuiProblemAnswer> buiProblemAnswerList = buiProblemAnswerService.getAnswerList(message);
if (buiProblemAnswerList.size() > 1){
String msg = "猜你想问：\t";
for (BuiProblemAnswer buiProblemAnswer : buiProblemAnswerList) {
msg = msg + buiProblemAnswer.getProblemTopic()+"\t";
}
messageVo.setMsg(msg);
}else if (buiProblemAnswerList.size() == 1){
messageVo.setMsg(buiProblemAnswerList.get(0).getProblemAnswer());
}else if (buiProblemAnswerList.size() == 0){
messageVo.setMsg("抱歉！没有找到您的问题！");
}
// messageVo.setMsg("猜你想问:\t"+);
// messageVo.setMsg("请输入正确的问题编号！！！");
}
}
//回复学生消息
messageVo.setUserId(null);
sendMsg(session,messageVo);
}
}
/**
* 关闭连接调用的方法，群成员退出
*
* @param session
* @param targetSessionId
*/
@OnClose
public void onClose(Session session, @PathParam("targetSessionId") String targetSessionId, @PathParam("userId") Integer userId) {
SocketUserInfo serverInfo = serverSessionMap.get(session.getId());
MessageVo messageVo = new MessageVo();
if (serverInfo != null){
//老师下线,将老师从map中移除
for (int i = 0; i < teacherUserMapList.size(); i++) {
Map<String, Object> map = teacherUserMapList.get(i);
if (userId == map.get("userId")){
teacherUserMapList.remove(i);
}
}
//更新缓存中的在线老师的数据
cacheProxy.set("websocket",teacherUserMapList);
serverSessionMap.remove(session.getId());
//查看当前是否有在服务的对象，如果有就给用户发送系统错误的信息
if (serverInfo.getTargetSessionId() != null){
messageVo.setMsg("系统错误，老师已经下线了");
messageVo.setUserId(serverInfo.getUserId());
sendMsg(userSessionMap.get(serverInfo.getTargetSessionId()).getSession(),messageVo);
}
log.info("老师编号：" + session.getId() + "退出了连接，当前在线老师共计：" + serverSessionMap.size());
}else if ("10000".equals(targetSessionId)){
//学生下线 从老师中解绑，解绑后，如果还有在排队的学生，则让这个学生和当前老师绑定起来
userSessionMap.remove(session.getId());
//获取当前老师信息
SocketUserInfo serverUserInfo = serverSessionMap.get(targetSessionId);
HashMap<String, Object> teacherMap = new HashMap<>();
teacherMap.put("userName",serverUserInfo.getUserName());
teacherMap.put("serssionId",serverUserInfo.getSessionId());
teacherMap.put("tel",serverUserInfo.getTerminalCode());
teacherMap.put("userId",serverUserInfo.getUserId());
//将当前老师添加到空闲老师列表并更新缓存
teacherUserMapList.add(teacherMap);
cacheProxy.set("websocket",teacherUserMapList);
log.info("用户编号：" + session.getId() + "退出了连接，当前在线学生共计：" + userSessionMap.size());
}
}
/**
* 传输消息错误调用的方法
*
* @param error
*/
@OnError
public void OnError(Throwable error) {
log.info("Connection error");
}
//查询排队用户
private synchronized String findLineUser(){
//判断是否有用户
if (userSessionMap.size() > 0){
//遍历所有用户，查找一个排队的用户
for (SocketUserInfo UserInfo: userSessionMap.values()) {
if (null == UserInfo.getTargetSessionId()){
return UserInfo.getSessionId();
}
}
}
return null;
}
//查询在线空闲的老师
private synchronized String findFreeServer(){
//判断是否有老师
if (serverSessionMap.size() > 0){
//遍历在线的所有老师，查找一个空闲的老师
for (SocketUserInfo serverInfo: serverSessionMap.values()) {
if (null == serverInfo.getTargetSessionId()){
return serverInfo.getSessionId();
}
}
}
return null;
}
//统一的发送消息方法
private synchronized void sendMsg(Session session, MessageVo messageVo) {
try {
session.getBasicRemote().sendText(JSON.toJSONString(messageVo));
} catch (IOException e) {
e.printStackTrace();
}
}
}
websocket中无法注入bean实例的问题： 解决：
这样就可以使用@Autowired来注入实例了
最后前端就可以使用 'ws://ip地址:端口/项目地址/groupChat/{参数1}/{参数2}' 来请求连接了
注意：如果后台做了用户认证配置了拦截器需要放行接口
相关推荐
- Vue3非兼容变更——函数式组件（vue 兼容）
-
在Vue2.X中，函数式组件有两个主要应用场景：作为性能优化，因为它们的初始化速度比有状态组件快得多；返回多个根节点。然而在Vue3.X中，有状态组件的性能已经提高到可以忽略不计的程度。此外，有状态组...
- 利用vue.js进行组件化开发，一学就会(一)
-
组件原理/组成组件（Component）扩展HTML元素，封装可重用的代码，核心目标是为了可重用性高，减少重复性的开发。组件预先定义好行为的ViewModel类。代码按照template\styl...
- Vue3 新趋势:10 个最强 X 操作!（vue.3）
-
Vue3为前端开发带来了诸多革新，它不仅提升了性能，还提供了...
- 总结 Vue3 组件管理 12 种高级写法，灵活使用才能提高效率
-
SFC单文件组件顾名思义，就是一个.vue文件只写一个组件...
- 前端流行框架Vue3教程:17. _组件数据传递
-
_组件数据传递我们之前讲解过了组件之间的数据传递，...
- 前端流行框架Vue3教程:14. 组件传递Props效验
-
组件传递Props效验Vue组件可以更细致地声明对传入的props的校验要求...
- 前端流行框架Vue3教程:25. 组件保持存活
-
25.组件保持存活当使用...
- 5 个被低估的 Vue3 实战技巧，让你的项目性能提升 300%?
-
前端圈最近都在卷性能优化和工程化，你还在用老一套的Vue3开发方法？作为摸爬滚打多年的老前端，今天就把私藏的几个Vue3实战技巧分享出来，帮你在开发效率、代码质量和项目性能上实现弯道超车！一、...
- 绝望!Vue3 组件频繁崩溃?7 个硬核技巧让性能暴涨 400%!
-
前端的兄弟姐妹们五一假期快乐，谁还没在Vue3项目上栽过跟头？满心欢喜写好的组件，一到实际场景就频频崩溃，页面加载慢得像蜗牛，操作卡顿到让人想砸电脑。用户疯狂吐槽，领导脸色难看，自己改代码改到怀疑...
- 前端流行框架Vue3教程:15. 组件事件
-
组件事件在组件的模板表达式中，可以直接使用...
- Vue3，看这篇就够了（vue3 从入门到实战）
-
一、前言最近很多技术网站，讨论的最多的无非就是Vue3了，大多数都是CompositionAPI和基于Proxy的原理分析。但是今天想着跟大家聊聊，Vue3对于一个低代码平台的前端更深层次意味着什么...
- 前端流行框架Vue3教程:24.动态组件
-
24.动态组件有些场景会需要在两个组件间来回切换，比如Tab界面...
- 前端流行框架Vue3教程:12. 组件的注册方式
-
组件的注册方式一个Vue组件在使用前需要先被“注册”，这样Vue才能在渲染模板时找到其对应的实现。组件注册有两种方式：全局注册和局部注册...
- 焦虑!Vue3 组件频繁假死?6 个奇招让页面流畅度狂飙 500%!
-
前端圈的朋友们，谁还没在Vue3项目上踩过性能的坑？满心期待开发出的组件，一到高并发场景就频繁假死，用户反馈页面点不动，产品经理追着问进度，自己调试到心态炸裂！别以为这是个例，不少人在电商大促、数...
- 前端流行框架Vue3教程:26. 异步组件
-
根据上节课的代码，我们在切换到B组件的时候，发现并没有网络请求：异步组件：...
欢迎 你 发表评论:
- 一周热门
- 最近发表
- 标签列表
-
- idea eval reset (50)
- vue dispatch (70)
- update canceled (42)
- order by asc (53)
- spring gateway (67)
- 简单代码编程 贪吃蛇 (40)
- transforms.resize (33)
- redisson trylock (35)
- 卸载node (35)
- np.reshape (33)
- torch.arange (34)
- npm 源 (35)
- vue3 deep (35)
- win10 ssh (35)
- vue foreach (34)
- idea设置编码为utf8 (35)
- vue 数组添加元素 (34)
- std find (34)
- tablefield注解用途 (35)
- python str转json (34)
- java websocket客户端 (34)
- tensor.view (34)
- java jackson (34)
- vmware17pro最新密钥 (34)
- mysql单表最大数据量 (35)