百度360必应搜狗淘宝本站头条
vue dispatchspring gatewayorder by ascidea eval resetupdate canceled
当前位置：网站首页 > 技术分类 > 正文

SpringBoot+Websocket实现即时通讯

ztj100 2025-01-19 02:00 94 浏览 0 评论

1. 介绍

基于Spring Boot和websocket实现点对点在线聊天和简单的机器人自动回复功能，学生可选择在线空闲的老师咨询，无在线老师接入机器人根据学生提问的内容和问题编号从问题库中获取对应的答案来回复学生。

2.后台

2.1. 引入依赖

    // 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

@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}' 来请求连接了

注意：如果后台做了用户认证配置了拦截器需要放行接口

相关推荐

离谱!写了5年Vue，还不会自动化测试?

前言大家好，我是倔强青铜三。是一名热情的软件工程师，我热衷于分享和传播IT技术，致力于通过我的知识和技能推动技术交流与创新，欢迎关注我，微信公众号：倔强青铜三。Playwright是一个功能强大的端到...

package.json 与 package-lock.json 的关系

模块化开发在前端越来越流行，使用node和npm可以很方便的下载管理项目所需的依赖模块。package.json用来描述项目及项目所依赖的模块信息。那package-lock.json和...

Github 标星35k 的 SpringBoot整合acvtiviti开源分享，看完献上膝盖

前言activiti是目前比较流行的工作流框架，但是activiti学起来还是费劲，还是有点难度的，如何整合在线编辑器，如何和业务表单绑定，如何和系统权限绑定，这些问题都是要考虑到的，不是说纯粹的把a...

Vue3 + TypeScript 前端研发模板仓库

我们把这个Vue3+TypeScript前端研发模板仓库的初始化脚本一次性补全到可直接运行的状态，包括：完整的目录结构所有配置文件研发规范文档示例功能模块（ExampleFeature）...

Vue 2迁移Vue 3:从响应式到性能优化

小伙伴们注意啦！Vue2已经在2023年底正式停止维护，再不升级就要面临安全漏洞没人管的风险啦！而且Vue3带来的性能提升可不是一点点——渲染速度快40%，内存占用少一半，更新速度直接翻倍！还在...

VUE学习笔记:声明式渲染详解，对比WEB与VUE

声明式渲染是指使用简洁的模板语法，声明式的方式将数据渲染进DOM系统。声明式是相对于编程式而言，声明式是面向对象的，告诉框架做什么，具体操作由框架完成。编程式是面向过程思想，需要手动编写代码完成具...

苏州web前端培训班， 苏州哪里有web前端工程师培训

前端+HTML5德学习内容：第一阶段：前端页面重构：PC端网站布局、HTML5+CSS3基础项目、WebAPP页面布局；第二阶段：高级程序设计：原生交互功能开发、面向对象开发与ES5/ES6、工具库...

跟我一起开发微信小程序——扩展组件的代码提示补全

用户自定义代码块步骤：1.HBuilderX中工具栏：工具-代码块设置-vue代码块2.通过“1”步骤打开设置文件...

JimuReport 积木报表 v1.9.3发布，免费可视化报表

项目介绍积木报表JimuReport，是一款免费的数据可视化报表，含报表、大屏和仪表盘，像搭建积木一样完全在线设计！功能涵盖：数据报表、打印设计、图表报表、门户设计、大屏设计等！...

软开企服开源的无忧企业文档(V2.1.3)产品说明书

目录1....

一款面向 AI 的下一代富文本编辑器，已开源

简介AiEditor是一个面向AI的下一代富文本编辑器。开箱即用、支持所有前端框架、支持Markdown书写模式什么是AiEditor？AiEditor是一个面向AI的下一代富文本编辑...

玩转Markdown(2)——抽象语法树的提取与操纵

上一篇玩转Markdown——数据的分离存储与组件的原生渲染发布，转眼已经鸽了大半年了。最近在操纵mdast生成md文件的时候，心血来潮，把玩转Markdown(2)给补上了。...

DeepseekR1+ollama+dify1.0.0搭建企业/个人知识库(入门避坑版)

找了网上的视频和相关文档看了之后，可能由于版本不对或文档格式不对，很容易走弯路，看完这一章，可以让你少踩三天的坑。步骤和注意事项我一一列出来：1，前提条件是在你的电脑上已配置好ollama，dify1...

升级JDK17的理由，核心是降低GC时间

升级前后对比升级方法...

一个vsCode格式化插件_vscode格式化插件缩进量

ESlint...

取消回复欢迎 你 发表评论:

一周热门
最近发表
标签列表
赣ICP备2023011147号-29