org.springframework.messaging.simp.user.DefaultUserDestinationResolver Java Examples

@Test
public void customPathMatcher() {
	ApplicationContext context = loadConfig(CustomConfig.class);

	SimpleBrokerMessageHandler broker = context.getBean(SimpleBrokerMessageHandler.class);
	DefaultSubscriptionRegistry registry = (DefaultSubscriptionRegistry) broker.getSubscriptionRegistry();
	assertEquals("a.a", registry.getPathMatcher().combine("a", "a"));

	PathMatcher pathMatcher =
			context.getBean(SimpAnnotationMethodMessageHandler.class).getPathMatcher();

	assertEquals("a.a", pathMatcher.combine("a", "a"));

	DefaultUserDestinationResolver resolver = context.getBean(DefaultUserDestinationResolver.class);
	assertNotNull(resolver);
	assertEquals(false, resolver.isRemoveLeadingSlash());
}
 

@Test
public void customPathMatcher() {
	ApplicationContext context = loadConfig(CustomConfig.class);

	SimpleBrokerMessageHandler broker = context.getBean(SimpleBrokerMessageHandler.class);
	DefaultSubscriptionRegistry registry = (DefaultSubscriptionRegistry) broker.getSubscriptionRegistry();
	assertEquals("a.a", registry.getPathMatcher().combine("a", "a"));

	PathMatcher pathMatcher =
			context.getBean(SimpAnnotationMethodMessageHandler.class).getPathMatcher();

	assertEquals("a.a", pathMatcher.combine("a", "a"));

	DefaultUserDestinationResolver resolver = context.getBean(DefaultUserDestinationResolver.class);
	assertNotNull(resolver);
	assertEquals(false, resolver.isRemoveLeadingSlash());
}
 

@Bean
public UserDestinationResolver userDestinationResolver() {
	DefaultUserDestinationResolver resolver = new DefaultUserDestinationResolver(userRegistry());
	String prefix = getBrokerRegistry().getUserDestinationPrefix();
	if (prefix != null) {
		resolver.setUserDestinationPrefix(prefix);
	}
	return resolver;
}
 

private RuntimeBeanReference registerUserDestResolver(Element brokerElem,
		RuntimeBeanReference userRegistry, ParserContext context, @Nullable Object source) {

	RootBeanDefinition beanDef = new RootBeanDefinition(DefaultUserDestinationResolver.class);
	beanDef.getConstructorArgumentValues().addIndexedArgumentValue(0, userRegistry);
	if (brokerElem.hasAttribute("user-destination-prefix")) {
		beanDef.getPropertyValues().add("userDestinationPrefix", brokerElem.getAttribute("user-destination-prefix"));
	}
	if (brokerElem.hasAttribute("path-matcher")) {
		String pathMatcherRef = brokerElem.getAttribute("path-matcher");
		beanDef.getPropertyValues().add("pathMatcher", new RuntimeBeanReference(pathMatcherRef));
	}
	return new RuntimeBeanReference(registerBeanDef(beanDef, context, source));
}
 

@Bean
public UserDestinationResolver userDestinationResolver() {
	DefaultUserDestinationResolver resolver = new DefaultUserDestinationResolver(userRegistry());
	String prefix = getBrokerRegistry().getUserDestinationPrefix();
	if (prefix != null) {
		resolver.setUserDestinationPrefix(prefix);
	}
	return resolver;
}
 

private RuntimeBeanReference registerUserDestResolver(Element brokerElem,
		RuntimeBeanReference userRegistry, ParserContext context, @Nullable Object source) {

	RootBeanDefinition beanDef = new RootBeanDefinition(DefaultUserDestinationResolver.class);
	beanDef.getConstructorArgumentValues().addIndexedArgumentValue(0, userRegistry);
	if (brokerElem.hasAttribute("user-destination-prefix")) {
		beanDef.getPropertyValues().add("userDestinationPrefix", brokerElem.getAttribute("user-destination-prefix"));
	}
	if (brokerElem.hasAttribute("path-matcher")) {
		String pathMatcherRef = brokerElem.getAttribute("path-matcher");
		beanDef.getPropertyValues().add("pathMatcher", new RuntimeBeanReference(pathMatcherRef));
	}
	return new RuntimeBeanReference(registerBeanDef(beanDef, context, source));
}
 

@Bean
public UserDestinationResolver userDestinationResolver() {
	DefaultUserDestinationResolver resolver = new DefaultUserDestinationResolver(userRegistry());
	String prefix = getBrokerRegistry().getUserDestinationPrefix();
	if (prefix != null) {
		resolver.setUserDestinationPrefix(prefix);
	}
	return resolver;
}
 

private RuntimeBeanReference registerUserDestResolver(Element brokerElem,
		RuntimeBeanReference userRegistry, ParserContext context, Object source) {

	RootBeanDefinition beanDef = new RootBeanDefinition(DefaultUserDestinationResolver.class);
	beanDef.getConstructorArgumentValues().addIndexedArgumentValue(0, userRegistry);
	if (brokerElem.hasAttribute("user-destination-prefix")) {
		beanDef.getPropertyValues().add("userDestinationPrefix", brokerElem.getAttribute("user-destination-prefix"));
	}
	return new RuntimeBeanReference(registerBeanDef(beanDef, context, source));
}
 

private void updateUserDestinationResolver(AbstractBrokerMessageHandler handler) {
	Collection<String> prefixes = handler.getDestinationPrefixes();
	if (!prefixes.isEmpty() && !prefixes.iterator().next().startsWith("/")) {
		((DefaultUserDestinationResolver) userDestinationResolver()).setRemoveLeadingSlash(true);
	}
}
 

@Test
public void stompBrokerRelay() {
	loadBeanDefinitions("websocket-config-broker-relay.xml");

	HandlerMapping hm = this.appContext.getBean(HandlerMapping.class);
	assertNotNull(hm);
	assertThat(hm, Matchers.instanceOf(SimpleUrlHandlerMapping.class));

	SimpleUrlHandlerMapping suhm = (SimpleUrlHandlerMapping) hm;
	assertThat(suhm.getUrlMap().keySet(), Matchers.hasSize(1));
	assertThat(suhm.getUrlMap().values(), Matchers.hasSize(1));
	assertEquals(2, suhm.getOrder());

	HttpRequestHandler httpRequestHandler = (HttpRequestHandler) suhm.getUrlMap().get("/foo/**");
	assertNotNull(httpRequestHandler);
	assertThat(httpRequestHandler, Matchers.instanceOf(SockJsHttpRequestHandler.class));
	SockJsHttpRequestHandler sockJsHttpRequestHandler = (SockJsHttpRequestHandler) httpRequestHandler;
	WebSocketHandler wsHandler = unwrapWebSocketHandler(sockJsHttpRequestHandler.getWebSocketHandler());
	assertNotNull(wsHandler);
	assertThat(wsHandler, Matchers.instanceOf(SubProtocolWebSocketHandler.class));
	assertNotNull(sockJsHttpRequestHandler.getSockJsService());

	UserDestinationResolver userDestResolver = this.appContext.getBean(UserDestinationResolver.class);
	assertNotNull(userDestResolver);
	assertThat(userDestResolver, Matchers.instanceOf(DefaultUserDestinationResolver.class));
	DefaultUserDestinationResolver defaultUserDestResolver = (DefaultUserDestinationResolver) userDestResolver;
	assertEquals("/user/", defaultUserDestResolver.getDestinationPrefix());

	StompBrokerRelayMessageHandler messageBroker = this.appContext.getBean(StompBrokerRelayMessageHandler.class);
	assertNotNull(messageBroker);
	assertEquals("clientlogin", messageBroker.getClientLogin());
	assertEquals("clientpass", messageBroker.getClientPasscode());
	assertEquals("syslogin", messageBroker.getSystemLogin());
	assertEquals("syspass", messageBroker.getSystemPasscode());
	assertEquals("relayhost", messageBroker.getRelayHost());
	assertEquals(1234, messageBroker.getRelayPort());
	assertEquals("spring.io", messageBroker.getVirtualHost());
	assertEquals(5000, messageBroker.getSystemHeartbeatReceiveInterval());
	assertEquals(5000, messageBroker.getSystemHeartbeatSendInterval());
	assertThat(messageBroker.getDestinationPrefixes(), Matchers.containsInAnyOrder("/topic","/queue"));
	assertTrue(messageBroker.isPreservePublishOrder());

	List<Class<? extends MessageHandler>> subscriberTypes = Arrays.asList(SimpAnnotationMethodMessageHandler.class,
			UserDestinationMessageHandler.class, StompBrokerRelayMessageHandler.class);
	testChannel("clientInboundChannel", subscriberTypes, 2);
	testExecutor("clientInboundChannel", Runtime.getRuntime().availableProcessors() * 2, Integer.MAX_VALUE, 60);

	subscriberTypes = Collections.singletonList(SubProtocolWebSocketHandler.class);
	testChannel("clientOutboundChannel", subscriberTypes, 2);
	testExecutor("clientOutboundChannel", Runtime.getRuntime().availableProcessors() * 2, Integer.MAX_VALUE, 60);

	subscriberTypes = Arrays.asList(StompBrokerRelayMessageHandler.class, UserDestinationMessageHandler.class);
	testChannel("brokerChannel", subscriberTypes, 1);
	try {
		this.appContext.getBean("brokerChannelExecutor", ThreadPoolTaskExecutor.class);
		fail("expected exception");
	}
	catch (NoSuchBeanDefinitionException ex) {
		// expected
	}

	String destination = "/topic/unresolved-user-destination";
	UserDestinationMessageHandler userDestHandler = this.appContext.getBean(UserDestinationMessageHandler.class);
	assertEquals(destination, userDestHandler.getBroadcastDestination());
	assertNotNull(messageBroker.getSystemSubscriptions());
	assertSame(userDestHandler, messageBroker.getSystemSubscriptions().get(destination));

	destination = "/topic/simp-user-registry";
	UserRegistryMessageHandler userRegistryHandler = this.appContext.getBean(UserRegistryMessageHandler.class);
	assertEquals(destination, userRegistryHandler.getBroadcastDestination());
	assertNotNull(messageBroker.getSystemSubscriptions());
	assertSame(userRegistryHandler, messageBroker.getSystemSubscriptions().get(destination));

	SimpUserRegistry userRegistry = this.appContext.getBean(SimpUserRegistry.class);
	assertEquals(MultiServerUserRegistry.class, userRegistry.getClass());

	String name = "webSocketMessageBrokerStats";
	WebSocketMessageBrokerStats stats = this.appContext.getBean(name, WebSocketMessageBrokerStats.class);
	String actual = stats.toString();
	String expected = "WebSocketSession\\[0 current WS\\(0\\)-HttpStream\\(0\\)-HttpPoll\\(0\\), " +
			"0 total, 0 closed abnormally \\(0 connect failure, 0 send limit, 0 transport error\\)\\], " +
			"stompSubProtocol\\[processed CONNECT\\(0\\)-CONNECTED\\(0\\)-DISCONNECT\\(0\\)\\], " +
			"stompBrokerRelay\\[0 sessions, relayhost:1234 \\(not available\\), " +
			"processed CONNECT\\(0\\)-CONNECTED\\(0\\)-DISCONNECT\\(0\\)\\], " +
			"inboundChannel\\[pool size = \\d, active threads = \\d, queued tasks = \\d, " +
			"completed tasks = \\d\\], " +
			"outboundChannel\\[pool size = \\d, active threads = \\d, queued tasks = \\d, " +
			"completed tasks = \\d\\], " +
			"sockJsScheduler\\[pool size = \\d, active threads = \\d, queued tasks = \\d, " +
			"completed tasks = \\d\\]";

	assertTrue("\nExpected: " + expected.replace("\\", "") + "\n  Actual: " + actual, actual.matches(expected));
}
 

private void updateUserDestinationResolver(AbstractBrokerMessageHandler handler) {
	Collection<String> prefixes = handler.getDestinationPrefixes();
	if (!prefixes.isEmpty() && !prefixes.iterator().next().startsWith("/")) {
		((DefaultUserDestinationResolver) userDestinationResolver()).setRemoveLeadingSlash(true);
	}
}
 

@Test
public void stompBrokerRelay() {
	loadBeanDefinitions("websocket-config-broker-relay.xml");

	HandlerMapping hm = this.appContext.getBean(HandlerMapping.class);
	assertNotNull(hm);
	assertThat(hm, Matchers.instanceOf(SimpleUrlHandlerMapping.class));

	SimpleUrlHandlerMapping suhm = (SimpleUrlHandlerMapping) hm;
	assertThat(suhm.getUrlMap().keySet(), Matchers.hasSize(1));
	assertThat(suhm.getUrlMap().values(), Matchers.hasSize(1));
	assertEquals(2, suhm.getOrder());

	HttpRequestHandler httpRequestHandler = (HttpRequestHandler) suhm.getUrlMap().get("/foo/**");
	assertNotNull(httpRequestHandler);
	assertThat(httpRequestHandler, Matchers.instanceOf(SockJsHttpRequestHandler.class));
	SockJsHttpRequestHandler sockJsHttpRequestHandler = (SockJsHttpRequestHandler) httpRequestHandler;
	WebSocketHandler wsHandler = unwrapWebSocketHandler(sockJsHttpRequestHandler.getWebSocketHandler());
	assertNotNull(wsHandler);
	assertThat(wsHandler, Matchers.instanceOf(SubProtocolWebSocketHandler.class));
	assertNotNull(sockJsHttpRequestHandler.getSockJsService());

	UserDestinationResolver userDestResolver = this.appContext.getBean(UserDestinationResolver.class);
	assertNotNull(userDestResolver);
	assertThat(userDestResolver, Matchers.instanceOf(DefaultUserDestinationResolver.class));
	DefaultUserDestinationResolver defaultUserDestResolver = (DefaultUserDestinationResolver) userDestResolver;
	assertEquals("/user/", defaultUserDestResolver.getDestinationPrefix());

	StompBrokerRelayMessageHandler messageBroker = this.appContext.getBean(StompBrokerRelayMessageHandler.class);
	assertNotNull(messageBroker);
	assertEquals("clientlogin", messageBroker.getClientLogin());
	assertEquals("clientpass", messageBroker.getClientPasscode());
	assertEquals("syslogin", messageBroker.getSystemLogin());
	assertEquals("syspass", messageBroker.getSystemPasscode());
	assertEquals("relayhost", messageBroker.getRelayHost());
	assertEquals(1234, messageBroker.getRelayPort());
	assertEquals("spring.io", messageBroker.getVirtualHost());
	assertEquals(5000, messageBroker.getSystemHeartbeatReceiveInterval());
	assertEquals(5000, messageBroker.getSystemHeartbeatSendInterval());
	assertThat(messageBroker.getDestinationPrefixes(), Matchers.containsInAnyOrder("/topic","/queue"));
	assertTrue(messageBroker.isPreservePublishOrder());

	List<Class<? extends MessageHandler>> subscriberTypes = Arrays.asList(SimpAnnotationMethodMessageHandler.class,
			UserDestinationMessageHandler.class, StompBrokerRelayMessageHandler.class);
	testChannel("clientInboundChannel", subscriberTypes, 2);
	testExecutor("clientInboundChannel", Runtime.getRuntime().availableProcessors() * 2, Integer.MAX_VALUE, 60);

	subscriberTypes = Collections.singletonList(SubProtocolWebSocketHandler.class);
	testChannel("clientOutboundChannel", subscriberTypes, 2);
	testExecutor("clientOutboundChannel", Runtime.getRuntime().availableProcessors() * 2, Integer.MAX_VALUE, 60);

	subscriberTypes = Arrays.asList(StompBrokerRelayMessageHandler.class, UserDestinationMessageHandler.class);
	testChannel("brokerChannel", subscriberTypes, 1);
	try {
		this.appContext.getBean("brokerChannelExecutor", ThreadPoolTaskExecutor.class);
		fail("expected exception");
	}
	catch (NoSuchBeanDefinitionException ex) {
		// expected
	}

	String destination = "/topic/unresolved-user-destination";
	UserDestinationMessageHandler userDestHandler = this.appContext.getBean(UserDestinationMessageHandler.class);
	assertEquals(destination, userDestHandler.getBroadcastDestination());
	assertNotNull(messageBroker.getSystemSubscriptions());
	assertSame(userDestHandler, messageBroker.getSystemSubscriptions().get(destination));

	destination = "/topic/simp-user-registry";
	UserRegistryMessageHandler userRegistryHandler = this.appContext.getBean(UserRegistryMessageHandler.class);
	assertEquals(destination, userRegistryHandler.getBroadcastDestination());
	assertNotNull(messageBroker.getSystemSubscriptions());
	assertSame(userRegistryHandler, messageBroker.getSystemSubscriptions().get(destination));

	SimpUserRegistry userRegistry = this.appContext.getBean(SimpUserRegistry.class);
	assertEquals(MultiServerUserRegistry.class, userRegistry.getClass());

	String name = "webSocketMessageBrokerStats";
	WebSocketMessageBrokerStats stats = this.appContext.getBean(name, WebSocketMessageBrokerStats.class);
	String actual = stats.toString();
	String expected = "WebSocketSession\\[0 current WS\\(0\\)-HttpStream\\(0\\)-HttpPoll\\(0\\), " +
			"0 total, 0 closed abnormally \\(0 connect failure, 0 send limit, 0 transport error\\)\\], " +
			"stompSubProtocol\\[processed CONNECT\\(0\\)-CONNECTED\\(0\\)-DISCONNECT\\(0\\)\\], " +
			"stompBrokerRelay\\[0 sessions, relayhost:1234 \\(not available\\), " +
			"processed CONNECT\\(0\\)-CONNECTED\\(0\\)-DISCONNECT\\(0\\)\\], " +
			"inboundChannel\\[pool size = \\d, active threads = \\d, queued tasks = \\d, " +
			"completed tasks = \\d\\], " +
			"outboundChannel\\[pool size = \\d, active threads = \\d, queued tasks = \\d, " +
			"completed tasks = \\d\\], " +
			"sockJsScheduler\\[pool size = \\d, active threads = \\d, queued tasks = \\d, " +
			"completed tasks = \\d\\]";

	assertTrue("\nExpected: " + expected.replace("\\", "") + "\n  Actual: " + actual, actual.matches(expected));
}
 

@Test
public void stompBrokerRelay() {
	loadBeanDefinitions("websocket-config-broker-relay.xml");

	HandlerMapping hm = this.appContext.getBean(HandlerMapping.class);
	assertNotNull(hm);
	assertThat(hm, Matchers.instanceOf(SimpleUrlHandlerMapping.class));

	SimpleUrlHandlerMapping suhm = (SimpleUrlHandlerMapping) hm;
	assertThat(suhm.getUrlMap().keySet(), Matchers.hasSize(1));
	assertThat(suhm.getUrlMap().values(), Matchers.hasSize(1));
	assertEquals(2, suhm.getOrder());

	HttpRequestHandler httpRequestHandler = (HttpRequestHandler) suhm.getUrlMap().get("/foo/**");
	assertNotNull(httpRequestHandler);
	assertThat(httpRequestHandler, Matchers.instanceOf(SockJsHttpRequestHandler.class));
	SockJsHttpRequestHandler sockJsHttpRequestHandler = (SockJsHttpRequestHandler) httpRequestHandler;
	WebSocketHandler wsHandler = unwrapWebSocketHandler(sockJsHttpRequestHandler.getWebSocketHandler());
	assertNotNull(wsHandler);
	assertThat(wsHandler, Matchers.instanceOf(SubProtocolWebSocketHandler.class));
	assertNotNull(sockJsHttpRequestHandler.getSockJsService());

	UserDestinationResolver userDestResolver = this.appContext.getBean(UserDestinationResolver.class);
	assertNotNull(userDestResolver);
	assertThat(userDestResolver, Matchers.instanceOf(DefaultUserDestinationResolver.class));
	DefaultUserDestinationResolver defaultUserDestResolver = (DefaultUserDestinationResolver) userDestResolver;
	assertEquals("/user/", defaultUserDestResolver.getDestinationPrefix());

	StompBrokerRelayMessageHandler messageBroker = this.appContext.getBean(StompBrokerRelayMessageHandler.class);
	assertNotNull(messageBroker);
	assertEquals("clientlogin", messageBroker.getClientLogin());
	assertEquals("clientpass", messageBroker.getClientPasscode());
	assertEquals("syslogin", messageBroker.getSystemLogin());
	assertEquals("syspass", messageBroker.getSystemPasscode());
	assertEquals("relayhost", messageBroker.getRelayHost());
	assertEquals(1234, messageBroker.getRelayPort());
	assertEquals("spring.io", messageBroker.getVirtualHost());
	assertEquals(5000, messageBroker.getSystemHeartbeatReceiveInterval());
	assertEquals(5000, messageBroker.getSystemHeartbeatSendInterval());
	assertThat(messageBroker.getDestinationPrefixes(), Matchers.containsInAnyOrder("/topic","/queue"));

	List<Class<? extends MessageHandler>> subscriberTypes =
			Arrays.<Class<? extends MessageHandler>>asList(SimpAnnotationMethodMessageHandler.class,
					UserDestinationMessageHandler.class, StompBrokerRelayMessageHandler.class);
	testChannel("clientInboundChannel", subscriberTypes, 2);
	testExecutor("clientInboundChannel", Runtime.getRuntime().availableProcessors() * 2, Integer.MAX_VALUE, 60);

	subscriberTypes = Collections.singletonList(SubProtocolWebSocketHandler.class);
	testChannel("clientOutboundChannel", subscriberTypes, 1);
	testExecutor("clientOutboundChannel", Runtime.getRuntime().availableProcessors() * 2, Integer.MAX_VALUE, 60);

	subscriberTypes = Arrays.<Class<? extends MessageHandler>>asList(
			StompBrokerRelayMessageHandler.class, UserDestinationMessageHandler.class);
	testChannel("brokerChannel", subscriberTypes, 1);
	try {
		this.appContext.getBean("brokerChannelExecutor", ThreadPoolTaskExecutor.class);
		fail("expected exception");
	}
	catch (NoSuchBeanDefinitionException ex) {
		// expected
	}

	String destination = "/topic/unresolved-user-destination";
	UserDestinationMessageHandler userDestHandler = this.appContext.getBean(UserDestinationMessageHandler.class);
	assertEquals(destination, userDestHandler.getBroadcastDestination());
	assertNotNull(messageBroker.getSystemSubscriptions());
	assertSame(userDestHandler, messageBroker.getSystemSubscriptions().get(destination));

	destination = "/topic/simp-user-registry";
	UserRegistryMessageHandler userRegistryHandler = this.appContext.getBean(UserRegistryMessageHandler.class);
	assertEquals(destination, userRegistryHandler.getBroadcastDestination());
	assertNotNull(messageBroker.getSystemSubscriptions());
	assertSame(userRegistryHandler, messageBroker.getSystemSubscriptions().get(destination));

	SimpUserRegistry userRegistry = this.appContext.getBean(SimpUserRegistry.class);
	assertEquals(MultiServerUserRegistry.class, userRegistry.getClass());

	String name = "webSocketMessageBrokerStats";
	WebSocketMessageBrokerStats stats = this.appContext.getBean(name, WebSocketMessageBrokerStats.class);
	String actual = stats.toString();
	String expected = "WebSocketSession\\[0 current WS\\(0\\)-HttpStream\\(0\\)-HttpPoll\\(0\\), " +
			"0 total, 0 closed abnormally \\(0 connect failure, 0 send limit, 0 transport error\\)\\], " +
			"stompSubProtocol\\[processed CONNECT\\(0\\)-CONNECTED\\(0\\)-DISCONNECT\\(0\\)\\], " +
			"stompBrokerRelay\\[0 sessions, relayhost:1234 \\(not available\\), processed CONNECT\\(0\\)-CONNECTED\\(0\\)-DISCONNECT\\(0\\)\\], " +
			"inboundChannel\\[pool size = \\d, active threads = \\d, queued tasks = \\d, completed tasks = \\d\\], " +
			"outboundChannelpool size = \\d, active threads = \\d, queued tasks = \\d, completed tasks = \\d\\], " +
			"sockJsScheduler\\[pool size = \\d, active threads = \\d, queued tasks = \\d, completed tasks = \\d\\]";

	assertTrue("\nExpected: " + expected.replace("\\", "") + "\n  Actual: " + actual, actual.matches(expected));
}