通过webSocket实现服务端推送消息到前端页面

pom文件中引入依赖

<!-- 开启websocket支持-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-websocket</artifactId>
        </dependency>

配置类

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

controller层代码

@RestController
@RequestMapping("/test")
@Slf4j
public class WebsocketTest {
    @Autowired
    private Websocket websocket;

    @RequestMapping(value = "/a",method = RequestMethod.GET)
    public String  test(String a){
        if (a!=null){
            websocket.sendMessage("heheh");
        }        return a;
    }
}

service层代码

@Component
@ServerEndpoint("/webSocket")
@Slf4j
public class Websocket {

    private Session session;

    private static CopyOnWriteArraySet<Websocket> websocket = new CopyOnWriteArraySet<>();

    @OnOpen
    public void onOpen(Session session){
        this.session =session;
        websocket.add(this);
        log.info("【websocket消息】有新的连接,总数:{}",websocket.size());
    }
    @OnClose
    public void onClose(){
        websocket.remove(this);
        log.info("【websocket消息】连接断开,总数:{}",websocket.size());
    }

    @OnMessage
    public void onMessage(String message){
        log.info("【websocket消息】,收到客户端发来的消息:{}",message);
    }
    public void sendMessage(String message){
        for (Websocket websocket : websocket) {
            log.info("【websocket消息】广播消息:{}",message);
            try {
                websocket.session.getBasicRemote().sendText(message);
            }
            catch (Exception e){
                e.printStackTrace();

            }        }
    }
}

前端页面

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>webSocket</title>
</head>
<body>

websocket测试

<script src="https://cdn.bootcss.com/jquery/1.12.4/jquery.min.js"></script>
<script src="https://cdn.bootcss.com/bootstrap/3.3.5/js/bootstrap.min.js"></script>
<script>
    var websocket = null;
    if ('WebSocket' in window) {
        websocket = new WebSocket('ws:/localhost:8080/webSocket');
    } else {
        alert('该浏览器不支持websocket!');
    }
    websocket.onopen = function (event) {
        console.log('建立连接');
    }

    websocket.onclose = function (event) {
        console.log('连接关闭');
    }

    websocket.onmessage = function (event) {
        console.log('收到消息:' + event.data)
        alert("收到服务端消息");
    }

    websocket.onerror = function () {
        alert('websocket通信发生错误!');
    }

    window.onbeforeunload = function () {
        websocket.close();
    }

</script>
</body>
</html>

运行代码

在这里插入图片描述

页面

在这里插入图片描述

postman测试

在这里插入图片描述

页面弹窗显示

在这里插入图片描述

在这里插入图片描述

  • 0
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
下面是一个使用 Spring WebFlux 实现 WebSocket 服务端推送 FLV 视频数据的示例代码: ```java @Configuration @EnableWebFlux public class WebSocketConfig { @Bean public HandlerMapping handlerMapping() { Map<String, WebSocketHandler> map = new HashMap<>(); map.put("/video", new VideoWebSocketHandler()); SimpleUrlHandlerMapping mapping = new SimpleUrlHandlerMapping(); mapping.setUrlMap(map); mapping.setOrder(1); return mapping; } @Bean public WebSocketHandlerAdapter handlerAdapter() { return new WebSocketHandlerAdapter(); } } public class VideoWebSocketHandler implements WebSocketHandler { private Flux<DataBuffer> videoData; public VideoWebSocketHandler() { // 从文件或网络中读取 FLV 视频数据 // 这里假设已经将数据读取为一个 Flux<DataBuffer> 对象 this.videoData = readVideoData(); } @Override public Mono<Void> handle(WebSocketSession session) { return session.send(videoData.map(session::binaryMessage)) .and(session.receive().map(WebSocketMessage::getPayloadAsText).log()); } private Flux<DataBuffer> readVideoData() { // 这里使用一个简单的随机数据生成器模拟视频数据 // 实际应用中需要从文件或网络读取真实的 FLV 视频数据 return Flux.generate(() -> new byte[1024], (bytes, sink) -> { new Random().nextBytes(bytes); sink.next(DefaultDataBufferFactory.sharedInstance.wrap(bytes)); return bytes; }); } } ``` 在上面的代码中,我们定义了一个 `VideoWebSocketHandler` 类,它实现了 `WebSocketHandler` 接口,用于处理 WebSocket 连接。在 `VideoWebSocketHandler` 类的构造方法中,我们读取了 FLV 视频数据并将其转换为一个 `Flux<DataBuffer>` 对象。在 `handle` 方法中,我们使用 `session.send` 方法将视频数据发送给客户端,并使用 `session.receive` 方法接收客户端的消息。这里我们假设客户端不会发送任何消息,因此我们只是简单地将接收到的消息记录到日志中。 在 `WebSocketConfig` 类中,我们定义了一个 `HandlerMapping` Bean,用于将 `/video` 路径映射到 `VideoWebSocketHandler` 处理器。我们还定义了一个 `WebSocketHandlerAdapter` Bean,用于将 `WebSocketHandler` 对象适配成 Spring WebFlux 的 `HandlerAdapter` 接口。 使用上面的代码,我们可以在 Spring WebFlux 中实现一个 WebSocket 服务端,用于推送 FLV 视频数据前端播放。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值