80 lines
2.3 KiB
Go
80 lines
2.3 KiB
Go
package mysqldao
|
||
|
||
import (
|
||
"context"
|
||
|
||
"git.lzcwy.cn/zhengchiliu/zt-closet-manage-backend/global"
|
||
"git.lzcwy.cn/zhengchiliu/zt-closet-manage-backend/internal/mysqlmodel"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// ClosetShareDao 衣柜分享数据访问
|
||
type ClosetShareDao struct {
|
||
db *gorm.DB
|
||
}
|
||
|
||
// NewClosetShareDao 创建衣柜分享 DAO
|
||
func NewClosetShareDao() *ClosetShareDao {
|
||
return &ClosetShareDao{db: global.MysqlEngine}
|
||
}
|
||
|
||
// ClosetShareListParams 分享分页列表筛选参数。
|
||
type ClosetShareListParams struct {
|
||
UserID int64 // 用户 ID，0 表示全部
|
||
Status int8 // 分享状态，0 表示全部
|
||
Page int
|
||
Size int
|
||
}
|
||
|
||
// ClosetShareList 分页查询衣柜分享
|
||
func (d *ClosetShareDao) ClosetShareList(ctx context.Context, p ClosetShareListParams) ([]mysqlmodel.ClosetShare, int64, error) {
|
||
ctx, span := global.Tracer.Start(ctx, "mysqldao.ClosetShareList")
|
||
defer span.End()
|
||
|
||
var list []mysqlmodel.ClosetShare
|
||
var total int64
|
||
q := d.db.WithContext(ctx).Model(&mysqlmodel.ClosetShare{})
|
||
if p.UserID != 0 {
|
||
q = q.Where("user_id = ?", p.UserID)
|
||
}
|
||
if p.Status != 0 {
|
||
q = q.Where("status = ?", p.Status)
|
||
}
|
||
if err := q.Count(&total).Error; err != nil {
|
||
span.RecordError(err)
|
||
return nil, 0, err
|
||
}
|
||
if err := q.Offset((p.Page - 1) * p.Size).Limit(p.Size).Order("id DESC").Find(&list).Error; err != nil {
|
||
span.RecordError(err)
|
||
return nil, 0, err
|
||
}
|
||
return list, total, nil
|
||
}
|
||
|
||
// ClosetShareGetByUUID 按分享 UUID 查询
|
||
func (d *ClosetShareDao) ClosetShareGetByUUID(ctx context.Context, shareUUID string) (*mysqlmodel.ClosetShare, error) {
|
||
ctx, span := global.Tracer.Start(ctx, "mysqldao.ClosetShareGetByUUID")
|
||
defer span.End()
|
||
|
||
var share mysqlmodel.ClosetShare
|
||
if err := d.db.WithContext(ctx).Where("share_uuid = ?", shareUUID).First(&share).Error; err != nil {
|
||
span.RecordError(err)
|
||
return nil, err
|
||
}
|
||
return &share, nil
|
||
}
|
||
|
||
// ClosetShareUpdateStatus 更新分享状态（封禁/解封）
|
||
func (d *ClosetShareDao) ClosetShareUpdateStatus(ctx context.Context, shareUUID string, status int8) error {
|
||
ctx, span := global.Tracer.Start(ctx, "mysqldao.ClosetShareUpdateStatus")
|
||
defer span.End()
|
||
|
||
if err := d.db.WithContext(ctx).Model(&mysqlmodel.ClosetShare{}).
|
||
Where("share_uuid = ?", shareUUID).
|
||
UpdateColumn("status", status).Error; err != nil {
|
||
span.RecordError(err)
|
||
return err
|
||
}
|
||
return nil
|
||
}
|