Introduce H2ChatMemoryRepositoryDialect

It seems that H2 is more populate than HSQLDB.

Signed-off-by: Yanming Zhou <[email protected]>

Author: quaff

memory/repository/spring-ai-model-chat-memory-repository-jdbc/pom.xml

@@ -77,6 +77,14 @@
 		</dependency>
 
 		<!-- TESTING -->
+
+		<dependency>
+			<groupId>com.h2database</groupId>
+			<artifactId>h2</artifactId>
+			<scope>test</scope>
+			<optional>true</optional>
+		</dependency>
+
 		<dependency>
 			<groupId>org.springframework.boot</groupId>
 			<artifactId>spring-boot-starter-test</artifactId>

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

@@ -0,0 +1,29 @@
+/*
+ * Copyright 2024-2025 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.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.ai.chat.memory.repository.jdbc;
+
+/**
+ * H2-specific SQL dialect for chat memory repository.
+ * <p>
+ * Currently it's compatible with HSQLDB, we could override methods for H2-specific syntax
+ * in the future.
+ *
+ * @author Yanming Zhou
+ */
+public class H2ChatMemoryRepositoryDialect extends HsqldbChatMemoryRepositoryDialect {
+
+}

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

@@ -57,14 +57,14 @@ static JdbcChatMemoryRepositoryDialect from(DataSource dataSource) {
 			String url = dataSource.getConnection().getMetaData().getURL().toLowerCase();
 			if (url.contains("postgresql"))
 				return new PostgresChatMemoryRepositoryDialect();
-			if (url.contains("mysql"))
-				return new MysqlChatMemoryRepositoryDialect();
-			if (url.contains("mariadb"))
+			if (url.contains("mysql") || url.contains("mariadb"))
 				return new MysqlChatMemoryRepositoryDialect();
 			if (url.contains("sqlserver"))
 				return new SqlServerChatMemoryRepositoryDialect();
 			if (url.contains("hsqldb"))
 				return new HsqldbChatMemoryRepositoryDialect();
+			if (url.contains("h2"))
+				return new H2ChatMemoryRepositoryDialect();
 			// Add more as needed
 		}
 		catch (Exception ignored) {

memory/repository/spring-ai-model-chat-memory-repository-jdbc/src/main/resources/org/springframework/ai/chat/memory/repository/jdbc/schema-h2.sql

@@ -0,0 +1,10 @@
+CREATE TABLE SPRING_AI_CHAT_MEMORY (
+    conversation_id VARCHAR(36) NOT NULL,
+    content LONGVARCHAR NOT NULL,
+    type VARCHAR(10) NOT NULL,
+    timestamp TIMESTAMP DEFAULT CURRENT_TIMESTAMP NOT NULL
+);
+
+CREATE INDEX SPRING_AI_CHAT_MEMORY_CONVERSATION_ID_TIMESTAMP_IDX ON SPRING_AI_CHAT_MEMORY(conversation_id, timestamp DESC);
+
+ALTER TABLE SPRING_AI_CHAT_MEMORY ADD CONSTRAINT TYPE_CHECK CHECK (type IN ('USER', 'ASSISTANT', 'SYSTEM', 'TOOL'));

memory/repository/spring-ai-model-chat-memory-repository-jdbc/src/test/java/org/springframework/ai/chat/memory/repository/jdbc/JdbcChatMemoryRepositoryH2IT.java

@@ -0,0 +1,40 @@
+/*
+ * Copyright 2023-2025 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.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.ai.chat.memory.repository.jdbc;
+
+import org.springframework.boot.SpringBootConfiguration;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.TestPropertySource;
+import org.springframework.test.context.jdbc.Sql;
+
+/**
+ * Integration tests for {@link JdbcChatMemoryRepository} with H2.
+ *
+ * @author Yanming Zhou
+ */
+@SpringBootTest(classes = JdbcChatMemoryRepositoryH2IT.TestConfiguration.class)
+@TestPropertySource(properties = { "spring.datasource.url=jdbc:h2:mem:mydb" })
+@Sql(scripts = "classpath:org/springframework/ai/chat/memory/repository/jdbc/schema-h2.sql",
+		executionPhase = Sql.ExecutionPhase.BEFORE_TEST_CLASS)
+class JdbcChatMemoryRepositoryH2IT extends AbstractJdbcChatMemoryRepositoryIT {
+
+	@SpringBootConfiguration
+	static class TestConfiguration extends BaseTestConfiguration {
+
+	}
+
+}