fix: 依赖改名

community-8073/src/main/java/com/ivmiku/tutorial/config/WebSocketConfig.java

@@ -0,0 +1,34 @@
+package com.ivmiku.tutorial.config;
+
+import jakarta.servlet.ServletContext;
+import jakarta.servlet.ServletException;
+import org.springframework.boot.web.servlet.ServletContextInitializer;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.web.socket.server.standard.ServerEndpointExporter;
+import org.springframework.web.socket.server.standard.ServletServerContainerFactoryBean;
+import org.springframework.web.util.WebAppRootListener;
+
+@Configuration
+public class WebSocketConfig implements ServletContextInitializer {
+    @Bean
+    public ServerEndpointExporter serverEndpointExporter (){
+        return new ServerEndpointExporter();
+    }
+
+    @Override
+    public void onStartup(ServletContext servletContext) throws ServletException {
+        servletContext.addListener(WebAppRootListener.class);
+        servletContext.setInitParameter("org.apache.tomcat.websocket.textBufferSize","102400000");
+    }
+
+    @Bean
+    public ServletServerContainerFactoryBean createWebSocketContainer() {
+        ServletServerContainerFactoryBean container = new ServletServerContainerFactoryBean();
+        // 在此处设置bufferSize
+        container.setMaxTextMessageBufferSize(50*1024*1024);
+        container.setMaxBinaryMessageBufferSize(50*1024*1024);
+        container.setMaxSessionIdleTimeout(15 * 60000L);
+        return container;
+    }
+}

community-8073/src/main/java/com/ivmiku/tutorial/controller/WebSocketServer.java

@@ -8,7 +8,9 @@ import com.ivmiku.tutorial.service.impl.SearchService;
 import jakarta.websocket.*;
 import jakarta.websocket.server.PathParam;
 import jakarta.websocket.server.ServerEndpoint;
+import org.springframework.beans.BeansException;
 import org.springframework.context.ApplicationContext;
+import org.springframework.context.ApplicationContextAware;
 import org.springframework.stereotype.Controller;
 
 import java.io.IOException;
@@ -18,7 +20,7 @@ import java.util.concurrent.ConcurrentHashMap;
 
 @Controller
 @ServerEndpoint(value = "/search/{satoken}")
-public class WebSocketServer {
+public class WebSocketServer implements ApplicationContextAware {
     public static Map<String, Session> sessionMap = new ConcurrentHashMap<>();
     private static ApplicationContext applicationContext;
     private SearchService searchService;
@@ -61,7 +63,7 @@ public class WebSocketServer {
             }
             case '#' -> {
                 String sub = message.substring(1);
-                List<String> list = searchService.getTag(message);
+                List<String> list = searchService.getTag(sub);
                 session.getBasicRemote().sendText(JSON.toJSONString(Result.ok(list)));
             }
             //预留给社区搜索
@@ -70,4 +72,9 @@ public class WebSocketServer {
             }
         }
     }
+
+    @Override
+    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
+        WebSocketServer.applicationContext = applicationContext;
+    }
 }