wms / wms_tianjinggelinmei · Files

package com.huaheng.framework.config;

import com.huaheng.framework.websocket.SessionAuthHandshakeInterceptor;
import com.huaheng.framework.websocket.StompHandshakeHandler;
import com.huaheng.framework.websocket.WebsocketConstants;
import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

	@Override
	public void configureMessageBroker(MessageBrokerRegistry config) {

		config.enableSimpleBroker(WebsocketConstants.BROKER_TOPIC, WebsocketConstants.BROKER_QUEUE);
		config.setUserDestinationPrefix(WebsocketConstants.USER_PRIFIX);
		config.setApplicationDestinationPrefixes(WebsocketConstants.APP_PRIFIX);
	}

	@Override
	public void registerStompEndpoints(StompEndpointRegistry registry) {
		registry.addEndpoint(WebsocketConstants.ENDPOINT)
				.setHandshakeHandler(new StompHandshakeHandler())
				.addInterceptors(new SessionAuthHandshakeInterceptor())
				.withSockJS();
	}

}