Spring boot 项目（十八）—— 集成WebSocket，实现群聊 + 单聊

原创

已于 2022-05-19 21:34:05 修改 · 1.1k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#websocket #spring boot #java

于 2022-03-01 23:36:23 首次发布

引言

对于某些项目来说，需要用户之间进行实时交流。本博客记录了实现群聊+单聊的功能的步骤。

项目结构

在这里插入图片描述

集成步骤

1、pom文件

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

2、config文件

@Configuration
public class WebSocketConfig {
   
   
    @Bean
    public ServerEndpointExporter serverEndpointExporter() {
   
   
        return new ServerEndpointExporter();
    }
}

3、entity类

@Data
public class SocketMsg {
   
   
    /**
     * 聊天类型0：群聊，1：单聊
     **/
    private int type;
    /**
     * 发送者
     **/
    private String fromUser;
    /**
     * 接受者
     **/
    private String toUser;
    /**
     * 消息
     **/
    private String msg;
}

4、具体实现类

import com.fasterxml.jackson.databind.ObjectMapper;
import com.information.entity.SocketMsg;
import org.springframework.stereotype.Component;
import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.CopyOnWriteArraySet;

/**
 * @author 流星
 * @Date: 2018/8/16 17:55
 * @Description: websocket的具体实现类
 * 使用springboot的唯一区别是要@Component声明下，而使用独立容器是由容器自己管理websocket的，
 * 但在springboot中连容器都是spring管理的。
 * 虽然@Component默认是单例模式的，但springboot还是会为每个websocket连接初始化一个bean，
 * 所以可以用一个静态set保存起来。
 */
@Component
@ServerEndpoint(value = "/websocket/{nickname}")
public class WebSocketServer{
   
   
    /**
     * 用来存放每个客户端对应的MyWebSocket对象。
     **/
    private static CopyOnWriteArraySet