国庆假期，整整七天，我使用SpringBoot终于做出了即时通信！！!-技术圈

前言：(本文用于辅助Flutter WebSocket的理解) 在这个假期，我完成了一个小Demo，Flutter 与 Springboot 进行websocket的通讯，为啥想要去做这个Demo呢，主要是在各大平台以及google搜索后发现，没有一个详细的例子来教大家进行一对一、一对多的通讯，大多数都是教你怎么连接，却没有教你怎么去进行下一步的功能实现，于是我利用了五天的假期，踩了无数的坑，终于是完成了它，所以，点个赞吧，不容易啊，兄弟们😭

Flutter移动端分析：Flutter WebSocket 即时通讯

先上效果图（我自己搜索这样功能性的问题时，没有效果图基本上都是不想看的）：

这个效果图为Flutter：

即时通讯最重要的功能是完成了（发送文字信息）

阅读本文的注意点：

本文参考www.zhihu.com/column/p/32…，在其基础上进行二次开发

1.需要一点WebSocket的原理知识

2.springboot使用WebSocket的方法，本章就是最普通的原生方法

WebSocket的原理知识在Flutter WebSocket这篇文章中已经讲了，这里就不再重复了

正文：

1.Springboot使用WebSocket的方法

juejin.cn/post/684490… 掘金里已经有大神详细的讲解了

这里推荐spring封装或者STOMP两种方式

讲几个注意的点（详细的步骤掘金里的大神都已经写过啦~）：

pom.xml 配置websocket


            org.springframework.boot
            spring-boot-starter-websocket

复制代码

如果需要用户存储功能可以使用Spring Security，用户授权非常方便

STOMP这个协议是非常优秀的，是一种简单的基于文本的消息传递协议

如果想学习的，我推荐这篇大佬的文章：juejin.cn/post/684490…

2.Springboot实现点对点通信

现在步入正文

配置pom.xml
在默认的环境下，加上一个websocket即可，这里使用的是JDK8


    1.8


    
        org.springframework.boot
        spring-boot-starter
    
    
        org.springframework.boot
        spring-boot-starter-websocket
    

    
        org.springframework.boot
        spring-boot-starter-test
        test
    

复制代码

封装前端传入的信息，以及return的值：

因为使用json进行消息的发送，所以需要先创建一个消息对象，包含了消息发送者，消息接受者，消息类型

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

    ...省略了get 和 set方法，不想写的话可以尝试lombook
}
复制代码

WebSocketConfig处理
这里就简单举了个例子，文章中其实没有怎么用到

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.server.standard.ServletServerContainerFactoryBean;

@Configuration
public class WebSocketConfig {
    @Bean
    public ServletServerContainerFactoryBean createWebSocketContainer() {
        ServletServerContainerFactoryBean container = new ServletServerContainerFactoryBean();
        // ws 传输数据的时候，数据过大有时候会接收不到，所以在此处设置bufferSize
        container.setMaxTextMessageBufferSize(512000);
        container.setMaxBinaryMessageBufferSize(512000);
        container.setMaxSessionIdleTimeout(15 * 60000L);
        return container;
    }
}
复制代码

最重要的内容，逻辑处理部分MyWebSocket来啦！

设置websocket连接点映射:

@ServerEndpoint(value = "/websocket/{nickname}")
复制代码

我们需要一个变量来存储每个客户端对应的MyWebSocket对象.

private static CopyOnWriteArraySet webSocketSet = new CopyOnWriteArraySet();
复制代码

一个变量用来记录sessionId和该session之间的绑定关系.

private static Map map = new HashMap();
复制代码

成功建立连接时

@OnOpen
public void onOpen(Session session, @PathParam("nickname") String nickname) {
    this.session = session;
    this.nickname = nickname;
    map.put(session.getId(), session);
    webSocketSet.add(this);//加入set中.
    this.session.getAsyncRemote().sendText(nickname + "上线了,（我的频道号是" + session.getId() + "）");
}
复制代码

收到客户端消息后调用

@OnMessage
public void onMessage(String message, Session session, @PathParam("nickname") String nickname) {
    //message 不是普通的string ，而是我们定义的SocketMsg json字符串.
    try {
        SocketMsg socketMsg = new ObjectMapper().readValue(message, SocketMsg.class);
        //一对一聊天
        if (socketMsg.getType() == 1) {
            //只需要找到发送者和接受者即可.
            socketMsg.setFromUser(session.getId());//发送者.
            //socketMsg.setToUser(toUser);//这个是由客户端进行设置.
            Session fromSession = map.get(socketMsg.getFromUser());
            Session toSession = map.get(socketMsg.getToUser());
            if (toSession != null) {
                //发送消息
                fromSession.getAsyncRemote().sendText(nickname + "：" + socketMsg.getMsg());
                toSession.getAsyncRemote().sendText(nickname + "：" + socketMsg.getMsg());
            } else {
                fromSession.getAsyncRemote().sendText("系统消息：对方不在线或者您输入的频道号有误");
            }
        } else {
            //群发给每个客户端
            broadcast(socketMsg, nickname);
        }

    } catch (JsonParseException e) {
        // TODO Auto-generated catch block
        e.printStackTrace();
    } catch (JsonMappingException e) {
        // TODO Auto-generated catch block
        e.printStackTrace();
    } catch (IOException e) {
        // TODO Auto-generated catch block
        e.printStackTrace();
    }
}
复制代码

发生错误时

public void onError(Session session, Throwable error) {
    System.out.println("发生错误");
    error.printStackTrace();
}
复制代码

如果选择群发时（每个在线的客户都可以收到信息）

private void broadcast(SocketMsg socketMsg, String nickname) {
    for (MyWebSocket item : webSocketSet) {
        //发送消息.
        item.session.getAsyncRemote().sendText(nickname + "：" + socketMsg.getMsg());
    }
}
复制代码

连接关闭

@OnClose
public void onClose(Session session) {
    webSocketSet.remove(this);//从set中移除.
    map.remove(session.getId());
}
复制代码

使用源码时注意，本章使用JDK8，websocket的版本可能略有不同，端口为9090，具体的使用方法可以参考Flutter WebSocket的文章最后~

作者：阿Tya
链接：https://juejin.cn/post/7016606239937265700
来源：稀土掘金
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。