import javax.annotation.Resource;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.EnableWebMvc;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
import com.hncy.optima.websocket.handler.MessageHandshakeInterceptor;
import com.hncy.optima.websocket.handler.MessageWebSocketHandler;
@Configuration
@EnableWebMvc
@EnableWebSocket
public class WebConfig extends WebMvcConfigurerAdapter implements WebSocketConfigurer{
@Resource
private MessageWebSocketHandler messageWebSocketHandler;
@Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
registry.addHandler(messageWebSocketHandler,"/message.msg").addInterceptors(new MessageHandshakeInterceptor());
registry.addHandler(messageWebSocketHandler,"/sockjs/message.msg").addInterceptors(new MessageHandshakeInterceptor())
.withSockJS();
}
@Bean
public WebSocketHandler messageWebSocketHandler(){
return new MessageWebSocketHandler();
}
}