package com.juick.ws.configuration; import com.juick.ws.WebsocketComponent; import com.juick.ws.XMPPConnection; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.PropertySource; import org.springframework.core.env.Environment; import org.springframework.jdbc.core.JdbcTemplate; import org.springframework.web.socket.config.annotation.EnableWebSocket; import org.springframework.web.socket.config.annotation.ServletWebSocketHandlerRegistry; import org.springframework.web.socket.config.annotation.WebSocketConfigurer; import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry; import org.springframework.web.socket.server.standard.ServletServerContainerFactoryBean; import javax.inject.Inject; /** * Created by aalexeev on 11/24/16. */ @Configuration @EnableWebSocket @PropertySource("classpath:juick.conf") class WebsocketConfiguration implements WebSocketConfigurer { @Inject private Environment env; @Inject private JdbcTemplate jdbc; @Bean public WebsocketComponent wsHandler() { return new WebsocketComponent(jdbc); } @Bean public XMPPConnection ws() { return new XMPPConnection(env, wsHandler(), jdbc); } @Override public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) { ((ServletWebSocketHandlerRegistry) registry).setOrder(2); registry.addHandler(wsHandler(), "/**").setAllowedOrigins("*"); } @Bean public ServletServerContainerFactoryBean createWebSocketContainer() { ServletServerContainerFactoryBean container = new ServletServerContainerFactoryBean(); container.setMaxTextMessageBufferSize(8192); container.setMaxBinaryMessageBufferSize(8192); return container; } }