repositories

package
v0.0.0-...-3d7ec5d Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Dec 18, 2025 License: Apache-2.0 Imports: 10 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func GetDB

func GetDB() *gorm.DB

GetDB 获取数据库实例

func InitDB

func InitDB(dsn string) error

InitDB 初始化数据库连接 dsn 格式: - MySQL: user:pass@tcp(host:port)/dbname?charset=utf8mb4&parseTime=True&loc=Local - SQLite: /path/to/file.db 或 /path/to/file.db?_journal_mode=WAL

Types

type AttachmentRepository

type AttachmentRepository struct {
	// contains filtered or unexported fields
}

AttachmentRepository 附件仓库

func NewAttachmentRepository

func NewAttachmentRepository(db *gorm.DB) *AttachmentRepository

NewAttachmentRepository 创建附件仓库实例

func (*AttachmentRepository) Create

func (r *AttachmentRepository) Create(attach *models.Attachment) error

Create 创建附件

func (*AttachmentRepository) Delete

func (r *AttachmentRepository) Delete(attachID string) error

Delete 删除附件

func (*AttachmentRepository) GetByID

func (r *AttachmentRepository) GetByID(attachID string) (*models.Attachment, error)

GetByID 根据ID获取附件

func (*AttachmentRepository) ListByMessage

func (r *AttachmentRepository) ListByMessage(messageID string) ([]*models.Attachment, error)

ListByMessage 获取消息的附件列表

func (*AttachmentRepository) Update

func (r *AttachmentRepository) Update(attach *models.Attachment) error

Update 更新附件

func (*AttachmentRepository) UpdateDataSummary

func (r *AttachmentRepository) UpdateDataSummary(attachID string, dataSummary string) error

UpdateDataSummary 更新附件数据摘要

func (*AttachmentRepository) UpdateVectorized

func (r *AttachmentRepository) UpdateVectorized(attachID string, vectorized bool) error

UpdateVectorized 更新附件向量化状态

type ConversationRepository

type ConversationRepository struct {
	// contains filtered or unexported fields
}

ConversationRepository 对话仓库

func NewConversationRepository

func NewConversationRepository(db *gorm.DB) *ConversationRepository

NewConversationRepository 创建对话仓库实例

func (*ConversationRepository) Archive

func (r *ConversationRepository) Archive(convID string) error

Archive 归档对话

func (*ConversationRepository) Create

Create 创建对话

func (*ConversationRepository) Delete

func (r *ConversationRepository) Delete(convID string) error

Delete 删除对话

func (*ConversationRepository) FirstOrCreat

func (r *ConversationRepository) FirstOrCreat(convID string) (*models.Conversation, error)

FirstOrCreat 根据ID判断,如果存在就返回,不存在就创建

func (*ConversationRepository) GetByID

func (r *ConversationRepository) GetByID(convID string) (*models.Conversation, error)

GetByID 根据ID获取对话

func (*ConversationRepository) List

func (r *ConversationRepository) List(offset, limit int) ([]*models.Conversation, error)

List 获取对话列表

func (*ConversationRepository) Pin

func (r *ConversationRepository) Pin(convID string) error

Pin 置顶对话

func (*ConversationRepository) Unarchive

func (r *ConversationRepository) Unarchive(convID string) error

Unarchive 取消归档对话

func (*ConversationRepository) Unpin

func (r *ConversationRepository) Unpin(convID string) error

Unpin 取消置顶对话

func (*ConversationRepository) Update

Update 更新对话

type MessageAttachmentRepository

type MessageAttachmentRepository struct {
	// contains filtered or unexported fields
}

MessageAttachmentRepository 消息附件关联仓库

func NewMessageAttachmentRepository

func NewMessageAttachmentRepository(db *gorm.DB) *MessageAttachmentRepository

NewMessageAttachmentRepository 创建消息附件关联仓库实例

func (*MessageAttachmentRepository) Create

Create 创建消息附件关联

func (*MessageAttachmentRepository) Delete

func (r *MessageAttachmentRepository) Delete(messageID, attachmentID string) error

Delete 删除消息附件关联

func (*MessageAttachmentRepository) ListByAttachment

func (r *MessageAttachmentRepository) ListByAttachment(attachmentID string) ([]*models.MessageAttachment, error)

ListByAttachment 获取附件的所有消息关联

func (*MessageAttachmentRepository) ListByMessage

func (r *MessageAttachmentRepository) ListByMessage(messageID string) ([]*models.MessageAttachment, error)

ListByMessage 获取消息的所有附件关联

type MessageRepository

type MessageRepository struct {
	// contains filtered or unexported fields
}

MessageRepository 消息仓库

func NewMessageRepository

func NewMessageRepository(db *gorm.DB) *MessageRepository

NewMessageRepository 创建消息仓库实例

func (*MessageRepository) Create

func (r *MessageRepository) Create(msg *models.Message) error

Create 创建消息

func (*MessageRepository) Delete

func (r *MessageRepository) Delete(msgID string) error

Delete 删除消息

func (*MessageRepository) GetByID

func (r *MessageRepository) GetByID(msgID string) (*models.Message, error)

GetByID 根据ID获取消息

func (*MessageRepository) ListByConversation

func (r *MessageRepository) ListByConversation(conversationID string, offset, limit int) ([]*models.Message, error)

ListByConversation 获取对话的消息列表

func (*MessageRepository) SetContextEdge

func (r *MessageRepository) SetContextEdge(msgID string, isContextEdge bool) error

SetContextEdge 设置消息为上下文边界

func (*MessageRepository) SetVariant

func (r *MessageRepository) SetVariant(msgID string, isVariant bool) error

SetVariant 设置消息为变体

func (*MessageRepository) Update

func (r *MessageRepository) Update(msg *models.Message) error

Update 更新消息

func (*MessageRepository) UpdateStatus

func (r *MessageRepository) UpdateStatus(msgID string, status string) error

UpdateStatus 更新消息状态

func (*MessageRepository) UpdateTokenCount

func (r *MessageRepository) UpdateTokenCount(msgID string, tokenCount int) error

UpdateTokenCount 更新消息token数量

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL