SpringBoot 整合 WebSocket 服务代码教程

import java.util.Objects;

@Slf4j

@Configuration

@EnableWebSocketMessageBroker

@ConditionalOnProperty(name = “websocket.enabled”, havingValue = “true”)

public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

/**

  • 方法描述: 端点配置

  • @param registry

  • @Return

  • @author caichengzhe

  • @date 2021年07月08日 10:43:55

*/

@Override

public void registerStompEndpoints(StompEndpointRegistry registry) {

registry.addEndpoint(“/ws”).setAllowedOrigins(“*”).withSockJS();

}

/**

  • 方法描述: 消息传输格式配置

  • @param registry

  • @Return

  • @author caichengzhe

  • @date 2021年07月08日 10:43:38

*/

@Override

public void configureWebSocketTransport(WebSocketTransportRegistration registry) {

registry.setMessageSizeLimit(500 * 1024 * 1024);//500Mb

registry.setSendBufferSizeLimit(1024 * 1024 * 1024);//1Gb

registry.setSendTimeLimit(200000);//200s

}

/**

  • 方法描述: 入界通道拦截,此处用来进行身份验证与用户信息设置

  • @param registration

  • @Return

  • @author caichengzhe

  • @date 2021年07月08日 10:43:15

*/

@Override

public void configureClientInboundChannel(ChannelRegistration registration) {

registration.interceptors(new ChannelInterceptor() {

@Override

public Message<?> preSend(Message<?> message, MessageChannel channel) {

StompHeaderAccessor accessor = MessageHeaderAccessor.getAccessor(message, StompHeaderAccessor.class);

if (StompCommand.CONNECT.equals(accessor.getCommand())) {

String token = accessor.getFirstNativeHeader(WebSocketConstant.WEBSOCKET_HEADER_TOKEN);

String userId = accessor.getFirstNativeHeader(WebSocketConstant.WEBSOCKET_HEADER_USER_ID);

if (StringUtil.isAnyBlank(token, userId)) {

return null;

}

//进行双重校验,两参数非空,且token中解析的用户id应于header传递的一致,否则视为非法请求

//websocket此处不再进行登录功能,均采用jwttoken进行身份验证

BladeUser bu = AuthUtil.getUser(token);

if (Objects.isNull(bu) || !StringUtil.equals(String.valueOf(bu.getUserId()), userId)) {

return null;

}

log.info(“用户:” + bu.getRealName() + “; userId:” + bu.getUserId() + " 连接成功");

accessor.setUser(() -> userId);

}

return message;

}

});

}

/**

  • 方法描述: 出界通道拦截配置,此处用来进行日志信息收集

  • @param registration

  • @Return

  • @author caichengzhe

  • @date 2021年07月08日 10:44:06

*/

@Override

public void configureClientOutboundChannel(ChannelRegistration registration) {

registration.interceptors(new ChannelInterceptor() {

@Override

public void postSend(Message<?> message, MessageChannel channel, boolean sent) {

log.info(“发出消息:” + JSON.toJSONString(message.getPayload()) + “;发送结果:” + (sent ? “成功” : “失败”) + “;消息通道:” + channel);

}

});

}

@Override

public void configureMessageBroker(MessageBrokerRegistry registry) {

//total message mapping include broadcast and specific topic

registry.enableSimpleBroker(“/topic/broadcast”, “/queue”);

// client-to-server message mapping prefix

registry.setApplicationDestinationPrefixes(“/unify-ws”);

// server to specific client mapping prefix

registry.setUserDestinationPrefix(“/user”);

}

}

或者 简单配置

import org.springframework.context.annotation.Bean;

import org.springframework.context.annotation.Configuration;

import org.springframework.web.socket.server.standard.ServerEndpointExporter;

@Configuration

public class WebSocketConfig {

/**

  • 注入一个ServerEndpointExporter,该Bean会自动注册使用@ServerEndpoint注解申明的websocket endpoint

*/

@Bean

public ServerEndpointExporter serverEndpointExporter() {

return new ServerEndpointExporter();

}

}

WebSocketService 服务类

import io.swagger.annotations.Api;

import lombok.extern.slf4j.Slf4j;

import org.springframework.stereotype.Component;

import javax.websocket.*;

import javax.websocket.server.ServerEndpoint;

import java.io.IOException;

import java.util.concurrent.ConcurrentHashMap;

import java.util.concurrent.atomic.AtomicInteger;

@Slf4j

@ServerEndpoint(value = “/point/change”)

@Component

@Api(value = “点位实时数据变更”, tags = “点位实时数据变更”)

public class WebSocketService {

/** concurrent包的线程安全Set,用来存放每个客户端对应的WebSocketServer对象。*/

private static ConcurrentHashMap<String, WebSocketService> webSocketMap = new ConcurrentHashMap<>();

private static ConcurrentHashMap<String, String> messageMap = new ConcurrentHashMap<>();

/*与某个客户端的连接会话,需要通过它来给客户端发送数据/

private Session session;

/** 记录当前在线连接数 */

private static AtomicInteger onlineCount = new AtomicInteger(0);

/**

  • 连接建立成功调用的方法

*/

@OnOpen

public void onOpen(Session session) {

this.session = session;

String sessionId = session.getId();

if(webSocketMap.containsKey(sessionId)){

//加入set中

webSocketMap.remove(sessionId);

webSocketMap.put(sessionId,this);

}else{

//加入set中

webSocketMap.put(sessionId,this);

// 在线数加1

onlineCount.incrementAndGet();

}

log.info(“有新连接加入:{},当前在线人数为:{}”, session.getId(), onlineCount.get());

}

/**

  • 连接关闭调用的方法

*/

@OnClose

public void onClose() {

if(webSocketMap.containsKey(session.getId())){

webSocketMap.remove(session.getId());

//从set中删除

onlineCount.decrementAndGet(); // 在线数减1

}

最后

自我介绍一下,小编13年上海交大毕业,曾经在小公司待过,也去过华为、OPPO等大厂,18年进入阿里一直到现在。

深知大多数初中级Android工程师,想要提升技能,往往是自己摸索成长,自己不成体系的自学效果低效漫长且无助。

因此收集整理了一份《2024年Web前端开发全套学习资料》,初衷也很简单,就是希望能够帮助到想自学提升又不知道该从何学起的朋友,同时减轻大家的负担。

img

既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,基本涵盖了95%以上Android开发知识点!不论你是刚入门Android开发的新手,还是希望在技术上不断提升的资深开发者,这些资料都将为你打开新的学习之门!

如果你觉得这些内容对你有帮助,需要这份全套学习资料的朋友可以戳我获取!!

由于文件比较大,这里只是将部分目录截图出来,每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频,并且会持续更新!
.(img-SvSh51fp-1715600019507)]

[外链图片转存中…(img-MGrNyX50-1715600019508)]

[外链图片转存中…(img-woZZOXKl-1715600019508)]

既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,基本涵盖了95%以上Android开发知识点!不论你是刚入门Android开发的新手,还是希望在技术上不断提升的资深开发者,这些资料都将为你打开新的学习之门!

如果你觉得这些内容对你有帮助,需要这份全套学习资料的朋友可以戳我获取!!

由于文件比较大,这里只是将部分目录截图出来,每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频,并且会持续更新!

  • 29
    点赞
  • 28
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
以下是一个简单的Spring Boot应用程序,它使用WebSocket进行通信: 1. 添加依赖 在pom.xml文件中添加以下依赖: ``` <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> </dependency> ``` 2. 创建WebSocket配置类 创建一个WebSocket配置类,用于配置WebSocket: ``` @Configuration @EnableWebSocket public class WebSocketConfig implements WebSocketConfigurer { @Override public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) { registry.addHandler(new MyWebSocketHandler(), "/my-websocket"); } } ``` 在这个配置类中,我们注册了一个WebSocket处理程序(MyWebSocketHandler),并将其映射到“/my-websocket”端点上。 3. 创建WebSocket处理程序 创建一个WebSocket处理程序,用于处理WebSocket请求: ``` public class MyWebSocketHandler extends TextWebSocketHandler { private final List<WebSocketSession> sessions = new CopyOnWriteArrayList<>(); @Override public void afterConnectionEstablished(WebSocketSession session) throws Exception { sessions.add(session); } @Override protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception { for (WebSocketSession s : sessions) { s.sendMessage(new TextMessage("Received message: " + message.getPayload())); } } @Override public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception { sessions.remove(session); } } ``` 在这个处理程序中,我们创建了一个会话列表,用于存储所有连接到WebSocket的会话。当一个新的WebSocket会话建立时,我们将其添加到会话列表中。当我们收到一个消息时,我们遍历会话列表,并将消息发送回每个会话。当WebSocket会话关闭时,我们从会话列表中删除该会话。 4. 创建WebSocket客户端 创建一个WebSocket客户端,用于连接到WebSocket服务器并发送消息: ``` public class MyWebSocketClient extends WebSocketClient { public MyWebSocketClient(URI serverUri, Draft draft) { super(serverUri, draft); } @Override public void onOpen(ServerHandshake handshakedata) { send("Hello, server!"); } @Override public void onMessage(String message) { System.out.println("Received message: " + message); } @Override public void onClose(int code, String reason, boolean remote) { System.out.println("WebSocket closed"); } @Override public void onError(Exception ex) { System.out.println("WebSocket error"); } } ``` 在这个WebSocket客户端中,我们定义了一些回调方法,用于处理WebSocket事件。当WebSocket连接建立时,我们发送一条消息。当我们收到一条消息时,我们将其打印到控制台上。当WebSocket连接关闭时,我们打印一条消息。当WebSocket发生错误时,我们打印一条消息。 5. 启动WebSocket服务器 最后,我们在Spring Boot应用程序的main方法中启动WebSocket服务器: ``` @SpringBootApplication public class Application { public static void main(String[] args) { SpringApplication.run(Application.class, args); WebSocketServerFactory factory = new DefaultWebSocketServerFactory(); WebSocketServer server = factory.createWebSocketServer(); server.start(); } } ``` 在这个main方法中,我们启动了Spring Boot应用程序,并创建了一个WebSocket服务器。我们使用默认的WebSocket服务器工厂来创建WebSocket服务器。我们调用start方法来启动WebSocket服务器。 现在,我们就完成了Spring Boot应用程序和WebSocket服务器的配置。运行应用程序,并使用WebSocket客户端连接到WebSocket服务器。您应该能够发送和接收消息。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值