Java HandshakeInterceptor类代码示例

import org.springframework.web.socket.server.HandshakeInterceptor; //导入依赖的package包/类
public boolean applyBeforeHandshake(ServerHttpRequest request, ServerHttpResponse response,
		Map<String, Object> attributes) throws Exception {

	for (int i = 0; i < this.interceptors.size(); i++) {
		HandshakeInterceptor interceptor = this.interceptors.get(i);
		if (!interceptor.beforeHandshake(request, response, this.wsHandler, attributes)) {
			if (logger.isDebugEnabled()) {
				logger.debug(interceptor + " returns false from beforeHandshake - precluding handshake");
			}
			applyAfterHandshake(request, response, null);
			return false;
		}
		this.interceptorIndex = i;
	}
	return true;
}
 

import org.springframework.web.socket.server.HandshakeInterceptor; //导入依赖的package包/类
@Override
public SockJsServiceRegistration withSockJS() {
	this.registration = new StompSockJsServiceRegistration(this.sockJsTaskScheduler);
	HandshakeInterceptor[] interceptors = getInterceptors();
	if (interceptors.length > 0) {
		this.registration.setInterceptors(interceptors);
	}
	if (this.handshakeHandler != null) {
		WebSocketTransportHandler transportHandler = new WebSocketTransportHandler(this.handshakeHandler);
		this.registration.setTransportHandlerOverrides(transportHandler);
	}
	if (!this.allowedOrigins.isEmpty()) {
		this.registration.setAllowedOrigins(this.allowedOrigins.toArray(new String[this.allowedOrigins.size()]));
	}
	return this.registration;
}
 

import org.springframework.web.socket.server.HandshakeInterceptor; //导入依赖的package包/类
@Override
public SockJsServiceRegistration withSockJS() {
	this.sockJsServiceRegistration = new SockJsServiceRegistration(this.sockJsTaskScheduler);
	HandshakeInterceptor[] interceptors = getInterceptors();
	if (interceptors.length > 0) {
		this.sockJsServiceRegistration.setInterceptors(interceptors);
	}
	if (this.handshakeHandler != null) {
		WebSocketTransportHandler transportHandler = new WebSocketTransportHandler(this.handshakeHandler);
		this.sockJsServiceRegistration.setTransportHandlerOverrides(transportHandler);
	}
	if (!this.allowedOrigins.isEmpty()) {
		this.sockJsServiceRegistration.setAllowedOrigins(this.allowedOrigins.toArray(new String[this.allowedOrigins.size()]));
	}
	return this.sockJsServiceRegistration;
}
 

import org.springframework.web.socket.server.HandshakeInterceptor; //导入依赖的package包/类
@Test
public void interceptorsPassedToSockJsRegistration() {
	WebSocketHandler handler = new TextWebSocketHandler();
	HttpSessionHandshakeInterceptor interceptor = new HttpSessionHandshakeInterceptor();

	this.registration.addHandler(handler, "/foo").addInterceptors(interceptor)
			.setAllowedOrigins("http://mydomain1.com").withSockJS();

	List<Mapping> mappings = this.registration.getMappings();
	assertEquals(1, mappings.size());

	Mapping mapping = mappings.get(0);
	assertEquals(handler, mapping.webSocketHandler);
	assertEquals("/foo/**", mapping.path);
	assertNotNull(mapping.sockJsService);
	assertTrue(mapping.sockJsService.getAllowedOrigins().contains("http://mydomain1.com"));
	List<HandshakeInterceptor> interceptors = mapping.sockJsService.getHandshakeInterceptors();
	assertEquals(interceptor, interceptors.get(0));
	assertEquals(OriginHandshakeInterceptor.class, interceptors.get(1).getClass());
}
 

import org.springframework.web.socket.server.HandshakeInterceptor; //导入依赖的package包/类
@Override
public SockJsServiceRegistration withSockJS() {
	this.registration = new WampSockJsServiceRegistration(this.sockJsTaskScheduler);
	HandshakeInterceptor[] handshakeInterceptors = getInterceptors();
	if (handshakeInterceptors.length > 0) {
		this.registration.setInterceptors(handshakeInterceptors);
	}
	if (this.handshakeHandler != null) {
		WebSocketTransportHandler transportHandler = new WebSocketTransportHandler(
				this.handshakeHandler);
		this.registration.setTransportHandlerOverrides(transportHandler);
	}
	if (!this.allowedOrigins.isEmpty()) {
		this.registration.setAllowedOrigins(
				this.allowedOrigins.toArray(new String[this.allowedOrigins.size()]));
	}
	return this.registration;
}
 

import org.springframework.web.socket.server.HandshakeInterceptor; //导入依赖的package包/类
@Bean
public HandlerMapping wampWebSocketHandlerMapping() {
	WebSocketHandler handler = subProtocolWebSocketHandler();
	handler = decorateWebSocketHandler(handler);

	WebMvcWampEndpointRegistry registry = new WebMvcWampEndpointRegistry(handler,
			getTransportRegistration(), messageBrokerSockJsTaskScheduler(),
			new MappingJsonFactory(lookupObjectMapper()));

	List<HandshakeInterceptor> handshakeInterceptors = new ArrayList<>();
	addHandshakeInterceptors(handshakeInterceptors);
	registry.addHandshakeInterceptors(handshakeInterceptors);

	registerWampEndpoints(registry);

	return registry.getHandlerMapping();
}
 

import org.springframework.web.socket.server.HandshakeInterceptor; //导入依赖的package包/类
public void applyAfterHandshake(ServerHttpRequest request, ServerHttpResponse response, Exception failure) {
	for (int i = this.interceptorIndex; i >= 0; i--) {
		HandshakeInterceptor interceptor = this.interceptors.get(i);
		try {
			interceptor.afterHandshake(request, response, this.wsHandler, failure);
		}
		catch (Throwable ex) {
			if (logger.isWarnEnabled()) {
				logger.warn(interceptor + " threw exception in afterHandshake: " + ex);
			}
		}
	}
}
 

import org.springframework.web.socket.server.HandshakeInterceptor; //导入依赖的package包/类
/**
 * Configure one or more WebSocket handshake request interceptors.
 */
public void setHandshakeInterceptors(List<HandshakeInterceptor> interceptors) {
	this.interceptors.clear();
	if (interceptors != null) {
		this.interceptors.addAll(interceptors);
	}
}
 

import org.springframework.web.socket.server.HandshakeInterceptor; //导入依赖的package包/类
@Override
protected void addWebSocketHandlerMapping(MultiValueMap<HttpRequestHandler, String> mappings,
		WebSocketHandler wsHandler, HandshakeHandler handshakeHandler,
		HandshakeInterceptor[] interceptors, String path) {

	WebSocketHttpRequestHandler httpHandler = new WebSocketHttpRequestHandler(wsHandler, handshakeHandler);
	if (!ObjectUtils.isEmpty(interceptors)) {
		httpHandler.setHandshakeInterceptors(Arrays.asList(interceptors));
	}
	mappings.add(httpHandler, path);
}
 

import org.springframework.web.socket.server.HandshakeInterceptor; //导入依赖的package包/类
@Override
public StompWebSocketEndpointRegistration addInterceptors(HandshakeInterceptor... interceptors) {
	if (!ObjectUtils.isEmpty(interceptors)) {
		this.interceptors.addAll(Arrays.asList(interceptors));
	}
	return this;
}
 

import org.springframework.web.socket.server.HandshakeInterceptor; //导入依赖的package包/类
public SockJsServiceRegistration setInterceptors(HandshakeInterceptor... interceptors) {
	this.interceptors.clear();
	if (!ObjectUtils.isEmpty(interceptors)) {
		this.interceptors.addAll(Arrays.asList(interceptors));
	}
	return this;
}
 

import org.springframework.web.socket.server.HandshakeInterceptor; //导入依赖的package包/类
@Override
public WebSocketHandlerRegistration addInterceptors(HandshakeInterceptor... interceptors) {
	if (!ObjectUtils.isEmpty(interceptors)) {
		this.interceptors.addAll(Arrays.asList(interceptors));
	}
	return this;
}
 

import org.springframework.web.socket.server.HandshakeInterceptor; //导入依赖的package包/类
@Before
public void setup() {
	i1 = mock(HandshakeInterceptor.class);
	i2 = mock(HandshakeInterceptor.class);
	i3 = mock(HandshakeInterceptor.class);
	interceptors = Arrays.asList(i1, i2, i3);
	wsHandler = mock(WebSocketHandler.class);
	attributes = new HashMap<String, Object>();
}
 

import org.springframework.web.socket.server.HandshakeInterceptor; //导入依赖的package包/类
public Mapping(WebSocketHandler h, String path, HandshakeHandler hh, HandshakeInterceptor[] interceptors) {
	this.webSocketHandler = h;
	this.path = path;
	this.handshakeHandler = hh;
	this.interceptors = interceptors;
	this.sockJsService = null;
}
 

import org.springframework.web.socket.server.HandshakeInterceptor; //导入依赖的package包/类
@Test
@SuppressWarnings("unchecked")
public void webSocketHandlersAttributes() {
	loadBeanDefinitions("websocket-config-handlers-attributes.xml");

	HandlerMapping handlerMapping = this.appContext.getBean(HandlerMapping.class);
	assertNotNull(handlerMapping);
	assertTrue(handlerMapping instanceof SimpleUrlHandlerMapping);

	SimpleUrlHandlerMapping urlHandlerMapping = (SimpleUrlHandlerMapping) handlerMapping;
	assertEquals(2, urlHandlerMapping.getOrder());

	WebSocketHttpRequestHandler handler = (WebSocketHttpRequestHandler) urlHandlerMapping.getUrlMap().get("/foo");
	assertNotNull(handler);
	unwrapAndCheckDecoratedHandlerType(handler.getWebSocketHandler(), FooWebSocketHandler.class);
	HandshakeHandler handshakeHandler = handler.getHandshakeHandler();
	assertNotNull(handshakeHandler);
	assertTrue(handshakeHandler instanceof TestHandshakeHandler);
	List<HandshakeInterceptor> interceptors = handler.getHandshakeInterceptors();
	assertThat(interceptors, contains(instanceOf(FooTestInterceptor.class),
			instanceOf(BarTestInterceptor.class), instanceOf(OriginHandshakeInterceptor.class)));

	handler = (WebSocketHttpRequestHandler) urlHandlerMapping.getUrlMap().get("/test");
	assertNotNull(handler);
	unwrapAndCheckDecoratedHandlerType(handler.getWebSocketHandler(), TestWebSocketHandler.class);
	handshakeHandler = handler.getHandshakeHandler();
	assertNotNull(handshakeHandler);
	assertTrue(handshakeHandler instanceof TestHandshakeHandler);
	interceptors = handler.getHandshakeInterceptors();
	assertThat(interceptors, contains(instanceOf(FooTestInterceptor.class),
			instanceOf(BarTestInterceptor.class), instanceOf(OriginHandshakeInterceptor.class)));
}