92 lines
3.3 KiB
Go
92 lines
3.3 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// CopilotThreadRepo provides data access for CopilotThread.
|
|
type CopilotThreadRepo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewCopilotThreadRepo(db *gorm.DB) *CopilotThreadRepo {
|
|
return &CopilotThreadRepo{db: db}
|
|
}
|
|
|
|
func (r *CopilotThreadRepo) DB() *gorm.DB {
|
|
return r.db
|
|
}
|
|
|
|
func (r *CopilotThreadRepo) Create(ctx context.Context, thread *model.CopilotThread) error {
|
|
return r.db.WithContext(ctx).Create(thread).Error
|
|
}
|
|
|
|
func (r *CopilotThreadRepo) GetByID(ctx context.Context, id uint) (*model.CopilotThread, error) {
|
|
var thread model.CopilotThread
|
|
if err := r.db.WithContext(ctx).Preload("Messages").Preload("User").Preload("Assistant").First(&thread, id).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &thread, nil
|
|
}
|
|
|
|
func (r *CopilotThreadRepo) GetByAccountUserAndID(ctx context.Context, accountID, userID, id uint) (*model.CopilotThread, error) {
|
|
var thread model.CopilotThread
|
|
if err := r.db.WithContext(ctx).
|
|
Preload("Messages").Preload("User").Preload("Assistant").
|
|
Where("account_id = ? AND user_id = ? AND id = ?", accountID, userID, id).
|
|
First(&thread).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &thread, nil
|
|
}
|
|
|
|
func (r *CopilotThreadRepo) Update(ctx context.Context, thread *model.CopilotThread) error {
|
|
return r.db.WithContext(ctx).Save(thread).Error
|
|
}
|
|
|
|
func (r *CopilotThreadRepo) Delete(ctx context.Context, id uint) error {
|
|
return r.db.WithContext(ctx).Delete(&model.CopilotThread{}, id).Error
|
|
}
|
|
|
|
func (r *CopilotThreadRepo) DeleteByAccountUser(ctx context.Context, accountID, userID, id uint) error {
|
|
return r.db.WithContext(ctx).Where("account_id = ? AND user_id = ? AND id = ?", accountID, userID, id).Delete(&model.CopilotThread{}).Error
|
|
}
|
|
|
|
func (r *CopilotThreadRepo) ListByUser(ctx context.Context, accountID, userID uint, offset, limit int) ([]model.CopilotThread, int64, error) {
|
|
var threads []model.CopilotThread
|
|
var count int64
|
|
db := r.db.WithContext(ctx).Model(&model.CopilotThread{}).Where("account_id = ? AND user_id = ?", accountID, userID)
|
|
db.Count(&count)
|
|
if err := db.Preload("User").Preload("Assistant").Order("created_at DESC").Offset(offset).Limit(limit).Find(&threads).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
return threads, count, nil
|
|
}
|
|
|
|
// FindByAccountID retrieves all threads for a given account with pagination.
|
|
func (r *CopilotThreadRepo) FindByAccountID(ctx context.Context, accountID uint, offset, limit int) ([]model.CopilotThread, int64, error) {
|
|
var threads []model.CopilotThread
|
|
var count int64
|
|
db := r.db.WithContext(ctx).Model(&model.CopilotThread{}).Where("account_id = ?", accountID)
|
|
db.Count(&count)
|
|
if err := db.Offset(offset).Limit(limit).Find(&threads).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
return threads, count, nil
|
|
}
|
|
|
|
// FindByAssistantID retrieves all threads linked to a given assistant.
|
|
func (r *CopilotThreadRepo) FindByAssistantID(ctx context.Context, assistantID uint, offset, limit int) ([]model.CopilotThread, int64, error) {
|
|
var threads []model.CopilotThread
|
|
var count int64
|
|
db := r.db.WithContext(ctx).Model(&model.CopilotThread{}).Where("assistant_id = ?", assistantID)
|
|
db.Count(&count)
|
|
if err := db.Offset(offset).Limit(limit).Find(&threads).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
return threads, count, nil
|
|
}
|