Fix KV Stores for same message in multiple groups (#8737)

* Fix KV Stores for same message in multiple groups

If same message is stored into different groups with the same KV store,
the removal of one group would lead to removal the message for the other one.

* Improve KV Store to save message with the key including a group id
* Respectively, refine the removal API to include group id into keys
* Also change the `MESSAGE_GROUP_KEY_PREFIX` for group records to `GROUP_OF_MESSAGES_`
since the `MESSAGE_` prefix includes group records as well for various operations
based on key pattern

* * Use `[^GROUP_]` in the pattern for messages count

**Cherry-pick to `6.1.x`**

* * Fix MongoDB MS for message removal logic

* * Bring back `GROUP_OF_MESSAGES_` prefix
to avoid complex regexp and don't bother for edge cases,
where even that regexp may fail

spring-integration-core/src/main/java/org/springframework/integration/store/AbstractKeyValueMessageStore.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2002-2021 the original author or authors.
+ * Copyright 2002-2023 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -27,6 +27,7 @@ import java.util.stream.Collectors;
 import java.util.stream.Stream;
 
 import org.springframework.jmx.export.annotation.ManagedAttribute;
+import org.springframework.lang.Nullable;
 import org.springframework.messaging.Message;
 import org.springframework.util.Assert;
 
@@ -45,7 +46,7 @@ public abstract class AbstractKeyValueMessageStore extends AbstractMessageGroupS
 
 	protected static final String MESSAGE_KEY_PREFIX = "MESSAGE_";
 
-	protected static final String MESSAGE_GROUP_KEY_PREFIX = "MESSAGE_GROUP_";
+	protected static final String MESSAGE_GROUP_KEY_PREFIX = "GROUP_OF_MESSAGES_";
 
 	private final String messagePrefix;
 
@@ -57,9 +58,9 @@ public abstract class AbstractKeyValueMessageStore extends AbstractMessageGroupS
 
 	/**
 	 * Construct an instance based on the provided prefix for keys to distinguish between
-	 * different store instances in the same target key-value data base. Defaults to an
+	 * different store instances in the same target key-value database. Defaults to an
 	 * empty string - no prefix. The actual prefix for messages is
-	 * {@code prefix + MESSAGE_}; for message groups - {@code prefix + MESSAGE_GROUP_}
+	 * {@code prefix + MESSAGE_}; for message groups - {@code prefix + GROUP_OF_MESSAGES_}
 	 * @param prefix the prefix to use
 	 * @since 4.3.12
 	 */
@@ -71,18 +72,18 @@ public abstract class AbstractKeyValueMessageStore extends AbstractMessageGroupS
 
 	/**
 	 * Return the configured prefix for message keys to distinguish between different
-	 * store instances in the same target key-value data base. Defaults to the
+	 * store instances in the same target key-value database. Defaults to the
 	 * {@value MESSAGE_KEY_PREFIX} - without a custom prefix.
 	 * @return the prefix for keys
 	 * @since 4.3.12
 	 */
-	protected String getMessagePrefix() {
+	public String getMessagePrefix() {
 		return this.messagePrefix;
 	}
 
 	/**
 	 * Return the configured prefix for message group keys to distinguish between
-	 * different store instances in the same target key-value data base. Defaults to the
+	 * different store instances in the same target key-value database. Defaults to the
 	 * {@value MESSAGE_GROUP_KEY_PREFIX} - without custom prefix.
 	 * @return the prefix for keys
 	 * @since 4.3.12
@@ -140,10 +141,15 @@ public abstract class AbstractKeyValueMessageStore extends AbstractMessageGroupS
 	}
 
 	protected void doAddMessage(Message<?> message) {
+		doAddMessage(message, null);
+	}
+
+	protected void doAddMessage(Message<?> message, @Nullable Object groupId) {
 		Assert.notNull(message, "'message' must not be null");
 		UUID messageId = message.getHeaders().getId();
 		Assert.notNull(messageId, "Cannot store messages without an ID header");
-		doStoreIfAbsent(this.messagePrefix + messageId, new MessageHolder(message));
+		String messageKey = this.messagePrefix + (groupId != null ? groupId.toString() + '_' : "") + messageId;
+		doStoreIfAbsent(messageKey, new MessageHolder(message));
 	}
 
 	@Override
@@ -165,7 +171,6 @@ public abstract class AbstractKeyValueMessageStore extends AbstractMessageGroupS
 		return (messageIds != null) ? messageIds.size() : 0;
 	}
 
-
 	// MessageGroupStore methods
 
 	/**
@@ -211,7 +216,7 @@ public abstract class AbstractKeyValueMessageStore extends AbstractMessageGroupS
 		}
 
 		for (Message<?> message : messages) {
-			doAddMessage(message);
+			doAddMessage(message, groupId);
 			if (metadata != null) {
 				metadata.add(message.getHeaders().getId());
 			}
@@ -253,7 +258,7 @@ public abstract class AbstractKeyValueMessageStore extends AbstractMessageGroupS
 
 			List<Object> messageIds = new ArrayList<>();
 			for (UUID id : ids) {
-				messageIds.add(this.messagePrefix + id);
+				messageIds.add(this.messagePrefix + groupId + '_' + id);
 			}
 
 			doRemoveAll(messageIds);
@@ -288,7 +293,7 @@ public abstract class AbstractKeyValueMessageStore extends AbstractMessageGroupS
 			List<Object> messageIds =
 					messageGroupMetadata.getMessageIds()
 							.stream()
-							.map(id -> this.messagePrefix + id)
+							.map(id -> this.messagePrefix + groupId + '_' + id)
 							.collect(Collectors.toList());
 
 			doRemoveAll(messageIds);
@@ -326,24 +331,47 @@ public abstract class AbstractKeyValueMessageStore extends AbstractMessageGroupS
 				groupMetadata.remove(firstId);
 				groupMetadata.setLastModified(System.currentTimeMillis());
 				doStore(this.groupPrefix + groupId, groupMetadata);
-				return removeMessage(firstId);
+				return removeMessageFromGroup(firstId, groupId);
 			}
 		}
 		return null;
 	}
 
+	private Message<?> removeMessageFromGroup(UUID id, Object groupId) {
+		Assert.notNull(id, "'id' must not be null");
+		Object object = doRemove(this.messagePrefix + groupId + '_' + id);
+		if (object != null) {
+			return extractMessage(object);
+		}
+		else {
+			return null;
+		}
+	}
+
 	@Override
 	public Message<?> getOneMessageFromGroup(Object groupId) {
 		MessageGroupMetadata groupMetadata = getGroupMetadata(groupId);
 		if (groupMetadata != null) {
 			UUID messageId = groupMetadata.firstId();
 			if (messageId != null) {
-				return getMessage(messageId);
+				return getMessageFromGroup(messageId, groupId);
 			}
 		}
 		return null;
 	}
 
+	@Nullable
+	private Message<?> getMessageFromGroup(UUID messageId, Object groupId) {
+		Assert.notNull(messageId, "'messageId' must not be null");
+		Object object = doRetrieve(this.messagePrefix + groupId + '_' + messageId);
+		if (object != null) {
+			return extractMessage(object);
+		}
+		else {
+			return null;
+		}
+	}
+
 	@Override
 	public Collection<Message<?>> getMessagesForGroup(Object groupId) {
 		MessageGroupMetadata groupMetadata = getGroupMetadata(groupId);
@@ -351,7 +379,7 @@ public abstract class AbstractKeyValueMessageStore extends AbstractMessageGroupS
 		if (groupMetadata != null) {
 			Iterator<UUID> messageIds = groupMetadata.messageIdIterator();
 			while (messageIds.hasNext()) {
-				messages.add(getMessage(messageIds.next()));
+				messages.add(getMessageFromGroup(messageIds.next(), groupId));
 			}
 		}
 		return messages;
@@ -362,7 +390,7 @@ public abstract class AbstractKeyValueMessageStore extends AbstractMessageGroupS
 		return getGroupMetadata(groupId)
 				.getMessageIds()
 				.stream()
-				.map(this::getMessage);
+				.map((messageId) -> getMessageFromGroup(messageId, groupId));
 	}
 
 	@Override
@@ -376,8 +404,8 @@ public abstract class AbstractKeyValueMessageStore extends AbstractMessageGroupS
 
 	private Collection<String> normalizeKeys(Collection<String> keys) {
 		Set<String> normalizedKeys = new HashSet<>();
-		for (Object key : keys) {
-			String strKey = (String) key;
+		for (String key : keys) {
+			String strKey = key;
 			if (strKey.startsWith(this.groupPrefix)) {
 				strKey = strKey.replace(this.groupPrefix, "");
 			}

spring-integration-hazelcast/src/main/java/org/springframework/integration/hazelcast/store/HazelcastMessageStore.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2017-2022 the original author or authors.
+ * Copyright 2017-2023 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.

spring-integration-hazelcast/src/test/java/org/springframework/integration/hazelcast/store/HazelcastMessageStoreTests.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2017-2022 the original author or authors.
+ * Copyright 2017-2023 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -23,10 +23,10 @@ import java.util.Properties;
 import com.hazelcast.core.Hazelcast;
 import com.hazelcast.core.HazelcastInstance;
 import com.hazelcast.map.IMap;
-import org.junit.AfterClass;
-import org.junit.Before;
-import org.junit.BeforeClass;
-import org.junit.Test;
+import org.junit.jupiter.api.AfterAll;
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
 
 import org.springframework.integration.channel.DirectChannel;
 import org.springframework.integration.history.MessageHistory;
@@ -50,26 +50,25 @@ public class HazelcastMessageStoreTests {
 
 	private static IMap<Object, Object> map;
 
-	@BeforeClass
+	@BeforeAll
 	public static void init() {
 		instance = Hazelcast.newHazelcastInstance();
 		map = instance.getMap("customTestsMessageStore");
 		store = new HazelcastMessageStore(map);
 	}
 
-	@AfterClass
+	@AfterAll
 	public static void destroy() {
 		instance.shutdown();
 	}
 
-	@Before
+	@BeforeEach
 	public void clean() {
 		map.clear();
 	}
 
 	@Test
 	public void testWithMessageHistory() {
-
 		Message<?> message = new GenericMessage<>("Hello");
 		DirectChannel fooChannel = new DirectChannel();
 		fooChannel.setBeanName("fooChannel");
@@ -107,7 +106,6 @@ public class HazelcastMessageStoreTests {
 
 	@Test
 	public void addAndGetMessage() {
-
 		Message<?> message = MessageBuilder.withPayload("test").build();
 		store.addMessage(message);
 		Message<?> retrieved = store.getMessage(message.getHeaders().getId());
@@ -145,4 +143,34 @@ public class HazelcastMessageStoreTests {
 		assertThat(groupCount).isEqualTo(1);
 	}
 
+	@Test
+	public void sameMessageInTwoGroupsNotRemovedByFirstGroup() {
+		GenericMessage<String> testMessage = new GenericMessage<>("test data");
+
+		store.addMessageToGroup("1", testMessage);
+		store.addMessageToGroup("2", testMessage);
+
+		store.removeMessageGroup("1");
+
+		assertThat(store.getMessageCount()).isEqualTo(1);
+
+		store.removeMessageGroup("2");
+
+		assertThat(store.getMessageCount()).isEqualTo(0);
+	}
+
+	@Test
+	public void removeMessagesFromGroupDontRemoveSameMessageInOtherGroup() {
+		GenericMessage<String> testMessage = new GenericMessage<>("test data");
+
+		store.addMessageToGroup("1", testMessage);
+		store.addMessageToGroup("2", testMessage);
+
+		store.removeMessagesFromGroup("1", testMessage);
+
+		assertThat(store.getMessageCount()).isEqualTo(1);
+		assertThat(store.messageGroupSize("1")).isEqualTo(0);
+		assertThat(store.messageGroupSize("2")).isEqualTo(1);
+	}
+
 }

spring-integration-mongodb/src/main/java/org/springframework/integration/mongodb/store/ConfigurableMongoDbMessageStore.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2013-2022 the original author or authors.
+ * Copyright 2013-2023 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -101,7 +101,8 @@ public class ConfigurableMongoDbMessageStore extends AbstractConfigurableMongoDb
 	@Override
 	public Message<?> removeMessage(UUID id) {
 		Assert.notNull(id, "'id' must not be null");
-		Query query = Query.query(Criteria.where(MessageDocumentFields.MESSAGE_ID).is(id));
+		Query query = Query.query(Criteria.where(MessageDocumentFields.MESSAGE_ID).is(id)
+				.and(MessageDocumentFields.GROUP_ID).exists(false));
 		MessageDocument document = getMongoTemplate().findAndRemove(query, MessageDocument.class, this.collectionName);
 		return (document != null) ? document.getMessage() : null;
 	}

spring-integration-mongodb/src/main/java/org/springframework/integration/mongodb/store/MongoDbMessageStore.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2002-2022 the original author or authors.
+ * Copyright 2002-2023 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -258,14 +258,15 @@ public class MongoDbMessageStore extends AbstractMessageGroupStore
 	@Override
 	@ManagedAttribute
 	public long getMessageCount() {
-		return this.template.getCollection(this.collectionName).countDocuments();
+		Query query = Query.query(Criteria.where("headers.id").exists(true).and(GROUP_ID_KEY).exists(false));
+		return this.template.getCollection(this.collectionName).countDocuments(query.getQueryObject());
 	}
 
 	@Override
 	public Message<?> removeMessage(UUID id) {
 		Assert.notNull(id, "'id' must not be null");
-		MessageWrapper messageWrapper =
-				this.template.findAndRemove(whereMessageIdIs(id), MessageWrapper.class, this.collectionName);
+		Query query = Query.query(Criteria.where("headers.id").is(id).and(GROUP_ID_KEY).exists(false));
+		MessageWrapper messageWrapper = this.template.findAndRemove(query, MessageWrapper.class, this.collectionName);
 		return (messageWrapper != null ? messageWrapper.getMessage() : null);
 	}
 

spring-integration-mongodb/src/test/java/org/springframework/integration/mongodb/store/AbstractMongoDbMessageGroupStoreTests.java

@@ -536,6 +536,28 @@ public abstract class AbstractMongoDbMessageGroupStoreTests implements MongoDbCo
 				.containsEntry("type", "channel");
 	}
 
+	@Test
+	public void removeMessageDoesntRemoveSameMessageInTheGroup() {
+		GenericMessage<String> testMessage = new GenericMessage<>("test data");
+
+		MessageGroupStore store = getMessageGroupStore();
+
+		store.addMessageToGroup("1", testMessage);
+
+		MessageStore messageStore = (MessageStore) store;
+
+		messageStore.removeMessage(testMessage.getHeaders().getId());
+
+		assertThat(messageStore.getMessageCount()).isEqualTo(0);
+		assertThat(store.getMessageCountForAllMessageGroups()).isEqualTo(1);
+		assertThat(store.messageGroupSize("1")).isEqualTo(1);
+
+		store.removeMessageGroup("1");
+
+		assertThat(store.getMessageCountForAllMessageGroups()).isEqualTo(0);
+		assertThat(store.messageGroupSize("1")).isEqualTo(0);
+	}
+
 	protected abstract MessageGroupStore getMessageGroupStore();
 
 	protected abstract MessageStore getMessageStore();

spring-integration-redis/src/test/java/org/springframework/integration/redis/store/RedisMessageGroupStoreTests.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2007-2022 the original author or authors.
+ * Copyright 2007-2023 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -56,6 +56,7 @@ import org.springframework.messaging.support.ErrorMessage;
 import org.springframework.messaging.support.GenericMessage;
 
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.assertj.core.api.Assertions.assertThatNoException;
 import static org.assertj.core.api.Assertions.fail;
 
 /**
@@ -65,6 +66,7 @@ import static org.assertj.core.api.Assertions.fail;
  * @author Artem Vozhdayenko
  */
 class RedisMessageGroupStoreTests implements RedisContainerTest {
+
 	private static RedisConnectionFactory redisConnectionFactory;
 
 	@BeforeAll
@@ -74,17 +76,18 @@ class RedisMessageGroupStoreTests implements RedisContainerTest {
 
 	private final UUID groupId = UUID.randomUUID();
 
+	RedisMessageStore store = new RedisMessageStore(redisConnectionFactory);
+
 	@BeforeEach
 	@AfterEach
 	void setUpTearDown() {
 		StringRedisTemplate template = RedisContainerTest.createStringRedisTemplate(redisConnectionFactory);
-		template.delete(template.keys("MESSAGE_GROUP_*"));
+		template.delete(template.keys("MESSAGE_*"));
+		template.delete(template.keys("GROUP_OF_MESSAGES_*"));
 	}
 
 	@Test
 	void testNonExistingEmptyMessageGroup() {
-		RedisMessageStore store = new RedisMessageStore(redisConnectionFactory);
-
 		MessageGroup messageGroup = store.getMessageGroup(this.groupId);
 		assertThat(messageGroup).isNotNull();
 		assertThat(messageGroup).isInstanceOf(SimpleMessageGroup.class);
@@ -93,8 +96,6 @@ class RedisMessageGroupStoreTests implements RedisContainerTest {
 
 	@Test
 	void testMessageGroupUpdatedDateChangesWithEachAddedMessage() throws Exception {
-		RedisMessageStore store = new RedisMessageStore(redisConnectionFactory);
-
 		Message<?> message = new GenericMessage<>("Hello");
 		MessageGroup messageGroup = store.addMessageToGroup(this.groupId, message);
 		assertThat(messageGroup.size()).isEqualTo(1);
@@ -117,8 +118,6 @@ class RedisMessageGroupStoreTests implements RedisContainerTest {
 
 	@Test
 	void testMessageGroupWithAddedMessage() {
-		RedisMessageStore store = new RedisMessageStore(redisConnectionFactory);
-
 		Message<?> message = new GenericMessage<>("Hello");
 		MessageGroup messageGroup = store.addMessageToGroup(this.groupId, message);
 		assertThat(messageGroup.size()).isEqualTo(1);
@@ -132,8 +131,6 @@ class RedisMessageGroupStoreTests implements RedisContainerTest {
 
 	@Test
 	void testRemoveMessageGroup() {
-		RedisMessageStore store = new RedisMessageStore(redisConnectionFactory);
-
 		MessageGroup messageGroup = store.getMessageGroup(this.groupId);
 		Message<?> message = new GenericMessage<>("Hello");
 		messageGroup = store.addMessageToGroup(messageGroup.getGroupId(), message);
@@ -157,8 +154,6 @@ class RedisMessageGroupStoreTests implements RedisContainerTest {
 
 	@Test
 	void testCompleteMessageGroup() {
-		RedisMessageStore store = new RedisMessageStore(redisConnectionFactory);
-
 		MessageGroup messageGroup = store.getMessageGroup(this.groupId);
 		Message<?> message = new GenericMessage<>("Hello");
 		messageGroup = store.addMessageToGroup(messageGroup.getGroupId(), message);
@@ -169,8 +164,6 @@ class RedisMessageGroupStoreTests implements RedisContainerTest {
 
 	@Test
 	void testLastReleasedSequenceNumber() {
-		RedisMessageStore store = new RedisMessageStore(redisConnectionFactory);
-
 		MessageGroup messageGroup = store.getMessageGroup(this.groupId);
 		Message<?> message = new GenericMessage<>("Hello");
 		messageGroup = store.addMessageToGroup(messageGroup.getGroupId(), message);
@@ -181,8 +174,6 @@ class RedisMessageGroupStoreTests implements RedisContainerTest {
 
 	@Test
 	void testRemoveMessageFromTheGroup() {
-		RedisMessageStore store = new RedisMessageStore(redisConnectionFactory);
-
 		MessageGroup messageGroup = store.getMessageGroup(this.groupId);
 		Message<?> message = new GenericMessage<>("2");
 		store.addMessagesToGroup(messageGroup.getGroupId(), new GenericMessage<>("1"), message);
@@ -202,8 +193,6 @@ class RedisMessageGroupStoreTests implements RedisContainerTest {
 
 	@Test
 	void testWithMessageHistory() {
-		RedisMessageStore store = new RedisMessageStore(redisConnectionFactory);
-
 		Message<?> message = new GenericMessage<>("Hello");
 		DirectChannel fooChannel = new DirectChannel();
 		fooChannel.setBeanName("fooChannel");
@@ -227,17 +216,16 @@ class RedisMessageGroupStoreTests implements RedisContainerTest {
 
 	@Test
 	void testRemoveNonExistingMessageFromTheGroup() {
-		RedisMessageStore store = new RedisMessageStore(redisConnectionFactory);
-
 		MessageGroup messageGroup = store.getMessageGroup(this.groupId);
 		store.addMessagesToGroup(messageGroup.getGroupId(), new GenericMessage<>("1"));
-		store.removeMessagesFromGroup(this.groupId, new GenericMessage<>("2"));
+		assertThatNoException()
+				.isThrownBy(() -> store.removeMessagesFromGroup(this.groupId, new GenericMessage<>("2")));
 	}
 
 	@Test
 	void testRemoveNonExistingMessageFromNonExistingTheGroup() {
-		RedisMessageStore store = new RedisMessageStore(redisConnectionFactory);
-		store.removeMessagesFromGroup(this.groupId, new GenericMessage<>("2"));
+		assertThatNoException()
+				.isThrownBy(() -> store.removeMessagesFromGroup(this.groupId, new GenericMessage<>("2")));
 	}
 
 
@@ -283,14 +271,9 @@ class RedisMessageGroupStoreTests implements RedisContainerTest {
 		while (messageGroups.hasNext()) {
 			MessageGroup group = messageGroups.next();
 			String groupId = (String) group.getGroupId();
-			if (groupId.equals("1")) {
-				assertThat(group.getMessages().size()).isEqualTo(1);
-			}
-			else if (groupId.equals("2")) {
-				assertThat(group.getMessages().size()).isEqualTo(1);
-			}
-			else if (groupId.equals("3")) {
-				assertThat(group.getMessages().size()).isEqualTo(2);
+			switch (groupId) {
+				case "1", "2" -> assertThat(group.getMessages().size()).isEqualTo(1);
+				case "3" -> assertThat(group.getMessages().size()).isEqualTo(2);
 			}
 			counter++;
 		}
@@ -390,25 +373,22 @@ class RedisMessageGroupStoreTests implements RedisContainerTest {
 
 	@Test
 	void testAddAndRemoveMessagesFromMessageGroup() {
-		RedisMessageStore messageStore = new RedisMessageStore(redisConnectionFactory);
-		List<Message<?>> messages = new ArrayList<Message<?>>();
+		List<Message<?>> messages = new ArrayList<>();
 		for (int i = 0; i < 25; i++) {
 			Message<String> message = MessageBuilder.withPayload("foo").setCorrelationId(this.groupId).build();
-			messageStore.addMessagesToGroup(this.groupId, message);
+			store.addMessagesToGroup(this.groupId, message);
 			messages.add(message);
 		}
-		MessageGroup group = messageStore.getMessageGroup(this.groupId);
+		MessageGroup group = store.getMessageGroup(this.groupId);
 		assertThat(group.size()).isEqualTo(25);
-		messageStore.removeMessagesFromGroup(this.groupId, messages);
-		group = messageStore.getMessageGroup(this.groupId);
+		store.removeMessagesFromGroup(this.groupId, messages);
+		group = store.getMessageGroup(this.groupId);
 		assertThat(group.size()).isZero();
-		messageStore.removeMessageGroup(this.groupId);
+		store.removeMessageGroup(this.groupId);
 	}
 
 	@Test
 	void testJsonSerialization() {
-		RedisMessageStore store = new RedisMessageStore(redisConnectionFactory);
-
 		ObjectMapper mapper = JacksonJsonUtils.messagingAwareMapper();
 
 		GenericJackson2JsonRedisSerializer serializer = new GenericJackson2JsonRedisSerializer(mapper);
@@ -476,6 +456,36 @@ class RedisMessageGroupStoreTests implements RedisContainerTest {
 		assertThat(messageGroup.getMessages().iterator().next()).isEqualTo(fooMessage);
 	}
 
+	@Test
+	public void sameMessageInTwoGroupsNotRemovedByFirstGroup() {
+		GenericMessage<String> testMessage = new GenericMessage<>("test data");
+
+		store.addMessageToGroup("1", testMessage);
+		store.addMessageToGroup("2", testMessage);
+
+		store.removeMessageGroup("1");
+
+		assertThat(store.getMessageCount()).isEqualTo(1);
+
+		store.removeMessageGroup("2");
+
+		assertThat(store.getMessageCount()).isEqualTo(0);
+	}
+
+	@Test
+	public void removeMessagesFromGroupDontRemoveSameMessageInOtherGroup() {
+		GenericMessage<String> testMessage = new GenericMessage<>("test data");
+
+		store.addMessageToGroup("1", testMessage);
+		store.addMessageToGroup("2", testMessage);
+
+		store.removeMessagesFromGroup("1", testMessage);
+
+		assertThat(store.getMessageCount()).isEqualTo(1);
+		assertThat(store.messageGroupSize("1")).isEqualTo(0);
+		assertThat(store.messageGroupSize("2")).isEqualTo(1);
+	}
+
 	private static class Foo {
 
 		private String foo;