68 lines
2.3 KiB
Go
68 lines
2.3 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// CaptainDocumentRepo provides data access for CaptainDocument.
|
|
type CaptainDocumentRepo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewCaptainDocumentRepo(db *gorm.DB) *CaptainDocumentRepo {
|
|
return &CaptainDocumentRepo{db: db}
|
|
}
|
|
|
|
func (r *CaptainDocumentRepo) Create(ctx context.Context, doc *model.CaptainDocument) error {
|
|
return r.db.WithContext(ctx).Create(doc).Error
|
|
}
|
|
|
|
func (r *CaptainDocumentRepo) GetByID(ctx context.Context, id uint) (*model.CaptainDocument, error) {
|
|
var doc model.CaptainDocument
|
|
if err := r.db.WithContext(ctx).First(&doc, id).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &doc, nil
|
|
}
|
|
|
|
func (r *CaptainDocumentRepo) Update(ctx context.Context, doc *model.CaptainDocument) error {
|
|
return r.db.WithContext(ctx).Save(doc).Error
|
|
}
|
|
|
|
func (r *CaptainDocumentRepo) Delete(ctx context.Context, id uint) error {
|
|
return r.db.WithContext(ctx).Delete(&model.CaptainDocument{}, id).Error
|
|
}
|
|
|
|
func (r *CaptainDocumentRepo) ListByAssistant(ctx context.Context, assistantID uint, offset, limit int) ([]model.CaptainDocument, int64, error) {
|
|
var docs []model.CaptainDocument
|
|
var count int64
|
|
db := r.db.WithContext(ctx).Model(&model.CaptainDocument{}).Where("assistant_id = ?", assistantID)
|
|
db.Count(&count)
|
|
if err := db.Offset(offset).Limit(limit).Find(&docs).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
return docs, count, nil
|
|
}
|
|
|
|
func (r *CaptainDocumentRepo) FindByExternalLink(ctx context.Context, assistantID uint, externalLink string) (*model.CaptainDocument, error) {
|
|
var doc model.CaptainDocument
|
|
if err := r.db.WithContext(ctx).Where("assistant_id = ? AND external_link = ?", assistantID, externalLink).First(&doc).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &doc, nil
|
|
}
|
|
|
|
// FindByStatus retrieves documents matching a specific status.
|
|
func (r *CaptainDocumentRepo) FindByStatus(ctx context.Context, assistantID uint, status model.DocumentStatus, offset, limit int) ([]model.CaptainDocument, int64, error) {
|
|
var docs []model.CaptainDocument
|
|
var count int64
|
|
db := r.db.WithContext(ctx).Model(&model.CaptainDocument{}).Where("assistant_id = ? AND status = ?", assistantID, status)
|
|
db.Count(&count)
|
|
if err := db.Offset(offset).Limit(limit).Find(&docs).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
return docs, count, nil
|
|
} |