Adapt Oracle JDBC ChatMemoryRepository support

  - Add Oracle Dialect and Schema
  - Update docs and tests

Signed-off-by: fanxt0218 <[email protected]>
Co-authored-by: psilberk <[email protected]>

Author: 2 people

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

@@ -56,7 +56,6 @@
 		</dependency>
 
 		<!-- TESTING -->
-
 		<dependency>
 			<groupId>org.postgresql</groupId>
 			<artifactId>postgresql</artifactId>
@@ -99,6 +98,20 @@
 			<optional>true</optional>
 		</dependency>
 
+		<dependency>
+			<groupId>com.oracle.database.jdbc</groupId>
+			<artifactId>ojdbc11</artifactId>
+			<version>23.4.0.24.05</version>
+			<scope>test</scope>
+			<optional>true</optional>
+		</dependency>
+
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-starter-jdbc</artifactId>
+			<scope>test</scope>
+		</dependency>
+
 		<dependency>
 			<groupId>org.springframework.boot</groupId>
 			<artifactId>spring-boot-starter-test</artifactId>
@@ -111,6 +124,13 @@
 			<scope>test</scope>
 		</dependency>
 
+		<dependency>
+			<groupId>org.testcontainers</groupId>
+			<artifactId>testcontainers-oracle-free</artifactId>
+			<version>2.0.1</version>
+			<scope>test</scope>
+		</dependency>
+
 		<dependency>
 			<groupId>org.testcontainers</groupId>
 			<artifactId>postgresql</artifactId>

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

@@ -76,6 +76,7 @@ static JdbcChatMemoryRepositoryDialect from(DataSource dataSource) {
 			case "HSQL Database Engine" -> new HsqldbChatMemoryRepositoryDialect();
 			case "SQLite" -> new SqliteChatMemoryRepositoryDialect();
 			case "H2" -> new H2ChatMemoryRepositoryDialect();
+			case "Oracle" -> new OracleChatMemoryRepositoryDialect();
 			default -> // Add more as needed
 				new PostgresChatMemoryRepositoryDialect();
 		};

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

@@ -0,0 +1,49 @@
+/*
+ * Copyright 2025-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;
+
+/**
+ * Dialect for Oracle.
+ *
+ * @author Xiaotong Fan
+ * @author Pablo Silberkasten
+ * @since 1.1.0
+ */
+
+public class OracleChatMemoryRepositoryDialect implements JdbcChatMemoryRepositoryDialect {
+
+	@Override
+	public String getSelectMessagesSql() {
+		return "SELECT content, type FROM SPRING_AI_CHAT_MEMORY WHERE CONVERSATION_ID = ? ORDER BY \"TIMESTAMP\"";
+	}
+
+	@Override
+	public String getInsertMessageSql() {
+		return "INSERT INTO SPRING_AI_CHAT_MEMORY (CONVERSATION_ID, CONTENT, TYPE, \"TIMESTAMP\") VALUES (?, ?, ?, ?)";
+	}
+
+	@Override
+	public String getSelectConversationIdsSql() {
+		return "SELECT DISTINCT conversation_id FROM SPRING_AI_CHAT_MEMORY";
+	}
+
+	@Override
+	public String getDeleteMessagesSql() {
+		return "DELETE FROM SPRING_AI_CHAT_MEMORY WHERE CONVERSATION_ID = ?";
+	}
+
+}

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

@@ -0,0 +1,21 @@
+DROP TABLE IF EXISTS SPRING_AI_CHAT_MEMORY;
+
+CREATE TABLE SPRING_AI_CHAT_MEMORY (
+                                       CONVERSATION_ID VARCHAR2(36 CHAR) NOT NULL,
+                                       CONTENT CLOB NOT NULL,
+                                       "TYPE" VARCHAR2(10 CHAR) NOT NULL CHECK (
+        "TYPE" IN (
+            'USER',
+            'ASSISTANT',
+            'SYSTEM',
+            'TOOL'
+        )
+    ),
+                                       "TIMESTAMP" TIMESTAMP NOT NULL
+);
+
+CREATE INDEX SPRING_AI_CHAT_MEMORY_CONVERSATION_ID_TIMESTAMP_IDX
+    ON SPRING_AI_CHAT_MEMORY(
+                             CONVERSATION_ID,
+        'TIMESTAMP'
+        );

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

@@ -37,6 +37,7 @@
  *
  * @author Mark Pollack
  * @author Yanming Zhou
+ * @author Xiaotong Fan
  */
 public class JdbcChatMemoryRepositoryBuilderTests {
 
@@ -136,6 +137,23 @@ void testBuilderWithHsqldbDialectFromDataSource() throws SQLException {
 		assertThat(repository).isNotNull();
 	}
 
+	@Test
+	void testBuilderWithOracleDialectFromDataSource() throws SQLException {
+		// Setup mocks for Oracle
+		DataSource dataSource = mock(DataSource.class);
+		Connection connection = mock(Connection.class);
+		DatabaseMetaData metaData = mock(DatabaseMetaData.class);
+
+		when(dataSource.getConnection()).thenReturn(connection);
+		when(connection.getMetaData()).thenReturn(metaData);
+		when(metaData.getURL()).thenReturn("jdbc:oracle:thin:@//192.168.19.129:1521/ORCL");
+
+		// Test with dialect from datasource
+		JdbcChatMemoryRepository repository = JdbcChatMemoryRepository.builder().dataSource(dataSource).build();
+
+		assertThat(repository).isNotNull();
+	}
+
 	@Test
 	void testBuilderWithUnknownDialectFromDataSource() throws SQLException {
 		// Setup mocks for unknown database

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

@@ -0,0 +1,35 @@
+/*
+ * 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.test.context.SpringBootTest;
+import org.springframework.test.context.TestPropertySource;
+import org.springframework.test.context.jdbc.Sql;
+
+/**
+ * Integration tests for {@link JdbcChatMemoryRepository} with Oracle.
+ *
+ * @author Xiaotong Fan
+ */
+
+@SpringBootTest
+@TestPropertySource(properties = { "spring.datasource.url=jdbc:tc:oracle:slim-faststart:///FREEPDB1",
+		"spring.datasource.username=test", "spring.datasource.password=test" })
+@Sql(scripts = "classpath:org/springframework/ai/chat/memory/repository/jdbc/schema-oracle.sql")
+public class JdbcChatMemoryRepositoryOracleIT extends AbstractJdbcChatMemoryRepositoryIT {
+
+}

spring-ai-docs/src/main/antora/modules/ROOT/pages/api/chat-memory.adoc

@@ -128,6 +128,7 @@ Spring AI supports multiple relational databases via a dialect abstraction. The
 - MySQL / MariaDB
 - SQL Server
 - HSQLDB
+- Oracle Database
 
 The correct dialect can be auto-detected from the JDBC URL when using `JdbcChatMemoryRepositoryDialect.from(DataSource)`. You can extend support for other databases by implementing the `JdbcChatMemoryRepositoryDialect` interface.