Examples with SimpAnnotationMethodMessageHandler org.springframework.messaging.simp.annotation.support.SimpAnnotationMethodMessageHandler used on opensource projects

Example 1 with SimpAnnotationMethodMessageHandler

use of org.springframework.messaging.simp.annotation.support.SimpAnnotationMethodMessageHandler in project spring-framework by spring-projects.

the class AbstractMessageBrokerConfiguration method simpAnnotationMethodMessageHandler.

@Bean
public SimpAnnotationMethodMessageHandler simpAnnotationMethodMessageHandler() {
    SimpAnnotationMethodMessageHandler handler = createAnnotationMethodMessageHandler();
    handler.setDestinationPrefixes(getBrokerRegistry().getApplicationDestinationPrefixes());
    handler.setMessageConverter(brokerMessageConverter());
    handler.setValidator(simpValidator());
    List<HandlerMethodArgumentResolver> argumentResolvers = new ArrayList<>();
    addArgumentResolvers(argumentResolvers);
    handler.setCustomArgumentResolvers(argumentResolvers);
    List<HandlerMethodReturnValueHandler> returnValueHandlers = new ArrayList<>();
    addReturnValueHandlers(returnValueHandlers);
    handler.setCustomReturnValueHandlers(returnValueHandlers);
    PathMatcher pathMatcher = getBrokerRegistry().getPathMatcher();
    if (pathMatcher != null) {
        handler.setPathMatcher(pathMatcher);
    }
    return handler;
}

Example 2 with SimpAnnotationMethodMessageHandler

use of org.springframework.messaging.simp.annotation.support.SimpAnnotationMethodMessageHandler in project spring-framework by spring-projects.

the class TestValidator method customChannels.

@Test
public void customChannels() {
    loadBeanDefinitions("websocket-config-broker-customchannels.xml");
    SimpAnnotationMethodMessageHandler annotationMethodMessageHandler = this.appContext.getBean(SimpAnnotationMethodMessageHandler.class);
    Validator validator = annotationMethodMessageHandler.getValidator();
    assertThat(validator).isNotNull();
    assertThat(validator).isSameAs(this.appContext.getBean("myValidator"));
    assertThat(validator).isInstanceOf(TestValidator.class);
    List<Class<? extends MessageHandler>> subscriberTypes = Arrays.asList(SimpAnnotationMethodMessageHandler.class, UserDestinationMessageHandler.class, SimpleBrokerMessageHandler.class);
    testChannel("clientInboundChannel", subscriberTypes, 3);
    testExecutor("clientInboundChannel", 100, 200, 600);
    subscriberTypes = Collections.singletonList(SubProtocolWebSocketHandler.class);
    testChannel("clientOutboundChannel", subscriberTypes, 3);
    testExecutor("clientOutboundChannel", 101, 201, 601);
    subscriberTypes = Arrays.asList(SimpleBrokerMessageHandler.class, UserDestinationMessageHandler.class);
    testChannel("brokerChannel", subscriberTypes, 1);
    testExecutor("brokerChannel", 102, 202, 602);
}

Example 3 with SimpAnnotationMethodMessageHandler

use of org.springframework.messaging.simp.annotation.support.SimpAnnotationMethodMessageHandler in project spring-framework by spring-projects.

the class AbstractMessageBrokerConfiguration method simpAnnotationMethodMessageHandler.

@Bean
public SimpAnnotationMethodMessageHandler simpAnnotationMethodMessageHandler(AbstractSubscribableChannel clientInboundChannel, AbstractSubscribableChannel clientOutboundChannel, SimpMessagingTemplate brokerMessagingTemplate, CompositeMessageConverter brokerMessageConverter) {
    SimpAnnotationMethodMessageHandler handler = createAnnotationMethodMessageHandler(clientInboundChannel, clientOutboundChannel, brokerMessagingTemplate);
    MessageBrokerRegistry brokerRegistry = getBrokerRegistry(clientInboundChannel, clientOutboundChannel);
    handler.setDestinationPrefixes(brokerRegistry.getApplicationDestinationPrefixes());
    handler.setMessageConverter(brokerMessageConverter);
    handler.setValidator(simpValidator());
    List<HandlerMethodArgumentResolver> argumentResolvers = new ArrayList<>();
    addArgumentResolvers(argumentResolvers);
    handler.setCustomArgumentResolvers(argumentResolvers);
    List<HandlerMethodReturnValueHandler> returnValueHandlers = new ArrayList<>();
    addReturnValueHandlers(returnValueHandlers);
    handler.setCustomReturnValueHandlers(returnValueHandlers);
    PathMatcher pathMatcher = brokerRegistry.getPathMatcher();
    if (pathMatcher != null) {
        handler.setPathMatcher(pathMatcher);
    }
    return handler;
}

Example 4 with SimpAnnotationMethodMessageHandler

use of org.springframework.messaging.simp.annotation.support.SimpAnnotationMethodMessageHandler in project spring-framework by spring-projects.

the class MessageBrokerConfigurationTests method customArgumentAndReturnValueTypes.

@Test
public void customArgumentAndReturnValueTypes() {
    ApplicationContext context = loadConfig(CustomConfig.class);
    SimpAnnotationMethodMessageHandler handler = context.getBean(SimpAnnotationMethodMessageHandler.class);
    List<HandlerMethodArgumentResolver> customResolvers = handler.getCustomArgumentResolvers();
    assertThat(customResolvers.size()).isEqualTo(1);
    assertThat(handler.getArgumentResolvers().contains(customResolvers.get(0))).isTrue();
    List<HandlerMethodReturnValueHandler> customHandlers = handler.getCustomReturnValueHandlers();
    assertThat(customHandlers.size()).isEqualTo(1);
    assertThat(handler.getReturnValueHandlers().contains(customHandlers.get(0))).isTrue();
}

Example 5 with SimpAnnotationMethodMessageHandler

use of org.springframework.messaging.simp.annotation.support.SimpAnnotationMethodMessageHandler in project spring-framework by spring-projects.

the class MessageBrokerConfigurationTests method clientOutboundChannelUsedByAnnotatedMethod.

@Test
public void clientOutboundChannelUsedByAnnotatedMethod() {
    ApplicationContext context = loadConfig(SimpleBrokerConfig.class);
    TestChannel channel = context.getBean("clientOutboundChannel", TestChannel.class);
    SimpAnnotationMethodMessageHandler messageHandler = context.getBean(SimpAnnotationMethodMessageHandler.class);
    StompHeaderAccessor headers = StompHeaderAccessor.create(StompCommand.SUBSCRIBE);
    headers.setSessionId("sess1");
    headers.setSessionAttributes(new ConcurrentHashMap<>());
    headers.setSubscriptionId("subs1");
    headers.setDestination("/foo");
    Message<?> message = MessageBuilder.withPayload(new byte[0]).setHeaders(headers).build();
    messageHandler.handleMessage(message);
    message = channel.messages.get(0);
    headers = StompHeaderAccessor.wrap(message);
    assertThat(headers.getMessageType()).isEqualTo(SimpMessageType.MESSAGE);
    assertThat(headers.getDestination()).isEqualTo("/foo");
    assertThat(new String((byte[]) message.getPayload())).isEqualTo("bar");
}