From a30a7797dfdad782050a20e7df975a227a8ddbba Mon Sep 17 00:00:00 2001
From: liusuyi <1951119284@qq.com>
Date: Mon, 04 Nov 2024 16:08:02 +0800
Subject: [PATCH] 优化

---
 ard-work/src/main/java/com/ard/tianyu/websocket/config/WebSocketConfig.java |    9 ++++++++-
 1 files changed, 8 insertions(+), 1 deletions(-)

diff --git a/ard-work/src/main/java/com/ard/tianyu/websocket/config/WebSocketConfig.java b/ard-work/src/main/java/com/ard/tianyu/websocket/config/WebSocketConfig.java
index e43f086..f6579b0 100644
--- a/ard-work/src/main/java/com/ard/tianyu/websocket/config/WebSocketConfig.java
+++ b/ard-work/src/main/java/com/ard/tianyu/websocket/config/WebSocketConfig.java
@@ -1,6 +1,7 @@
 package com.ard.tianyu.websocket.config;
 
 import com.ard.tianyu.websocket.handler.TyMessageHandler;
+import com.ard.tianyu.websocket.handler.TyStreamHandler;
 import com.ard.tianyu.websocket.interceptor.TyHandshakeInterceptor;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.web.socket.config.annotation.EnableWebSocket;
@@ -14,14 +15,20 @@
 
     @Resource
     private TyMessageHandler tyMessageHandler;
+    @Resource
+    private TyStreamHandler tyStreamHandler;
 
     @Override
     public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
         //添加myWebSocketHandler消息处理对象以及websocket连接地址
-        registry.addHandler(tyMessageHandler, "/ws/**")
+        registry.addHandler(tyMessageHandler, "/ws/state/**")
                 //设置允许跨域访问
                 .setAllowedOrigins("*")
                 //添加拦截器可实现用户链接前进行权限校验等操作
                 .addInterceptors(new TyHandshakeInterceptor());
+        // 注册流媒体 WebSocket 处理器
+        registry.addHandler(tyStreamHandler, "/ws/stream/**")
+                .setAllowedOrigins("*")
+                .addInterceptors(new TyHandshakeInterceptor()); // 可选的拦截器
     }
 }
\ No newline at end of file

--
Gitblit v1.9.3