Package com.bytedesk.ai.alibaba.memory

Class JdbcChatMemoryRepository

java.lang.Object

com.bytedesk.ai.alibaba.memory.JdbcChatMemoryRepository

All Implemented Interfaces:

org.springframework.ai.chat.memory.ChatMemoryRepository

Direct Known Subclasses:

MysqlChatMemoryRepository

public abstract class JdbcChatMemoryRepository
extends Object
implements org.springframework.ai.chat.memory.ChatMemoryRepository

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
private static final record	JdbcChatMemoryRepository.AddBatchPreparedStatement
private static class	JdbcChatMemoryRepository.MessageRowMapper

Field Summary

Fields

Modifier and Type	Field	Description
private final org.springframework.jdbc.core.JdbcTemplate	jdbcTemplate
private static final String	QUERY_ADD
private static final String	QUERY_CLEAR
private static final String	QUERY_GET
private static final String	QUERY_GET_IDS
static final String	TABLE_NAME

Constructor Summary

Constructors

Constructor	Description
JdbcChatMemoryRepository(org.springframework.jdbc.core.JdbcTemplate jdbcTemplate)

Method Summary

Modifier and Type	Method	Description
private void	checkAndCreateTable()
protected abstract String	createTableSql(String tableName)
void	deleteByConversationId(String conversationId)
List<org.springframework.ai.chat.messages.Message>	findByConversationId(String conversationId)
List<String>	findConversationIds()
protected String	getAddSql()	Get the SQL statement used to add records
protected String	getGetSql()	Get the SQL statement used for querying records
protected abstract String	hasTableSql(String tableName)
void	saveAll(String conversationId, List<org.springframework.ai.chat.messages.Message> messages)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

TABLE_NAME

public static final String TABLE_NAME

See Also:

• Constant Field Values

QUERY_GET_IDS

private static final String QUERY_GET_IDS

See Also:

• Constant Field Values

QUERY_ADD

private static final String QUERY_ADD

See Also:

• Constant Field Values

QUERY_GET

private static final String QUERY_GET

See Also:

• Constant Field Values

QUERY_CLEAR

private static final String QUERY_CLEAR

See Also:

• Constant Field Values

jdbcTemplate

private final org.springframework.jdbc.core.JdbcTemplate jdbcTemplate

Constructor Details

JdbcChatMemoryRepository

public JdbcChatMemoryRepository(org.springframework.jdbc.core.JdbcTemplate jdbcTemplate)

Method Details

checkAndCreateTable

private void checkAndCreateTable()

findConversationIds

public List<String> findConversationIds()

Specified by:

findConversationIds in interface org.springframework.ai.chat.memory.ChatMemoryRepository

findByConversationId

public List<org.springframework.ai.chat.messages.Message> findByConversationId(String conversationId)

Specified by:

findByConversationId in interface org.springframework.ai.chat.memory.ChatMemoryRepository

saveAll

public void saveAll(String conversationId,
 List<org.springframework.ai.chat.messages.Message> messages)

Specified by:

saveAll in interface org.springframework.ai.chat.memory.ChatMemoryRepository

deleteByConversationId

public void deleteByConversationId(String conversationId)

Specified by:

deleteByConversationId in interface org.springframework.ai.chat.memory.ChatMemoryRepository

getAddSql

protected String getAddSql()

Get the SQL statement used to add records

Returns:

SQL

getGetSql

protected String getGetSql()

Get the SQL statement used for querying records

Returns:

SQL

hasTableSql

protected abstract String hasTableSql(String tableName)

createTableSql

protected abstract String createTableSql(String tableName)