50 lines
1.7 KiB
Go
50 lines
1.7 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// CopilotSuggestionRepo provides data access for CopilotSuggestionMessage.
|
|
type CopilotSuggestionRepo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewCopilotSuggestionRepo(db *gorm.DB) *CopilotSuggestionRepo {
|
|
return &CopilotSuggestionRepo{db: db}
|
|
}
|
|
|
|
func (r *CopilotSuggestionRepo) Create(ctx context.Context, msg *model.CopilotSuggestionMessage) error {
|
|
return r.db.WithContext(ctx).Create(msg).Error
|
|
}
|
|
|
|
func (r *CopilotSuggestionRepo) GetByID(ctx context.Context, id uint) (*model.CopilotSuggestionMessage, error) {
|
|
var msg model.CopilotSuggestionMessage
|
|
if err := r.db.WithContext(ctx).First(&msg, id).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &msg, nil
|
|
}
|
|
|
|
func (r *CopilotSuggestionRepo) FindByConversationID(ctx context.Context, accountID, conversationID uint, offset, limit int) ([]model.CopilotSuggestionMessage, int64, error) {
|
|
var msgs []model.CopilotSuggestionMessage
|
|
var count int64
|
|
db := r.db.WithContext(ctx).Model(&model.CopilotSuggestionMessage{}).
|
|
Where("account_id = ? AND conversation_id = ?", accountID, conversationID)
|
|
db.Count(&count)
|
|
if err := db.Offset(offset).Limit(limit).Order("created_at ASC").Find(&msgs).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
return msgs, count, nil
|
|
}
|
|
|
|
func (r *CopilotSuggestionRepo) UpdateStatus(ctx context.Context, id uint, status model.CopilotSuggestionStatus) error {
|
|
return r.db.WithContext(ctx).Model(&model.CopilotSuggestionMessage{}).
|
|
Where("id = ?", id).Update("status", status).Error
|
|
}
|
|
|
|
func (r *CopilotSuggestionRepo) Delete(ctx context.Context, id uint) error {
|
|
return r.db.WithContext(ctx).Delete(&model.CopilotSuggestionMessage{}, id).Error
|
|
} |