mirror of
https://github.com/ctwj/urldb.git
synced 2025-11-25 03:15:04 +08:00
101 lines
2.9 KiB
Go
101 lines
2.9 KiB
Go
package repo
|
||
|
||
import (
|
||
"time"
|
||
|
||
"github.com/ctwj/urldb/db/entity"
|
||
"github.com/ctwj/urldb/utils"
|
||
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// CksRepository Cks的Repository接口
|
||
type CksRepository interface {
|
||
BaseRepository[entity.Cks]
|
||
FindByPanID(panID uint) ([]entity.Cks, error)
|
||
FindByIds(ids []uint) ([]*entity.Cks, error)
|
||
FindByIsValid(isValid bool) ([]entity.Cks, error)
|
||
UpdateSpace(id uint, space, leftSpace int64) error
|
||
DeleteByPanID(panID uint) error
|
||
UpdateWithAllFields(cks *entity.Cks) error
|
||
}
|
||
|
||
// CksRepositoryImpl Cks的Repository实现
|
||
type CksRepositoryImpl struct {
|
||
BaseRepositoryImpl[entity.Cks]
|
||
}
|
||
|
||
// NewCksRepository 创建Cks Repository
|
||
func NewCksRepository(db *gorm.DB) CksRepository {
|
||
return &CksRepositoryImpl{
|
||
BaseRepositoryImpl: BaseRepositoryImpl[entity.Cks]{db: db},
|
||
}
|
||
}
|
||
|
||
// FindByPanID 根据PanID查找
|
||
func (r *CksRepositoryImpl) FindByPanID(panID uint) ([]entity.Cks, error) {
|
||
var cks []entity.Cks
|
||
err := r.db.Where("pan_id = ?", panID).Find(&cks).Error
|
||
return cks, err
|
||
}
|
||
|
||
// FindByIsValid 根据有效性查找
|
||
func (r *CksRepositoryImpl) FindByIsValid(isValid bool) ([]entity.Cks, error) {
|
||
var cks []entity.Cks
|
||
err := r.db.Where("is_valid = ?", isValid).Find(&cks).Error
|
||
return cks, err
|
||
}
|
||
|
||
// UpdateSpace 更新空间信息
|
||
func (r *CksRepositoryImpl) UpdateSpace(id uint, space, leftSpace int64) error {
|
||
return r.db.Model(&entity.Cks{}).Where("id = ?", id).
|
||
Updates(map[string]interface{}{
|
||
"space": space,
|
||
"left_space": leftSpace,
|
||
}).Error
|
||
}
|
||
|
||
// DeleteByPanID 根据PanID删除
|
||
func (r *CksRepositoryImpl) DeleteByPanID(panID uint) error {
|
||
return r.db.Where("pan_id = ?", panID).Delete(&entity.Cks{}).Error
|
||
}
|
||
|
||
// FindAll 查找所有Cks，预加载Pan关联数据
|
||
func (r *CksRepositoryImpl) FindAll() ([]entity.Cks, error) {
|
||
var cks []entity.Cks
|
||
err := r.db.Preload("Pan").Find(&cks).Error
|
||
return cks, err
|
||
}
|
||
|
||
// FindByID 根据ID查找Cks，预加载Pan关联数据
|
||
func (r *CksRepositoryImpl) FindByID(id uint) (*entity.Cks, error) {
|
||
startTime := utils.GetCurrentTime()
|
||
var cks entity.Cks
|
||
err := r.db.Preload("Pan").First(&cks, id).Error
|
||
queryDuration := time.Since(startTime)
|
||
if err != nil {
|
||
utils.Debug("FindByID失败: ID=%d, 错误=%v, 查询耗时=%v", id, err, queryDuration)
|
||
return nil, err
|
||
}
|
||
utils.Debug("FindByID成功: ID=%d, 查询耗时=%v", id, queryDuration)
|
||
return &cks, nil
|
||
}
|
||
|
||
func (r *CksRepositoryImpl) FindByIds(ids []uint) ([]*entity.Cks, error) {
|
||
startTime := utils.GetCurrentTime()
|
||
var cks []*entity.Cks
|
||
err := r.db.Preload("Pan").Where("id IN ?", ids).Find(&cks).Error
|
||
queryDuration := time.Since(startTime)
|
||
if err != nil {
|
||
utils.Debug("FindByIds失败: IDs数量=%d, 错误=%v, 查询耗时=%v", len(ids), err, queryDuration)
|
||
return nil, err
|
||
}
|
||
utils.Debug("FindByIds成功: 找到%d个账号，查询耗时=%v", len(cks), queryDuration)
|
||
return cks, nil
|
||
}
|
||
|
||
// UpdateWithAllFields 更新Cks，包括零值字段
|
||
func (r *CksRepositoryImpl) UpdateWithAllFields(cks *entity.Cks) error {
|
||
return r.db.Save(cks).Error
|
||
}
|