Examples with MessageGroupStore org.springframework.integration.store.MessageGroupStore used on opensource projects

Example 11 with MessageGroupStore

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

the class AbstractMongoDbMessageGroupStoreTests method testCountMessagesInGroup.

@Test
@MongoDbAvailable
public void testCountMessagesInGroup() throws Exception {
    this.cleanupCollections(new SimpleMongoDbFactory(new MongoClient(), "test"));
    MessageGroupStore store = this.getMessageGroupStore();
    Message<?> messageA = new GenericMessage<String>("A");
    Message<?> messageB = new GenericMessage<String>("B");
    store.addMessagesToGroup(1, messageA, messageB);
    assertEquals(2, store.messageGroupSize(1));
}

Example 12 with MessageGroupStore

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

the class AbstractMongoDbMessageGroupStoreTests method testMessageGroupUpdatedDateChangesWithEachAddedMessage.

@Test
@MongoDbAvailable
public void testMessageGroupUpdatedDateChangesWithEachAddedMessage() throws Exception {
    this.cleanupCollections(new SimpleMongoDbFactory(new MongoClient(), "test"));
    MessageGroupStore store = this.getMessageGroupStore();
    MessageGroup messageGroup = store.getMessageGroup(1);
    Message<?> message = new GenericMessage<String>("Hello");
    messageGroup = store.addMessageToGroup(1, message);
    assertNotNull(messageGroup);
    assertEquals(1, messageGroup.size());
    long createdTimestamp = messageGroup.getTimestamp();
    long updatedTimestamp = messageGroup.getLastModified();
    assertEquals(createdTimestamp, updatedTimestamp);
    Thread.sleep(10);
    message = new GenericMessage<String>("Hello again");
    messageGroup = store.addMessageToGroup(1, message);
    createdTimestamp = messageGroup.getTimestamp();
    updatedTimestamp = messageGroup.getLastModified();
    assertTrue(updatedTimestamp > createdTimestamp);
    assertEquals(2, messageGroup.size());
    // make sure the store is properly rebuild from MongoDB
    store = this.getMessageGroupStore();
    messageGroup = store.getMessageGroup(1);
    assertEquals(2, messageGroup.size());
}

Example 13 with MessageGroupStore

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

the class AbstractMongoDbMessageGroupStoreTests method testRemoveMessageFromTheGroup.

@Test
@MongoDbAvailable
public void testRemoveMessageFromTheGroup() throws Exception {
    this.cleanupCollections(new SimpleMongoDbFactory(new MongoClient(), "test"));
    MessageGroupStore store = this.getMessageGroupStore();
    MessageGroup messageGroup = store.getMessageGroup(1);
    Message<?> message = new GenericMessage<String>("2");
    store.addMessagesToGroup(1, new GenericMessage<String>("1"), message);
    messageGroup = store.addMessageToGroup(1, new GenericMessage<String>("3"));
    assertNotNull(messageGroup);
    assertEquals(3, messageGroup.size());
    store.removeMessagesFromGroup(1, message);
    messageGroup = store.getMessageGroup(1);
    assertEquals(2, messageGroup.size());
}

Example 14 with MessageGroupStore

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

the class AbstractMongoDbMessageGroupStoreTests method testAddAndRemoveMessagesFromMessageGroup.

@Test
@MongoDbAvailable
public void testAddAndRemoveMessagesFromMessageGroup() throws Exception {
    MessageGroupStore messageStore = (MessageGroupStore) this.getMessageStore();
    String groupId = "X";
    messageStore.removeMessageGroup("X");
    ((AbstractBatchingMessageGroupStore) messageStore).setRemoveBatchSize(10);
    List<Message<?>> messages = new ArrayList<Message<?>>();
    for (int i = 0; i < 25; i++) {
        Message<String> message = MessageBuilder.withPayload("foo").setCorrelationId(groupId).build();
        messageStore.addMessagesToGroup(groupId, message);
        messages.add(message);
    }
    MessageGroup group = messageStore.getMessageGroup(groupId);
    assertEquals(25, group.size());
    messageStore.removeMessagesFromGroup(groupId, messages);
    group = messageStore.getMessageGroup(groupId);
    assertEquals(0, group.size());
}

Example 15 with MessageGroupStore

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

the class AbstractMongoDbMessageGroupStoreTests method testMultipleMessageStores.

@Test
@MongoDbAvailable
public void testMultipleMessageStores() throws Exception {
    this.cleanupCollections(new SimpleMongoDbFactory(new MongoClient(), "test"));
    MessageGroupStore store1 = this.getMessageGroupStore();
    MessageGroupStore store2 = this.getMessageGroupStore();
    Message<?> message = new GenericMessage<String>("1");
    store1.addMessagesToGroup(1, message, new GenericMessage<String>("2"), new GenericMessage<String>("3"));
    MessageGroupStore store3 = this.getMessageGroupStore();
    MessageGroup messageGroup = store3.getMessageGroup(1);
    assertNotNull(messageGroup);
    assertEquals(3, messageGroup.size());
    store3.removeMessagesFromGroup(1, message);
    messageGroup = store2.getMessageGroup(1);
    assertEquals(2, messageGroup.size());
}