src/main/java/com/example/websocketproxy/config/WebSocketConfig.java - master - dixsu/websocket-proxy

src/main/java/com/example/websocketproxy/config/WebSocketConfig.java
105 строк5 KB
Сахно Роман Александрович
начал делать wss подключение с tls сертификатами
15 фев 2025, 19:09
15 фев 2025, 19:09512630f
package com.example.websocketproxy.config;

import com.example.websocketproxy.websocket.WebSocketProxyHandler;
import com.example.websocketproxy.websocket.ProxyWebSocketHandler;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.client.RestTemplate;
import org.springframework.web.servlet.config.annotation.CorsRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
import org.springframework.web.socket.config.annotation.*;

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer, WebSocketMessageBrokerConfigurer {
   public static final int BUFFER_SIZE = 4096;
   public static final boolean DEBUG = true;
   private final ProxyWebSocketHandler proxyWebSocketHandler;
    private final WebSocketProxyHandler webSocketProxyHandler;


    public WebSocketConfig(ProxyWebSocketHandler proxyWebSocketHandler, WebSocketProxyHandler webSocketProxyHandler) {
        this.proxyWebSocketHandler = proxyWebSocketHandler;
        this.webSocketProxyHandler = webSocketProxyHandler;
    }

    @Override
    public void configureWebSocketTransport(WebSocketTransportRegistration registry) {
        // Увеличиваем размер сообщений, если это необходимо
        registry.setMessageSizeLimit(10 * 1024 * 1024); // Максимальный размер сообщения
        registry.setSendBufferSizeLimit(10 * 1024 * 1024); // Максимальный размер буфера отправки
      //  registry.setSendTimeLimit(60 * 1000); // Таймаут отправки
    //    registry.setSupportsPartialMessages(true); // Включаем поддержку фрагментированных сообщений
    }


    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        // Регистрируем ProxyWebSocketHandler на /proxy-ws
        registry.addHandler(proxyWebSocketHandler, "/proxy-ws")
                .setAllowedOrigins("*"); // Разрешаем запросы с любых доменов

        // Регистрируем WebSocketProxyHandler на /device-ws
        registry.addHandler(webSocketProxyHandler, "/ws")
//                .setAllowedOrigins("*");  // Разрешаем запросы с любых доменов
         .setAllowedOriginPatterns("http://localhost:*", "https://*.example.com");
//                .setAllowedOriginPatterns("https://*.example.com", "http://localhost:*","0.0.0.0","127.0.0.1") // Используем allowedOriginPatterns
//                .setAllowedOrigins(null) // Полностью отключаем проверку источников
//                .withSockJS(); // Добавляем поддержку SockJS для обратной совместимости
    }

    // Bean для RestTemplate, если требуется HTTP-клиент
    @Bean
    public RestTemplate restTemplate() {
        return new RestTemplate();
    }


}








//package com.example.websocketproxy.config;
//
//import com.example.websocketproxy.websocket.WebSocketProxyHandler;
//import org.springframework.context.annotation.Bean;
//import org.springframework.context.annotation.Configuration;
//import org.springframework.web.client.RestTemplate;
//import org.springframework.web.socket.config.annotation.*;
//import com.example.websocketproxy.websocket.ProxyWebSocketHandler;
//
//@Configuration
//@EnableWebSocket
//public class WebSocketConfig implements WebSocketConfigurer, WebSocketMessageBrokerConfigurer {
//
//    private final ProxyWebSocketHandler proxyWebSocketHandler;
//    private final WebSocketProxyHandler webSocketProxyHandler;
//
//    public WebSocketConfig(ProxyWebSocketHandler proxyWebSocketHandler, WebSocketProxyHandler webSocketProxyHandler) {
//        this.proxyWebSocketHandler = proxyWebSocketHandler;
//        this.webSocketProxyHandler = webSocketProxyHandler;
//    }
//    @Override
//    public void configureWebSocketTransport(WebSocketTransportRegistration registry) {
//        registry.setMessageSizeLimit(65536); // Увеличьте лимит
//    }
//    @Override
//    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
//        registry.addHandler(proxyWebSocketHandler, "/ws")
//                .setAllowedOrigins("*"); // Разрешить запросы с любых доменов
//        registry.addHandler(webSocketProxyHandler, "/ws")
//                .setAllowedOrigins("*"); // Разрешить запросы с любых доменов
//    }
//
////Добавьте этот метод
////Метод restTemplate в контексте Spring-приложений используется для выполнения HTTP-запросов к RESTful веб-сервисам. Он позволяет отправлять запросы и получать ответы от внешних API
////    @Bean
////    public RestTemplate restTemplate() {
////        return new RestTemplate();
////    }
//}