Search in sources :

Example 11 with MqttPahoMessageHandler

use of org.springframework.integration.mqtt.outbound.MqttPahoMessageHandler in project spring-integration by spring-projects.

the class BackToBackAdapterTests method testAsyncPersisted.

@Test
public void testAsyncPersisted() throws Exception {
    DefaultMqttPahoClientFactory factory = new DefaultMqttPahoClientFactory();
    MqttClientPersistence persistence = new MqttDefaultFilePersistence(folder.getRoot().getAbsolutePath());
    factory.setPersistence(persistence);
    MqttPahoMessageHandler adapter = new MqttPahoMessageHandler("tcp://localhost:1883", "si-test-out", factory);
    adapter.setDefaultTopic("mqtt-foo");
    adapter.setBeanFactory(mock(BeanFactory.class));
    adapter.setAsync(true);
    adapter.setAsyncEvents(true);
    adapter.setDefaultQos(1);
    EventPublisher publisher1 = new EventPublisher();
    adapter.setApplicationEventPublisher(publisher1);
    adapter.afterPropertiesSet();
    adapter.start();
    MqttPahoMessageDrivenChannelAdapter inbound = new MqttPahoMessageDrivenChannelAdapter("tcp://localhost:1883", "si-test-in", "mqtt-foo", "mqtt-bar");
    QueueChannel outputChannel = new QueueChannel();
    inbound.setOutputChannel(outputChannel);
    ThreadPoolTaskScheduler taskScheduler = new ThreadPoolTaskScheduler();
    taskScheduler.initialize();
    inbound.setTaskScheduler(taskScheduler);
    inbound.setBeanFactory(mock(BeanFactory.class));
    inbound.afterPropertiesSet();
    inbound.start();
    Message<String> message1 = new GenericMessage<String>("foo");
    adapter.handleMessage(message1);
    verifyEvents(adapter, publisher1, message1);
    Message<String> message2 = MessageBuilder.withPayload("bar").setHeader(MqttHeaders.TOPIC, "mqtt-bar").build();
    EventPublisher publisher2 = new EventPublisher();
    adapter.setApplicationEventPublisher(publisher2);
    adapter.handleMessage(message2);
    verifyEvents(adapter, publisher2, message2);
    verifyMessageIds(publisher1, publisher2);
    int clientInstance = publisher1.delivered.getClientInstance();
    adapter.stop();
    // new client instance
    adapter.start();
    publisher1 = new EventPublisher();
    adapter.setApplicationEventPublisher(publisher1);
    adapter.handleMessage(message1);
    verifyEvents(adapter, publisher1, message1);
    publisher2 = new EventPublisher();
    adapter.setApplicationEventPublisher(publisher2);
    adapter.handleMessage(message2);
    verifyEvents(adapter, publisher2, message2);
    verifyMessageIds(publisher1, publisher2);
    assertNotEquals(clientInstance, publisher1.delivered.getClientInstance());
    Message<?> out = null;
    for (int i = 0; i < 4; i++) {
        out = outputChannel.receive(20000);
        assertNotNull(out);
        if ("foo".equals(out.getPayload())) {
            assertEquals("mqtt-foo", out.getHeaders().get(MqttHeaders.RECEIVED_TOPIC));
        } else if ("bar".equals(out.getPayload())) {
            assertEquals("mqtt-bar", out.getHeaders().get(MqttHeaders.RECEIVED_TOPIC));
        } else {
            fail("unexpected payload " + out.getPayload());
        }
    }
    adapter.stop();
    inbound.stop();
}
Also used : DefaultMqttPahoClientFactory(org.springframework.integration.mqtt.core.DefaultMqttPahoClientFactory) ApplicationEventPublisher(org.springframework.context.ApplicationEventPublisher) QueueChannel(org.springframework.integration.channel.QueueChannel) MqttClientPersistence(org.eclipse.paho.client.mqttv3.MqttClientPersistence) ThreadPoolTaskScheduler(org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler) GenericMessage(org.springframework.messaging.support.GenericMessage) MqttPahoMessageDrivenChannelAdapter(org.springframework.integration.mqtt.inbound.MqttPahoMessageDrivenChannelAdapter) MqttPahoMessageHandler(org.springframework.integration.mqtt.outbound.MqttPahoMessageHandler) BeanFactory(org.springframework.beans.factory.BeanFactory) MqttDefaultFilePersistence(org.eclipse.paho.client.mqttv3.persist.MqttDefaultFilePersistence) Test(org.junit.Test)

Example 12 with MqttPahoMessageHandler

use of org.springframework.integration.mqtt.outbound.MqttPahoMessageHandler in project spring-integration by spring-projects.

the class DownstreamExceptionTests method testWithErrorChannel.

@Test
public void testWithErrorChannel() throws Exception {
    assertSame(this.errors, TestUtils.getPropertyValue(this.withErrorChannel, "errorChannel"));
    service.n = 0;
    MqttPahoMessageHandler adapter = new MqttPahoMessageHandler("tcp://localhost:1883", "si-test-out");
    adapter.setDefaultTopic("mqtt-fooEx2");
    adapter.setBeanFactory(mock(BeanFactory.class));
    adapter.afterPropertiesSet();
    adapter.start();
    adapter.handleMessage(new GenericMessage<String>("foo"));
    service.barrier.await(10, TimeUnit.SECONDS);
    service.barrier.reset();
    adapter.handleMessage(new GenericMessage<String>("foo"));
    service.barrier.await(10, TimeUnit.SECONDS);
    assertNotNull(errors.receive(10000));
    service.barrier.reset();
    adapter.stop();
}
Also used : MqttPahoMessageHandler(org.springframework.integration.mqtt.outbound.MqttPahoMessageHandler) BeanFactory(org.springframework.beans.factory.BeanFactory) ArgumentMatchers.anyString(org.mockito.ArgumentMatchers.anyString) Test(org.junit.Test)

Aggregations

MqttPahoMessageHandler (org.springframework.integration.mqtt.outbound.MqttPahoMessageHandler)12 Test (org.junit.Test)10 BeanFactory (org.springframework.beans.factory.BeanFactory)9 QueueChannel (org.springframework.integration.channel.QueueChannel)6 MqttPahoMessageDrivenChannelAdapter (org.springframework.integration.mqtt.inbound.MqttPahoMessageDrivenChannelAdapter)6 ThreadPoolTaskScheduler (org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler)6 ArgumentMatchers.anyString (org.mockito.ArgumentMatchers.anyString)4 GenericMessage (org.springframework.messaging.support.GenericMessage)3 ApplicationEventPublisher (org.springframework.context.ApplicationEventPublisher)2 Bean (org.springframework.context.annotation.Bean)2 DefaultMqttPahoClientFactory (org.springframework.integration.mqtt.core.DefaultMqttPahoClientFactory)2 DefaultPahoMessageConverter (org.springframework.integration.mqtt.support.DefaultPahoMessageConverter)2 Properties (java.util.Properties)1 CountDownLatch (java.util.concurrent.CountDownLatch)1 AtomicBoolean (java.util.concurrent.atomic.AtomicBoolean)1 SocketFactory (javax.net.SocketFactory)1 Log (org.apache.commons.logging.Log)1 IMqttToken (org.eclipse.paho.client.mqttv3.IMqttToken)1 MqttAsyncClient (org.eclipse.paho.client.mqttv3.MqttAsyncClient)1 MqttClientPersistence (org.eclipse.paho.client.mqttv3.MqttClientPersistence)1