国庆假期,整整七天,我使用SpringBoot终于做出了即时通信!!!

愿天堂没有BUG

共 4848字,需浏览 10分钟

 ·

2021-10-12 13:47

前言:(本文用于辅助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
来源:稀土掘金
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。



浏览 25
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报