使用WebSocket编写聊天室笔记

最新推荐文章于 2024-05-24 17:56:50 发布
最新推荐文章于 2024-05-24 17:56:50 发布
阅读量262 收藏 5
点赞数 5
文章标签： websocket 笔记 网络协议
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_65325528/article/details/138165094
版权
        本人是java小白，本章节属于我得毕设项目中的小块个功能点，如果有编写不对的地方还请指出，非常感谢！！！
        聊天系统采用最简单的scoket插件写的，真正的聊天系统最好使用原生udp/tcp等方式编写

       本章默认已经将SpringBoot项目搭建完成，下面操作是在SpringBoot的基础上编写的

1、创建SpringBoot项目,添加依赖
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
2、创建WebSocketConfig配置
@Configuration
public class WebSocketConfig {
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
        return new ServerEndpointExporter();
    }
}
3、定义保存用户会话信息的map
//    保存用户的map
    public static ConcurrentHashMap<String,MyWebSocket> users = new ConcurrentHashMap<>();
//    保存客服的map
    public static ConcurrentHashMap<String,MyWebSocket> kf = new ConcurrentHashMap<>();
4、创建控制类
/**
 * @author: 赵帅峰
 * @create: 2023/10/5
 * @FileName: MyWebSocket
 */
@ServerEndpoint(value = "/websocket/{sid}/{identify}",encoders = {ServerEncoder.class})
@Component
public class MyWebSocket extends ChatOperate  {
​
    private static final Lock lock = new ReentrantLock();
​
    private Session session;
​
    private String identify;
​
    private String sid = "";
​
    private static LoginService loginService;
​
    @Autowired
    public void setINoticeService(LoginService loginService) {
        MyWebSocket.loginService = loginService;
    }
​
    @OnOpen
    public void onOpen(@PathParam("sid") String sid, @PathParam("identify") String identify, Session session) throws IOException {
​
        this.session = session;
        this.sid = sid;
        this.identify = identify;
​
        if(this.identify.equals("3")){
            WebSocketUser.kf.put(this.sid,this);
        }else{
            WebSocketUser.users.put(this.sid,this);
        }
​
        if(this.identify.equals("3")){
            this.sendMessage(getServiceDialog(sid),0);
            this.AllUserSend();
//            AllSendMessage(this.sid,"客服","客服已上线",1);
        }else {
            this.sendMessage(getUserDialog(sid),0);
            this.AllKfSend();
//            UserIdToSend(sid,sid,identify,"","已上线",1);
        }
    }
​
​
    @OnMessage
    public void onMessage(String message, Session session) throws IOException {
​
        HashMap hashMap = JSON.parseObject(message, HashMap.class);
//        这是接收人的id
        String id = (String) hashMap.get("id");
//        这是发送人的id
        String sid = (String) hashMap.get("sid");
//        这是发送人的标识
        String identify = (String) hashMap.get("identify");
//        这是发送人的用户名
        String name = (String) hashMap.get("name");
​
        String context = (String) hashMap.get("context");
        Dialogue dialogue = new Dialogue(Integer.valueOf(sid), Integer.valueOf(id), name, "", context, Integer.valueOf(identify));
        if(id.equals("All") || id.equals("all")){
//            this.AllSendMessage(sid,name,context,2);
        }else{
//            this.sendMessage(dialogue);
            this.UserIdToSend(dialogue);
//            this.UserIdToSend(sid,id,identify,name,context,2);
        }
    }
​
    public void UserIdToSend(Dialogue dialogue) throws IOException {
        //        将信息保存Redis
        saveService(dialogue);
        saveUserMessage(dialogue);
​
        MyWebSocket myWebSocket = null;
        if(dialogue.getIdentify() == 3){
            myWebSocket = WebSocketUser.users.get(String.valueOf(dialogue.getRecipient()));
            if(myWebSocket==null){
                dialogue.setContext("不在线");
            }else{
                myWebSocket.sendMessage(dialogue,1);
            }
            WebSocketUser.kf.get(String.valueOf(dialogue.getSender())).sendMessage(dialogue,1);
        }else{
            myWebSocket = WebSocketUser.kf.get(String.valueOf(dialogue.getRecipient()));
            if(myWebSocket==null){
                dialogue.setContext("不在线");
            }else{
                myWebSocket.sendMessage(dialogue,1);
            }
            WebSocketUser.users.get(String.valueOf(dialogue.getSender())).sendMessage(dialogue,1);
        }
​
    }
​
    public void sendMessage(Object o,int flag) throws IOException {
        HashMap<String, Object> result = new HashMap<>();
        if("3".equals(this.identify)){
            result.put("user",WebSocketUser.users.keys());
        }else {
            result.put("user",WebSocketUser.kf.keys());
        }
        result.put("flag",flag);
        result.put("data",o);
        lock.lock();
        try{
            this.session.getBasicRemote().sendObject(result);//同步
            //this.session.getAsyncRemote().sendText(message);//异步
        } catch (EncodeException e) {
            e.printStackTrace();
        } finally {
            lock.unlock();
        }
    }
​
    public void AllKfSend(){
       WebSocketUser.kf.forEach(new BiConsumer<String, MyWebSocket>() {
            @SneakyThrows
            @Override
            public void accept(String s, MyWebSocket myWebSocket) {
                myWebSocket.sendMessage(null,1);
            }
        });
    }
​
    public void AllUserSend(){
        WebSocketUser.users.forEach(new BiConsumer<String, MyWebSocket>() {
            @SneakyThrows
            @Override
            public void accept(String s, MyWebSocket myWebSocket) {
                myWebSocket.sendMessage(null,1);
            }
        });
    }
    @OnError
    public void onError(Session session, Throwable error) {
        System.out.println("发生错误");
        error.printStackTrace();
    }
​
    @OnClose
    public void onclose(Session session) throws IOException {
        if (!sid.equals("")) {
            WebSocketUser.users.remove(sid);  //从set中删除
            WebSocketUser.kf.remove(sid);  //从set中删除
        }
    }
}
利用Redis设置会话持久
5、添加redis依赖
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>
<dependency>
    <groupId>com.alibaba</groupId>
    <artifactId>fastjson</artifactId>
    <version>1.2.62</version>
</dependency>
6、创建RedisConfig配置
@Configuration
public class RedisConfig {
​
    @Bean
    public RedisTemplate<String,Object> redisTemplate(LettuceConnectionFactory con){
        RedisTemplate<String,Object> template = new RedisTemplate<String,Object>();
        template.setConnectionFactory(con);
        template.setKeySerializer(new StringRedisSerializer());  //key采用字符串方式管理
        template.setValueSerializer(new GenericJackson2JsonRedisSerializer());  //value使用jdk的序列化
        template.setHashKeySerializer(new StringRedisSerializer());
        template.setHashValueSerializer(new GenericJackson2JsonRedisSerializer());
        return template;
    }
}
7、在application.properties中添加redis的配置项
#配置redis
# Redis服务器地址
spring.redis.host=localhost
# Redis服务器连接端口
spring.redis.port=6379
# Redis数据库索引（默认为0）
spring.redis.database=0
# Redis服务器连接密码（默认为空）
spring.redis.password=
# 连接池最大连接数（使用负值表示没有限制） 默认 8
spring.redis.lettuce.pool.max-active=8
# 连接池最大阻塞等待时间（使用负值表示没有限制） 默认 -1
spring.redis.lettuce.pool.max-wait=-1
# 连接池中的最大空闲连接 默认 8
spring.redis.lettuce.pool.max-idle=8
# 连接池中的最小空闲连接 默认 0
spring.redis.lettuce.pool.min-idle=0
8、定义存储的json格式

客服：

{

客服id:{

用户Id_1:[{},{}],

用户Id_2:[{},{}]

}

客服2：{

用户Id_1:[{内容},{内容}],

用户Id_2:[{内容},{内容}]

}

}

客户：{

用户Id_1:[{内容},{内容},{内容}],

用户Id_2:[{内容},{内容}]

}

9、添加实体类，并实现序列化,get和set方法自己添加
public class Dialogue implements Serializable {
​
//    发送者id
    private int sender = -1;
//    接收者id
    private int recipient = -1;
//    发送者名称
    private String SenderName;
//    接收者名称
    private String recipientName;
//    内容
    private String message;
//    0：客服  1：用户
    private int isCustomer;
    //    创建时间
    private String createDate;
​
    public Dialogue(){}
    public Dialogue(int sender, int recipient, String senderName, String recipientName, String message,int isCustomer) {
        this.sender = sender;
        this.recipient = recipient;
        this.SenderName = senderName;
        this.recipientName = recipientName;
        this.message = message;
        this.isCustomer = isCustomer;
        Date date = new Date();
        this.createDate = new SimpleDateFormat("MM-dd hh:mm:ss").format(date);;
    }
    @Override
    public String toString() {
        return "Dialogue{" +
                "sender=" + sender +
                ", recipient=" + recipient +
                ", SenderName='" + SenderName + '\'' +
                ", recipientName='" + recipientName + '\'' +
                ", message='" + message + '\'' +
                ", isCustomer=" + isCustomer +
                ", createDate='" + createDate + '\'' +
                '}';
    }
}
9、编写对会话存储的代码
/**
 * @author: 赵帅峰
 * @create: 2023/12/13
 * @FileName: ChatOperate
 */
@Component
public class ChatOperate {
​
​
    private static RedisTemplate redisTemplate;
​
    @Autowired
    public void setRedisTemplate(RedisTemplate redisTemplate) {
        ChatOperate.redisTemplate = redisTemplate;
    }
​
    /**
     *  这是在redis中添加客服聊天记录
     * @param dialogue
     * @return
     */
    public Boolean saveService(Dialogue dialogue){
​
        if(dialogue.getSender() == -1 || dialogue.getRecipient()== -1){
            throw new RuntimeException("添加异常，请查看信息是否正确");
        }
​
//        判断当前信息是用户发的还是客服发的,这里判断客户id
        String recipient = String.valueOf(dialogue.getIdentify()==3 ? dialogue.getSender() : dialogue.getRecipient());
//        判断当前信息是用户发的还是客服发的,这里判断用户id
        String Sender = String.valueOf(dialogue.getIdentify()==3 ? dialogue.getRecipient() : dialogue.getSender());
​
//        如果记录为空，则直接添加，否则原有基础上添加
        if (redisTemplate.opsForHash().get("chat:service", recipient)==null){
            HashMap<String, List> map = new HashMap<>();
            ArrayList<Dialogue> d = new ArrayList<>();
            d.add(dialogue);
            map.put(Sender,d);
            redisTemplate.opsForHash().put("chat:service",recipient,map);
        }else{
//        查询redis数据库中改客服与该用户的聊天记录
            HashMap<String,List> dialogues = (HashMap<String,List>) redisTemplate.opsForHash().get("chat:service", recipient);
            List<Dialogue> arr = dialogues.get(Sender)!=null ? dialogues.get(Sender) : new ArrayList<>();
            arr.add(dialogue);
            dialogues.put(Sender,arr);
            redisTemplate.opsForHash().put("chat:service",recipient,dialogues);
        }
        return true;
    }
​
    /**
     * 这是在redis中添加用户聊天记录
     * @param dialogue
     * @return
     */
    public Boolean saveUserMessage(Dialogue dialogue){
​
        if(dialogue.getSender() == -1 || dialogue.getRecipient()== -1){
            throw new RuntimeException("添加异常，请查看信息是否正确");
        }
​
//        判断当前信息是用户发的还是客服发的,这里判断用户
        String recipient = String.valueOf(dialogue.getIdentify()!=3 ? dialogue.getSender() : dialogue.getRecipient());
​
        if(redisTemplate.opsForHash().get("chat:user", recipient) == null){
            ArrayList<Dialogue> d = new ArrayList<>();
            d.add(dialogue);
            redisTemplate.opsForHash().put("chat:user",recipient,d);
        }else{
            List dialogues = (List) redisTemplate.opsForHash().get("chat:user", recipient);
            dialogues.add(dialogue);
            redisTemplate.opsForHash().put("chat:user",recipient,dialogues);
        }
        return true;
    }
​
    /**
     *  按照id查找客服的聊天记录
     * @param id
     * @return
     */
    public HashMap<String,List<Dialogue>> getServiceDialog(String id){
        if(redisTemplate.opsForHash().get("chat:service", id) == null){
            return null;
        }
        return (HashMap) redisTemplate.opsForHash().get("chat:service", id);
    }
​
    /**
     *  按照id查找用户的聊天记录
     * @param id
     * @return
     */
    public List<Dialogue> getUserDialog(String id){
        if(redisTemplate.opsForHash().get("chat:user", id) == null){
            return null;
        }
        return (List<Dialogue>) redisTemplate.opsForHash().get("chat:user", id);
    }
​
    /**
     * 根据指定id删除客服记录
     * @param id
     * @return
     */
    public Long deleteServiceById(String id){
        if(redisTemplate.opsForHash().get("chat:service",id)==null) return 0L;
        return redisTemplate.opsForHash().delete("chat:service",id);
    }
​
    /**
     * 根据指定id删除用户记录
     * @param id
     * @return
     */
    public Long deleteUserById(String id){
        if(redisTemplate.opsForHash().get("chat:user",id)==null) return 0L;
        return redisTemplate.opsForHash().delete("chat:user",id);
    }
}
10、测试方法
@Test
public void 添加记录(){
    Dialogue dialogue = new Dialogue(3,5,"admin","zsf1","nihao",0);
    saveUserMessage(dialogue);
    saveService(dialogue);
    Dialogue dialogue1 = new Dialogue(5,3,"zsf","admin","你好，我在呢",1);
    saveUserMessage(dialogue1);
    saveService(dialogue1);
}
@Test
public void 根据id删除记录(){
    System.out.println(deleteServiceById("3"));
    System.out.println(deleteUserById("4"));
}
@Test
public void 查询用户记录(){
    System.out.println(getServiceDialog("3"));
    System.out.println(getUserDialog("4"));
}
11、前端调用

前端调用时直接用websocket调用即可，将自己的id和身份标识传来

标识：3 -- 客服人员 other（其他） -- 用户

我爱吃干果
关注
ASP.NET Core通过Websocket建立简单聊天室
m0_64916060的博客
11-16 1844
初学者的傻瓜式笔记，备忘
06----3小时 用 websocket 实现聊天室 之实现聊天功能，股市干货分享
imtokenmax众筹
04-16 330
首先在页面上引用css文件和js文件，css文件一般在中添加，js文件一般在之前添加。另外，大推一个网上教程 现代 JavaScript 教程 ，文章深入浅出，很容易理解，上面的内容几乎都是重点，而且充分发挥了网上教程的时效性和资料链接。想要取得优秀的面试成绩，刷面试题是必须的，除非你样样精通。学习资料在精不在多，二者结合，定能构建你的 JavaScript 知识体系。// 需要把这个文件发送到服务器， 借助于H5新增的fileReader。// 把接收到的消息显示到聊天窗口中。//接受用户发送的图片。
SpringBoot利用redis集成消息队列的方法
08-29
主要介绍了SpringBoot利用redis集成消息队列的方法,需要的朋友可以参考下
sprintboot Redis配置详解
u010125432的博客
10-21 842
# REDIS (RedisProperties) # Redis数据库索引（默认为0） spring.redis.database=0 # Redis服务器地址 spring.redis.host=localhost # Redis服务器连接端口 spring.redis.port=6379 # Redis服务器连接密码（默认为空） spring.redis.password= # 连接池最大连接数（使用负值表示没有限制） spring.redis.pool.max-active=8 # 连接池最大阻塞等
Springboot项目中使用Redis作为消息队列
最新发布
StruggleRookie的博客
05-24 444
Springboot项目中使用Redis作为消息队列
启动spring gateway网关服务报错 Unable to find GatewayFilterFactory with name RequestRateLimiter
rdping的博客
03-30 4355
解决：启动spring gateway网关服务报错 Unable to find GatewayFilterFactory with name RequestRateLimiter
Unable to find GatewayFilterFactory with name RequestRateLimiter
淮水竹亭
05-09 2475
错误原因：由于可能有多个子工程使用redis，所以在父工程changgou-gateway引入spring-boot-starter-data-redis-reactive依赖。但是redis实际使用者是子工程changgou-gateway-web,点击File->Project Structure，发现子工程changgou-gateway-web没有redis依赖。 解决办法：更新父工程changgou-gatewaypom依赖，子工程changgou-gateway-web的redis依赖.
spring+springMVC+WebSocket 的学习笔记
10-15
相较于传统的 HTTP 协议，WebSocket 提供更低的延迟和更高的效率，尤其适用于实时通信应用，如聊天室、股票交易、在线游戏等。 **Spring 和 WebSocket 集成** Spring 通过 Spring WebSocket 模块提供了对 ...
WebSocket 开发
harvestUachieve的博客
08-17 916
websocketWebSocket是HTML5下一种新的协议（websocket协议本质上是一个基于tcp的协议）它实现了浏览器与服务器全双工通信，能更好的节省服务器资源和带宽并达到实时通讯的目的Websocket是一个持久化的协议WebSocket使得客户端和服务器之间的数据交换变得更加简单，允许服务端主动向客户端推送数据。在WebSocket API中，浏览器和服务器只需要完成一次握手，两者之间就直接可以创建持久性的连接，并进行双向数据传输。
SpringCloudGateway限流报Unable to find GatewayFilterFactory with name RequestRateLimiter异常
AE86的博客
05-12 3432
前言 最近在SpringCloudGateway网关加上动态路由和限流的功能，使用RequestRateLimiter配置令牌桶时，启动应用报了异常，异常信息为Unable to find GatewayFilterFactory with name RequestRateLimiter 原因是：引入 org.springframework.boot:spring-boot-starter-data-redis-reactive 依赖时，排除了lettuce-core依赖（当时一个同事使用Lettuce
springBoot整合redis使用介绍（详细案例）
热门推荐
weixin_43811057的博客
09-19 5万+
文章预览：一、创建springboot项目（采用骨架方式）二、配置文件三、使用redis1、添加字符串到redis2、将对象转换成jsonString并存入redis3、将对象集合转换成jsonString，并设置过期时间存入至redis4、获取对象5、获取对象集合6、添加 hash-set7、获取 hash-setvalue 一、创建springboot项目（采用骨架方式） 创建完成； 我们分析下pom文件中内容： 所使用到的关键依赖： <!--springBoot集成redis--&
Spring Boot 整合Redis实现消息队列
嘉禾嘉宁papa
02-27 1万+
Spring Boot 整合Redis实现消息队列，RedisMessageListenerContainer的使用，Pub/Sub模式的优缺点
Spring cloud Gateaway : Unable to find GatewayFilterFactory with name RequestRateLimiter
峡谷电光马仔的博客
01-07 1650
Spring cloud Gateaway : o.s.c.gateway.route.CachingRouteLocator : Refresh routes error !!! Unable to find GatewayFilterFactory with name RequestRateLimiter
Satoken+Redis实现短信登录、注册、鉴权
love_eat_peach的博客
06-05 2271
Satoken+Redis实现短信登录、注册、鉴权；Redis中缓存用户权限、验证码、登录 次数等信息
Satoken+Redis实现短信登录、注册、鉴权_sa-token redis
m0_60147147的博客
04-15 832
接口的调用往往需要权限的校验，一般的系统会给用户绑定某种角色，再给此角色分配权限，设置权限码，具有此角色或者权限码才放行请求。在调用后台服务时，我们可以在路由时做一些拦截，例如添加登陆权限拦截、放开一些接口白名单等。（一定要排除 登录、注册、发送验证码等接口 的拦截）填入手机号后，若调用登录接口根据手机号查询数据库发现不存在该用户，自动跳转到注册页面。
springboot redis 实现消息队列
qq_34874784的博客
02-10 1368
6. 测试发送和接收消息。2. 配置Redis。5. 配置消费者监听。
Sa-Token 集成 alone-redis
2201_76059856的博客
07-03 1022
搭建两个Redis服务器，一个专门用来做业务缓存，另一台专门存放Sa-Token权限数据。
使用SpringBoot利用Redis实现消息队列
果酱 の 博客
05-31 4115
随着互联网的发展，消息队列的应用越来越广泛。消息队列可以解决系统之间的异步通信问题，提高系统的可靠性和可扩展性。在Java开发中，Redis作为一种高性能的缓存和消息队列系统，被广泛应用。本文将介绍如何使用SpringBoot中利用Redis实现消息队列。
golang 使用nhooyr.io/websocket编写websocket 编写聊天室
07-28
当使用 nhooyr.io/websocket 编写聊天室时，您可以使用以下步骤： 1. 创建一个 WebSocket 服务器： ```go http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) { conn, err := websocket.Accept...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值