feat: implement refresh logic for each database type

Signed-off-by: potato <[email protected]>

Author: juntae6942

memory/repository/spring-ai-model-chat-memory-repository-cassandra/src/main/java/org/springframework/ai/chat/memory/repository/cassandra/CassandraChatMemoryRepository.java

@@ -165,6 +165,20 @@ public void deleteByConversationId(String conversationId) {
 		saveAll(conversationId, List.of());
 	}
 
+	@Override
+	public void refresh(String conversationId, List<Message> deletes, List<Message> adds) {
+		Assert.hasText(conversationId, "conversationId cannot be null or empty");
+		Assert.notNull(deletes, "deletes cannot be null");
+		Assert.notNull(adds, "adds cannot be null");
+
+		// RMW (Read-Modify-Write) is the only way with the current schema.
+		// This is not efficient, but it is correct.
+		List<Message> currentMessages = new ArrayList<>(this.findByConversationId(conversationId));
+		currentMessages.removeAll(deletes);
+		currentMessages.addAll(adds);
+		this.saveAll(conversationId, currentMessages);
+	}
+
 	private PreparedStatement prepareAddStmt() {
 		RegularInsert stmt = null;
 		InsertInto stmtStart = QueryBuilder.insertInto(this.conf.schema.keyspace(), this.conf.schema.table());

memory/repository/spring-ai-model-chat-memory-repository-jdbc/src/main/java/org/springframework/ai/chat/memory/repository/jdbc/JdbcChatMemoryRepository.java

@@ -106,10 +106,51 @@ public void deleteByConversationId(String conversationId) {
 		this.jdbcTemplate.update(this.dialect.getDeleteMessagesSql(), conversationId);
 	}
 
+	@Override
+	public void refresh(String conversationId, List<Message> deletes, List<Message> adds) {
+		Assert.hasText(conversationId, "conversationId cannot be null or empty");
+		Assert.notNull(deletes, "deletes cannot be null");
+		Assert.notNull(adds, "adds cannot be null");
+
+		this.transactionTemplate.execute(status -> {
+			if (!deletes.isEmpty()) {
+				// This is a simplification. In a real implementation, we would need a
+				// stable
+				// way to identify messages to delete, perhaps by adding a message_id
+				// column.
+				// For now, we delete based on content and type, which is not robust.
+				this.jdbcTemplate.batchUpdate(this.dialect.getDeleteMessageSql(),
+						new DeleteBatchPreparedStatement(conversationId, deletes));
+			}
+			if (!adds.isEmpty()) {
+				this.jdbcTemplate.batchUpdate(this.dialect.getInsertMessageSql(),
+						new AddBatchPreparedStatement(conversationId, adds));
+			}
+			return null;
+		});
+	}
+
 	public static Builder builder() {
 		return new Builder();
 	}
 
+	private record DeleteBatchPreparedStatement(String conversationId,
+			List<Message> messages) implements BatchPreparedStatementSetter {
+
+		@Override
+		public void setValues(PreparedStatement ps, int i) throws SQLException {
+			var message = this.messages.get(i);
+			ps.setString(1, this.conversationId);
+			ps.setString(2, message.getText());
+			ps.setString(3, message.getMessageType().name());
+		}
+
+		@Override
+		public int getBatchSize() {
+			return this.messages.size();
+		}
+	}
+
 	private record AddBatchPreparedStatement(String conversationId, List<Message> messages,
 			AtomicLong instantSeq) implements BatchPreparedStatementSetter {
 

memory/repository/spring-ai-model-chat-memory-repository-neo4j/src/main/java/org/springframework/ai/chat/memory/repository/neo4j/Neo4jChatMemoryRepository.java

@@ -166,6 +166,34 @@ OPTIONAL MATCH (m)-[:HAS_TOOL_CALL]->(tc:%s)
 		}
 	}
 
+	@Override
+	public void refresh(String conversationId, List<Message> deletes, List<Message> adds) {
+		try (Session s = this.config.getDriver().session()) {
+			s.executeWriteWithoutResult(tx -> {
+				if (!deletes.isEmpty()) {
+					List<String> messageIds = deletes.stream().map(m -> (String) m.getMetadata().get("id")).toList();
+
+					String deleteStatement = """
+							MATCH (m:%s) WHERE m.id IN $messageIds
+							OPTIONAL MATCH (m)-[:HAS_METADATA]->(metadata:%s)
+							OPTIONAL MATCH (m)-[:HAS_MEDIA]->(media:%s)
+							OPTIONAL MATCH (m)-[:HAS_TOOL_RESPONSE]-(tr:%s)
+							OPTIONAL MATCH (m)-[:HAS_TOOL_CALL]->(tc:%s)
+							DETACH DELETE m, metadata, media, tr, tc
+							""".formatted(this.config.getMessageLabel(), this.config.getMetadataLabel(),
+							this.config.getMediaLabel(), this.config.getToolResponseLabel(),
+							this.config.getToolCallLabel());
+					tx.run(deleteStatement, Map.of("messageIds", messageIds));
+				}
+				if (!adds.isEmpty()) {
+					for (Message m : adds) {
+						addMessageToTransaction(tx, conversationId, m);
+					}
+				}
+			});
+		}
+	}
+
 	public Neo4jChatMemoryRepositoryConfig getConfig() {
 		return this.config;
 	}