org.springframework.web.socket.server.HandshakeHandler Java Examples

@Test
public void handshakeHandlerPassedToSockJsRegistration() {
	WebSocketHandler handler = new TextWebSocketHandler();
	HandshakeHandler handshakeHandler = new DefaultHandshakeHandler();

	this.registration.addHandler(handler, "/foo").setHandshakeHandler(handshakeHandler).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);

	WebSocketTransportHandler transportHandler =
			(WebSocketTransportHandler) mapping.sockJsService.getTransportHandlers().get(TransportType.WEBSOCKET);
	assertSame(handshakeHandler, transportHandler.getHandshakeHandler());
}
 

@Test
public void handshakeHandlerPassedToSockJsRegistration() {
	WebSocketHandler handler = new TextWebSocketHandler();
	HandshakeHandler handshakeHandler = new DefaultHandshakeHandler();

	this.registration.addHandler(handler, "/foo").setHandshakeHandler(handshakeHandler).withSockJS();
	this.registration.getSockJsServiceRegistration().setTaskScheduler(this.taskScheduler);

	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);

	WebSocketTransportHandler transportHandler =
			(WebSocketTransportHandler) mapping.sockJsService.getTransportHandlers().get(TransportType.WEBSOCKET);
	assertSame(handshakeHandler, transportHandler.getHandshakeHandler());
}
 

@Test
public void handshakeHandlerPassedToSockJsRegistration() {
	WebSocketHandler handler = new TextWebSocketHandler();
	HandshakeHandler handshakeHandler = new DefaultHandshakeHandler();

	this.registration.addHandler(handler, "/foo").setHandshakeHandler(handshakeHandler).withSockJS();
	this.registration.getSockJsServiceRegistration().setTaskScheduler(this.taskScheduler);

	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);

	WebSocketTransportHandler transportHandler =
			(WebSocketTransportHandler) mapping.sockJsService.getTransportHandlers().get(TransportType.WEBSOCKET);
	assertSame(handshakeHandler, transportHandler.getHandshakeHandler());
}
 

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;
}
 

@Override
protected void addWebSocketHandlerMapping(MultiValueMap<HttpRequestHandler, String> mappings,
		WebSocketHandler webSocketHandler, HandshakeHandler handshakeHandler,
		HandshakeInterceptor[] interceptors, String path) {

	WebSocketHttpRequestHandler httpHandler =
			new WebSocketHttpRequestHandler(webSocketHandler, handshakeHandler);

	if (!ObjectUtils.isEmpty(interceptors)) {
		httpHandler.setHandshakeInterceptors(Arrays.asList(interceptors));
	}
	mappings.add(httpHandler, path);
}
 

@Override
protected void addWebSocketHandlerMapping(MultiValueMap<HttpRequestHandler, String> mappings,
		WebSocketHandler webSocketHandler, HandshakeHandler handshakeHandler,
		HandshakeInterceptor[] interceptors, String path) {

	WebSocketHttpRequestHandler httpHandler =
			new WebSocketHttpRequestHandler(webSocketHandler, handshakeHandler);

	if (!ObjectUtils.isEmpty(interceptors)) {
		httpHandler.setHandshakeInterceptors(Arrays.asList(interceptors));
	}
	mappings.add(httpHandler, path);
}
 

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;
}
 

@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);
}
 

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;
}
 

private HandshakeHandler getOrCreateHandshakeHandler() {
	return (this.handshakeHandler != null ? this.handshakeHandler : new DefaultHandshakeHandler());
}
 

@Nullable
protected HandshakeHandler getHandshakeHandler() {
	return this.handshakeHandler;
}
 

protected abstract void addWebSocketHandlerMapping(M mappings, WebSocketHandler wsHandler,
HandshakeHandler handshakeHandler, HandshakeInterceptor[] interceptors, String path);
 

@Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
	HandshakeHandler handshakeHandler = new DefaultHandshakeHandler(this.upgradeStrategy);
	registry.addHandler(new EchoHandler(), "/echo").setHandshakeHandler(handshakeHandler).withSockJS();
	registry.addHandler(testServerHandler(), "/test").setHandshakeHandler(handshakeHandler).withSockJS();
}
 

@Override
protected void addWebSocketHandlerMapping(List<Mapping> mappings, WebSocketHandler handler,
		HandshakeHandler handshakeHandler, HandshakeInterceptor[] interceptors, String path) {

	mappings.add(new Mapping(handler, path, handshakeHandler, interceptors));
}
 

public WebSocketHttpRequestHandler(WebSocketHandler wsHandler, HandshakeHandler handshakeHandler) {
	Assert.notNull(wsHandler, "wsHandler must not be null");
	Assert.notNull(handshakeHandler, "handshakeHandler must not be null");
	this.wsHandler = new ExceptionWebSocketHandlerDecorator(new LoggingWebSocketHandlerDecorator(wsHandler));
	this.handshakeHandler = handshakeHandler;
}
 

/**
 * Return the HandshakeHandler.
 */
public HandshakeHandler getHandshakeHandler() {
	return this.handshakeHandler;
}
 

public WebSocketTransportHandler(HandshakeHandler handshakeHandler) {
	Assert.notNull(handshakeHandler, "handshakeHandler must not be null");
	this.handshakeHandler = handshakeHandler;
}
 

public HandshakeHandler getHandshakeHandler() {
	return this.handshakeHandler;
}
 

@Override
public StompWebSocketEndpointRegistration setHandshakeHandler(HandshakeHandler handshakeHandler) {
	Assert.notNull(handshakeHandler, "'handshakeHandler' must not be null");
	this.handshakeHandler = handshakeHandler;
	return this;
}
 

@Override
public WebSocketHandlerRegistration setHandshakeHandler(HandshakeHandler handshakeHandler) {
	this.handshakeHandler = handshakeHandler;
	return this;
}
 

protected HandshakeHandler getHandshakeHandler() {
	return this.handshakeHandler;
}
 

private HandshakeHandler getOrCreateHandshakeHandler() {
	return (this.handshakeHandler != null) ? this.handshakeHandler : new DefaultHandshakeHandler();
}
 

protected abstract void addWebSocketHandlerMapping(M mappings, WebSocketHandler wsHandler,
HandshakeHandler handshakeHandler, HandshakeInterceptor[] interceptors, String path);
 

@Override
public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
	HandshakeHandler handshakeHandler = new DefaultHandshakeHandler(this.upgradeStrategy);
	registry.addHandler(new EchoHandler(), "/echo").setHandshakeHandler(handshakeHandler).withSockJS();
	registry.addHandler(testServerHandler(), "/test").setHandshakeHandler(handshakeHandler).withSockJS();
}
 

@Override
protected void addWebSocketHandlerMapping(List<Mapping> mappings, WebSocketHandler handler,
		HandshakeHandler handshakeHandler, HandshakeInterceptor[] interceptors, String path) {

	mappings.add(new Mapping(handler, path, handshakeHandler, interceptors));
}
 

@Override
public WebSocketHandlerRegistration setHandshakeHandler(@Nullable HandshakeHandler handshakeHandler) {
	this.handshakeHandler = handshakeHandler;
	return this;
}
 

/**
 * Return the HandshakeHandler.
 */
public HandshakeHandler getHandshakeHandler() {
	return this.handshakeHandler;
}
 

public WebSocketTransportHandler(HandshakeHandler handshakeHandler) {
	Assert.notNull(handshakeHandler, "HandshakeHandler must not be null");
	this.handshakeHandler = handshakeHandler;
}
 

public HandshakeHandler getHandshakeHandler() {
	return this.handshakeHandler;
}
 

@Override
public StompWebSocketEndpointRegistration setHandshakeHandler(HandshakeHandler handshakeHandler) {
	this.handshakeHandler = handshakeHandler;
	return this;
}