Search in sources :

Example 96 with MessageGroup

use of org.springframework.integration.store.MessageGroup in project spring-integration by spring-projects.

the class RedisMessageGroupStoreTests method testMessageGroupWithAddedMessage.

@Test
@RedisAvailable
public void testMessageGroupWithAddedMessage() {
    RedisConnectionFactory jcf = getConnectionFactoryForTest();
    RedisMessageStore store = new RedisMessageStore(jcf);
    Message<?> message = new GenericMessage<>("Hello");
    MessageGroup messageGroup = store.addMessageToGroup(this.groupId, message);
    assertEquals(1, messageGroup.size());
    // make sure the store is properly rebuild from Redis
    store = new RedisMessageStore(jcf);
    messageGroup = store.getMessageGroup(this.groupId);
    assertEquals(1, messageGroup.size());
}
Also used : GenericMessage(org.springframework.messaging.support.GenericMessage) MessageGroup(org.springframework.integration.store.MessageGroup) SimpleMessageGroup(org.springframework.integration.store.SimpleMessageGroup) RedisConnectionFactory(org.springframework.data.redis.connection.RedisConnectionFactory) RedisAvailable(org.springframework.integration.redis.rules.RedisAvailable) Test(org.junit.Test)

Example 97 with MessageGroup

use of org.springframework.integration.store.MessageGroup in project spring-integration by spring-projects.

the class RedisMessageGroupStoreTests method testRemoveMessageFromTheGroup.

@Test
@RedisAvailable
public void testRemoveMessageFromTheGroup() {
    RedisConnectionFactory jcf = getConnectionFactoryForTest();
    RedisMessageStore store = new RedisMessageStore(jcf);
    MessageGroup messageGroup = store.getMessageGroup(this.groupId);
    Message<?> message = new GenericMessage<>("2");
    store.addMessagesToGroup(messageGroup.getGroupId(), new GenericMessage<>("1"), message);
    messageGroup = store.addMessageToGroup(messageGroup.getGroupId(), new GenericMessage<>("3"));
    assertEquals(3, messageGroup.size());
    store.removeMessagesFromGroup(this.groupId, message);
    messageGroup = store.getMessageGroup(this.groupId);
    assertEquals(2, messageGroup.size());
    // make sure the store is properly rebuild from Redis
    store = new RedisMessageStore(jcf);
    messageGroup = store.getMessageGroup(this.groupId);
    assertEquals(2, messageGroup.size());
}
Also used : GenericMessage(org.springframework.messaging.support.GenericMessage) MessageGroup(org.springframework.integration.store.MessageGroup) SimpleMessageGroup(org.springframework.integration.store.SimpleMessageGroup) RedisConnectionFactory(org.springframework.data.redis.connection.RedisConnectionFactory) RedisAvailable(org.springframework.integration.redis.rules.RedisAvailable) Test(org.junit.Test)

Example 98 with MessageGroup

use of org.springframework.integration.store.MessageGroup in project spring-integration by spring-projects.

the class RedisMessageGroupStoreTests method testIteratorOfMessageGroups.

@Test
@RedisAvailable
public void testIteratorOfMessageGroups() {
    RedisConnectionFactory jcf = getConnectionFactoryForTest();
    RedisMessageStore store1 = new RedisMessageStore(jcf);
    RedisMessageStore store2 = new RedisMessageStore(jcf);
    store1.removeMessageGroup(this.groupId);
    UUID group2 = UUID.randomUUID();
    store1.removeMessageGroup(group2);
    UUID group3 = UUID.randomUUID();
    store1.removeMessageGroup(group3);
    store1.addMessagesToGroup(this.groupId, new GenericMessage<>("1"));
    store2.addMessagesToGroup(group2, new GenericMessage<>("2"));
    store1.addMessagesToGroup(group3, new GenericMessage<>("3"), new GenericMessage<>("3A"));
    Iterator<MessageGroup> messageGroups = store1.iterator();
    int counter = 0;
    while (messageGroups.hasNext()) {
        MessageGroup group = messageGroups.next();
        String groupId = (String) group.getGroupId();
        if (groupId.equals("1")) {
            assertEquals(1, group.getMessages().size());
        } else if (groupId.equals("2")) {
            assertEquals(1, group.getMessages().size());
        } else if (groupId.equals("3")) {
            assertEquals(2, group.getMessages().size());
        }
        counter++;
    }
    assertEquals(3, counter);
    store2.removeMessageGroup(group3);
    messageGroups = store1.iterator();
    counter = 0;
    while (messageGroups.hasNext()) {
        messageGroups.next();
        counter++;
    }
    assertEquals(2, counter);
}
Also used : MessageGroup(org.springframework.integration.store.MessageGroup) SimpleMessageGroup(org.springframework.integration.store.SimpleMessageGroup) Matchers.containsString(org.hamcrest.Matchers.containsString) UUID(java.util.UUID) RedisConnectionFactory(org.springframework.data.redis.connection.RedisConnectionFactory) RedisAvailable(org.springframework.integration.redis.rules.RedisAvailable) Test(org.junit.Test)

Aggregations

MessageGroup (org.springframework.integration.store.MessageGroup)98 Test (org.junit.Test)79 SimpleMessageGroup (org.springframework.integration.store.SimpleMessageGroup)54 GenericMessage (org.springframework.messaging.support.GenericMessage)36 MessageGroupStore (org.springframework.integration.store.MessageGroupStore)25 Message (org.springframework.messaging.Message)20 ArrayList (java.util.ArrayList)19 RedisAvailable (org.springframework.integration.redis.rules.RedisAvailable)15 RedisConnectionFactory (org.springframework.data.redis.connection.RedisConnectionFactory)14 MongoDbAvailable (org.springframework.integration.mongodb.rules.MongoDbAvailable)13 AbstractBatchingMessageGroupStore (org.springframework.integration.store.AbstractBatchingMessageGroupStore)13 MongoClient (com.mongodb.MongoClient)12 SimpleMongoDbFactory (org.springframework.data.mongodb.core.SimpleMongoDbFactory)12 Transactional (org.springframework.transaction.annotation.Transactional)10 LinkedList (java.util.LinkedList)8 List (java.util.List)7 Matchers.containsString (org.hamcrest.Matchers.containsString)7 UUID (java.util.UUID)6 DirectFieldAccessor (org.springframework.beans.DirectFieldAccessor)6 HashMap (java.util.HashMap)5