766 lines
28 KiB
Go
766 lines
28 KiB
Go
package repository
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"time"
|
||
|
||
"fashionapi/internal/dto"
|
||
"fashionapi/internal/model"
|
||
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// ReviewRepository 审核草稿的数据访问接口。
|
||
//
|
||
// 草稿由爬虫 worker 写入（见 IngestRepository），本接口负责「读 + 晋升」：
|
||
// 列出待审草稿、查看详情、审核前微调字段、通过时晋升到正式表、驳回时标记状态。
|
||
type ReviewRepository interface {
|
||
// ListDrafts 分页列出草稿；status 为空表示全部，否则按状态过滤。
|
||
// 排序：pending 优先，其次按 id 倒序（最新在前）。
|
||
ListDrafts(ctx context.Context, status string, page, size int) ([]model.BrandRunwayDraft, int64, error)
|
||
// DraftDetail 查单条草稿（不含图片）。
|
||
DraftDetail(ctx context.Context, id uint32) (*model.BrandRunwayDraft, error)
|
||
// DraftImages 查草稿的全部图片，按排序值升序。
|
||
DraftImages(ctx context.Context, draftID uint32) ([]model.BrandRunwayDraftImage, error)
|
||
// UpdateDraft 审核前微调草稿字段（白名单，杜绝任意列注入）。
|
||
UpdateDraft(ctx context.Context, id uint32, fields map[string]any) error
|
||
// SetDraftStatus 设置草稿状态（通过/驳回）与审核人/驳回理由。
|
||
SetDraftStatus(ctx context.Context, id uint32, status, reviewer, reason string) error
|
||
// SaveRunwayFromDraft 晋升：按实体键（brand_id + season_code + collection_type）幂等 upsert 到 brand_runways 正式表并替换图片，返回正式表主键。
|
||
SaveRunwayFromDraft(ctx context.Context, draftID uint32) (uint32, error)
|
||
// 街拍草稿镜像方法（与 runway 同构，但对象为 street_snap_drafts / street_snaps）。
|
||
ListStreetDrafts(ctx context.Context, status string, page, size int) ([]model.StreetSnapDraft, int64, error)
|
||
StreetDraftDetail(ctx context.Context, id uint32) (*model.StreetSnapDraft, error)
|
||
StreetDraftImages(ctx context.Context, draftID uint32) ([]model.StreetSnapDraftImage, error)
|
||
// SoftDeleteRunwayDraftImage 软删单张走秀草稿图片（审核阶段剔除坏图），并同步 image_count。
|
||
SoftDeleteRunwayDraftImage(ctx context.Context, draftID, imgID uint32) error
|
||
// SoftDeleteStreetDraftImage 软删单张街拍草稿图片，并同步 image_count。
|
||
SoftDeleteStreetDraftImage(ctx context.Context, draftID, imgID uint32) error
|
||
// AttachStreetDraftImages 把 imgIDs 变成 mainImgID 的副图（批量并入）。
|
||
// mainImgID 必须属于同一草稿、未被软删、且自身是主图（is_detail=0），否则返回 ErrInvalidImageGroup。
|
||
// 若某个被并入的图自身还带着副图，则它原有的副图一并改挂到 mainImgID ——
|
||
// 避免出现「副图的副图」破坏不变量（见计划「实现裁决」第 1 条）。
|
||
// 幂等：已是 mainImgID 副图的行重复并入不报错。
|
||
AttachStreetDraftImages(ctx context.Context, draftID, mainImgID uint32, imgIDs []uint32) error
|
||
// AttachPrevStreetDraftImage 把 imgID 并入「同草稿内 sort_order 紧邻的前一行所属组的主图」。
|
||
// imgID 是首行（没有上一行）时返回 ErrNotFound。
|
||
AttachPrevStreetDraftImage(ctx context.Context, draftID, imgID uint32) error
|
||
// DetachStreetDraftImage 把副图恢复为主图（is_detail=0, parent_image_id=0）。
|
||
// 对主图调用是幂等成功（无副作用）。
|
||
DetachStreetDraftImage(ctx context.Context, draftID, imgID uint32) error
|
||
// GetDraftImageKey 取单张草稿图片的S4 key（删除前清理用）；不存在返回空串。
|
||
GetDraftImageKey(ctx context.Context, kind string, draftID, imgID uint32) (string, error)
|
||
UpdateStreetDraft(ctx context.Context, id uint32, fields map[string]any) error
|
||
SetStreetDraftStatus(ctx context.Context, id uint32, status, reviewer, reason string) error
|
||
SaveStreetSnapFromDraft(ctx context.Context, draftID uint32) (uint32, error)
|
||
// CountDraftPending 统计某草稿表（dest 传表对应的零值指针，如 &model.BrandRunwayDraft{}）中指定状态的草稿数。
|
||
// 供各 ReviewModule 复用，新增模块无需再单独写计数方法。
|
||
CountDraftPending(ctx context.Context, status string, dest any) (int64, error)
|
||
}
|
||
|
||
type reviewRepository struct {
|
||
db *gorm.DB
|
||
}
|
||
|
||
// NewReviewRepository 创建审核仓储。
|
||
func NewReviewRepository(db *gorm.DB) ReviewRepository {
|
||
return &reviewRepository{db: db}
|
||
}
|
||
|
||
func (r *reviewRepository) ListDrafts(ctx context.Context, status string, page, size int) ([]model.BrandRunwayDraft, int64, error) {
|
||
if page < 1 {
|
||
page = 1
|
||
}
|
||
if size < 1 || size > 100 {
|
||
size = 20
|
||
}
|
||
q := r.db.WithContext(ctx).Model(&model.BrandRunwayDraft{}).Where("is_deleted = 0")
|
||
if status != "" {
|
||
q = q.Where("status = ?", status)
|
||
}
|
||
var total int64
|
||
if err := q.Count(&total).Error; err != nil {
|
||
return nil, 0, err
|
||
}
|
||
if total == 0 {
|
||
return []model.BrandRunwayDraft{}, 0, nil
|
||
}
|
||
var rows []model.BrandRunwayDraft
|
||
if err := q.
|
||
Order("CASE status WHEN '" + model.DraftStatusPending + "' THEN 0 ELSE 1 END ASC, id DESC").
|
||
Offset((page - 1) * size).
|
||
Limit(size).
|
||
Find(&rows).Error; err != nil {
|
||
return nil, 0, err
|
||
}
|
||
return rows, total, nil
|
||
}
|
||
|
||
func (r *reviewRepository) DraftDetail(ctx context.Context, id uint32) (*model.BrandRunwayDraft, error) {
|
||
var d model.BrandRunwayDraft
|
||
if err := r.db.WithContext(ctx).Where("id = ? AND is_deleted = 0", id).First(&d).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, ErrNotFound
|
||
}
|
||
return nil, err
|
||
}
|
||
return &d, nil
|
||
}
|
||
|
||
func (r *reviewRepository) DraftImages(ctx context.Context, draftID uint32) ([]model.BrandRunwayDraftImage, error) {
|
||
var imgs []model.BrandRunwayDraftImage
|
||
err := r.db.WithContext(ctx).
|
||
Where("draft_id = ? AND is_deleted = 0", draftID).
|
||
Order("sort_order ASC, id ASC").
|
||
Find(&imgs).Error
|
||
return imgs, err
|
||
}
|
||
|
||
// draftEditable 审核前可微调的字段白名单（键即列名）。
|
||
var draftEditable = map[string]bool{
|
||
"title_en": true,
|
||
"title_cn": true,
|
||
"description_en": true,
|
||
"description_cn": true,
|
||
"year": true,
|
||
"season": true,
|
||
"collection_type": true,
|
||
"season_code": true,
|
||
}
|
||
|
||
// UpdateDraft 只更新白名单内的字段，其余忽略，杜绝任意列注入。
|
||
func (r *reviewRepository) UpdateDraft(ctx context.Context, id uint32, fields map[string]any) error {
|
||
clean := map[string]any{}
|
||
for k, v := range fields {
|
||
if draftEditable[k] {
|
||
clean[k] = v
|
||
}
|
||
}
|
||
if len(clean) == 0 {
|
||
return nil
|
||
}
|
||
clean["updated_at"] = uint32(time.Now().Unix())
|
||
return r.db.WithContext(ctx).
|
||
Model(&model.BrandRunwayDraft{}).
|
||
Where("id = ?", id).
|
||
Updates(clean).Error
|
||
}
|
||
|
||
func (r *reviewRepository) SetDraftStatus(ctx context.Context, id uint32, status, reviewer, reason string) error {
|
||
return r.db.WithContext(ctx).
|
||
Model(&model.BrandRunwayDraft{}).
|
||
Where("id = ?", id).
|
||
Updates(map[string]any{
|
||
"status": status,
|
||
"reviewer": reviewer,
|
||
"reject_reason": reason,
|
||
"updated_at": uint32(time.Now().Unix()),
|
||
}).Error
|
||
}
|
||
|
||
// SaveRunwayFromDraft 晋升草稿到正式表：
|
||
// 1. 按实体键（brand_id + season_code + collection_type）查找已存在的正式走秀（幂等：重审同一草稿只更新，不重复建行）；
|
||
// 2. 存在则更新全部可编辑字段 + 图片数；不存在则新建；
|
||
// 3. 替换图片：软删旧图（is_deleted=1）+ 按草稿图片重建；
|
||
// 4. 返回正式表主键。
|
||
//
|
||
// 在事务内完成，保证草稿与正式表一致性。
|
||
func (r *reviewRepository) SaveRunwayFromDraft(ctx context.Context, draftID uint32) (uint32, error) {
|
||
var draft model.BrandRunwayDraft
|
||
if err := r.db.WithContext(ctx).Where("id = ? AND is_deleted = 0", draftID).First(&draft).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return 0, ErrNotFound
|
||
}
|
||
return 0, err
|
||
}
|
||
// 多来源聚合：当前草稿图 + 同实体已 approved 的其他草稿图（按 image URL 去重，
|
||
// S4 key 已是 sha1，同图 URL 必相同）。当前正被审核的草稿强制纳入，其余只取已审核来源。
|
||
curImgs, err := r.DraftImages(ctx, draftID)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
sibImgs, err := r.runwayApprovedSiblingImages(ctx, draftID, draft.BrandID, draft.SeasonCode, draft.CollectionType)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
imgs := unionRunwayDraftImages(curImgs, sibImgs)
|
||
|
||
// image_count 仅计主图（is_detail=0）：对外列表卡片「N 张」与详情默认画廊都只展示主图，
|
||
// 细节图作为主图的附属按需展开，不计入主图数。
|
||
mainCount := 0
|
||
for _, im := range imgs {
|
||
if im.IsDetail == 0 {
|
||
mainCount++
|
||
}
|
||
}
|
||
|
||
now := uint32(time.Now().Unix())
|
||
var runwayID uint32
|
||
err = r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
|
||
// 1) 按实体键查找既有正式走秀
|
||
var existing model.BrandRunway
|
||
if eErr := tx.Where("brand_id = ? AND season_code = ? AND collection_type = ? AND is_deleted = 0", draft.BrandID, draft.SeasonCode, draft.CollectionType).
|
||
Limit(1).Find(&existing).Error; eErr != nil {
|
||
return eErr
|
||
}
|
||
|
||
common := map[string]any{
|
||
"title_en": draft.TitleEn,
|
||
"title_cn": draft.TitleCn,
|
||
"description_en": draft.DescriptionEn,
|
||
"description_cn": draft.DescriptionCn,
|
||
"year": draft.Year,
|
||
"season": draft.Season,
|
||
"collection_type": draft.CollectionType,
|
||
"season_code": draft.SeasonCode,
|
||
"cover": draft.Cover,
|
||
"image_count": uint16(mainCount),
|
||
"updated_at": now,
|
||
}
|
||
|
||
if existing.ID != 0 {
|
||
runwayID = existing.ID
|
||
common["brand_id"] = existing.BrandID // 保留正式表的品牌归属
|
||
if uErr := tx.Model(&model.BrandRunway{}).Where("id = ?", runwayID).Updates(common).Error; uErr != nil {
|
||
return uErr
|
||
}
|
||
} else {
|
||
rw := &model.BrandRunway{
|
||
BrandID: draft.BrandID,
|
||
TitleEn: draft.TitleEn,
|
||
TitleCn: draft.TitleCn,
|
||
DescriptionEn: draft.DescriptionEn,
|
||
DescriptionCn: draft.DescriptionCn,
|
||
Year: draft.Year,
|
||
Season: draft.Season,
|
||
CollectionType: draft.CollectionType,
|
||
SeasonCode: draft.SeasonCode,
|
||
Cover: draft.Cover,
|
||
ImageCount: uint16(mainCount),
|
||
CreatedAt: now,
|
||
UpdatedAt: now,
|
||
}
|
||
if cErr := tx.Create(rw).Error; cErr != nil {
|
||
return cErr
|
||
}
|
||
runwayID = rw.ID
|
||
}
|
||
|
||
// 2) 替换图片：软删旧 + 重建
|
||
if dErr := tx.Model(&model.BrandRunwayImage{}).
|
||
Where("runway_id = ? AND is_deleted = 0", runwayID).
|
||
Updates(map[string]any{"is_deleted": 1, "updated_at": now}).Error; dErr != nil {
|
||
return dErr
|
||
}
|
||
if len(imgs) > 0 {
|
||
rows := make([]model.BrandRunwayImage, 0, len(imgs))
|
||
for i, im := range imgs {
|
||
rows = append(rows, model.BrandRunwayImage{
|
||
RunwayID: runwayID,
|
||
BrandID: draft.BrandID,
|
||
Image: im.Image,
|
||
Name: im.Name,
|
||
SortOrder: uint32(i + 1),
|
||
LookIndex: im.LookIndex,
|
||
IsDetail: im.IsDetail,
|
||
Phash: im.Phash,
|
||
IsDuplicate: im.IsDuplicate,
|
||
DupOf: im.DupOf,
|
||
CreatedAt: now,
|
||
UpdatedAt: now,
|
||
})
|
||
}
|
||
if cErr := tx.Create(&rows).Error; cErr != nil {
|
||
return cErr
|
||
}
|
||
}
|
||
return nil
|
||
})
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
return runwayID, nil
|
||
}
|
||
|
||
// CountDraftPending 统计传入表（dest 零值指针）中指定状态的草稿数；供各 ReviewModule 复用。
|
||
func (r *reviewRepository) CountDraftPending(ctx context.Context, status string, dest any) (int64, error) {
|
||
var n int64
|
||
if err := r.db.WithContext(ctx).
|
||
Model(dest).
|
||
Where("status = ? AND is_deleted = 0", status).
|
||
Count(&n).Error; err != nil {
|
||
return 0, err
|
||
}
|
||
return n, nil
|
||
}
|
||
|
||
// ---- 街拍草稿镜像实现（与 runway 同构） ----
|
||
|
||
func (r *reviewRepository) ListStreetDrafts(ctx context.Context, status string, page, size int) ([]model.StreetSnapDraft, int64, error) {
|
||
if page < 1 {
|
||
page = 1
|
||
}
|
||
if size < 1 || size > 100 {
|
||
size = 20
|
||
}
|
||
q := r.db.WithContext(ctx).Model(&model.StreetSnapDraft{}).Where("is_deleted = 0")
|
||
if status != "" {
|
||
q = q.Where("status = ?", status)
|
||
}
|
||
var total int64
|
||
if err := q.Count(&total).Error; err != nil {
|
||
return nil, 0, err
|
||
}
|
||
if total == 0 {
|
||
return []model.StreetSnapDraft{}, 0, nil
|
||
}
|
||
var rows []model.StreetSnapDraft
|
||
if err := q.
|
||
Order("CASE status WHEN '" + model.DraftStatusPending + "' THEN 0 ELSE 1 END ASC, id DESC").
|
||
Offset((page - 1) * size).
|
||
Limit(size).
|
||
Find(&rows).Error; err != nil {
|
||
return nil, 0, err
|
||
}
|
||
return rows, total, nil
|
||
}
|
||
|
||
func (r *reviewRepository) StreetDraftDetail(ctx context.Context, id uint32) (*model.StreetSnapDraft, error) {
|
||
var d model.StreetSnapDraft
|
||
if err := r.db.WithContext(ctx).Where("id = ? AND is_deleted = 0", id).First(&d).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, ErrNotFound
|
||
}
|
||
return nil, err
|
||
}
|
||
return &d, nil
|
||
}
|
||
|
||
func (r *reviewRepository) StreetDraftImages(ctx context.Context, draftID uint32) ([]model.StreetSnapDraftImage, error) {
|
||
var imgs []model.StreetSnapDraftImage
|
||
err := r.db.WithContext(ctx).
|
||
Where("draft_id = ? AND is_deleted = 0", draftID).
|
||
Order("sort_order ASC, id ASC").
|
||
Find(&imgs).Error
|
||
return imgs, err
|
||
}
|
||
|
||
// streetDraftEditable 街拍草稿可微调字段白名单（title / year / city）。
|
||
var streetDraftEditable = map[string]bool{
|
||
"title": true,
|
||
"year": true,
|
||
"city": true,
|
||
}
|
||
|
||
func (r *reviewRepository) UpdateStreetDraft(ctx context.Context, id uint32, fields map[string]any) error {
|
||
clean := map[string]any{}
|
||
for k, v := range fields {
|
||
if streetDraftEditable[k] {
|
||
clean[k] = v
|
||
}
|
||
}
|
||
if len(clean) == 0 {
|
||
return nil
|
||
}
|
||
clean["updated_at"] = uint32(time.Now().Unix())
|
||
return r.db.WithContext(ctx).
|
||
Model(&model.StreetSnapDraft{}).
|
||
Where("id = ?", id).
|
||
Updates(clean).Error
|
||
}
|
||
|
||
func (r *reviewRepository) SetStreetDraftStatus(ctx context.Context, id uint32, status, reviewer, reason string) error {
|
||
return r.db.WithContext(ctx).
|
||
Model(&model.StreetSnapDraft{}).
|
||
Where("id = ?", id).
|
||
Updates(map[string]any{
|
||
"status": status,
|
||
"reviewer": reviewer,
|
||
"reject_reason": reason,
|
||
"updated_at": uint32(time.Now().Unix()),
|
||
}).Error
|
||
}
|
||
|
||
// SaveStreetSnapFromDraft 晋升街拍草稿到正式表 street_snaps（按实体键 city+year 幂等 upsert，替换图片）。
|
||
// 逻辑与 SaveRunwayFromDraft 同构，但无 brand_id、有 city，图片明细落到 street_snap_images。
|
||
// 在事务内完成，保证草稿与正式表一致性。
|
||
func (r *reviewRepository) SaveStreetSnapFromDraft(ctx context.Context, draftID uint32) (uint32, error) {
|
||
var draft model.StreetSnapDraft
|
||
if err := r.db.WithContext(ctx).Where("id = ? AND is_deleted = 0", draftID).First(&draft).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return 0, ErrNotFound
|
||
}
|
||
return 0, err
|
||
}
|
||
curImgs, err := r.StreetDraftImages(ctx, draftID)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
sibImgs, err := r.streetApprovedSiblingImages(ctx, draftID, draft.City, draft.Year)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
imgs := unionStreetDraftImages(curImgs, sibImgs)
|
||
|
||
now := uint32(time.Now().Unix())
|
||
var snapID uint32
|
||
err = r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
|
||
var existing model.StreetSnap
|
||
if eErr := tx.Where("city = ? AND year = ? AND is_deleted = 0", draft.City, draft.Year).
|
||
Limit(1).Find(&existing).Error; eErr != nil {
|
||
return eErr
|
||
}
|
||
|
||
common := map[string]any{
|
||
"title": draft.Title,
|
||
"year": draft.Year,
|
||
"city": draft.City,
|
||
"cover": draft.Cover,
|
||
"image_count": uint16(len(imgs)),
|
||
"updated_at": now,
|
||
}
|
||
|
||
if existing.ID != 0 {
|
||
snapID = existing.ID
|
||
if uErr := tx.Model(&model.StreetSnap{}).Where("id = ?", snapID).Updates(common).Error; uErr != nil {
|
||
return uErr
|
||
}
|
||
} else {
|
||
snap := &model.StreetSnap{
|
||
Title: draft.Title,
|
||
Year: draft.Year,
|
||
City: draft.City,
|
||
Cover: draft.Cover,
|
||
ImageCount: uint16(len(imgs)),
|
||
CreatedAt: now,
|
||
UpdatedAt: now,
|
||
}
|
||
if cErr := tx.Create(snap).Error; cErr != nil {
|
||
return cErr
|
||
}
|
||
snapID = snap.ID
|
||
}
|
||
|
||
// 替换图片：软删旧 + 重建
|
||
if dErr := tx.Model(&model.StreetSnapImage{}).
|
||
Where("snap_id = ? AND is_deleted = 0", snapID).
|
||
Updates(map[string]any{"is_deleted": 1, "updated_at": now}).Error; dErr != nil {
|
||
return dErr
|
||
}
|
||
if len(imgs) > 0 {
|
||
// 第一遍：插入主图（is_detail=0），并记下「草稿行 id → 新正式行 id」的映射。
|
||
//
|
||
// 必须两遍插入：副图的 parent_image_id 存的是**草稿表**的行 id，
|
||
// 而这里插入的正式行拿到的是全新 id。若先插副图，父引用就会指向草稿表的旧 id，
|
||
// 详情页折叠结构随之错乱（见计划「任务 2」标题下的说明）。
|
||
newIDByDraftID := make(map[uint32]uint32, len(imgs))
|
||
mains := make([]model.StreetSnapImage, 0, len(imgs))
|
||
for _, im := range imgs {
|
||
if im.IsDetail == 1 {
|
||
continue
|
||
}
|
||
mains = append(mains, model.StreetSnapImage{
|
||
SnapID: snapID,
|
||
Image: im.Image,
|
||
Name: im.Name,
|
||
SortOrder: uint32(len(mains) + 1),
|
||
IsDetail: 0,
|
||
Phash: im.Phash,
|
||
IsDuplicate: im.IsDuplicate,
|
||
DupOf: im.DupOf,
|
||
CreatedAt: now,
|
||
UpdatedAt: now,
|
||
})
|
||
}
|
||
for i := range mains {
|
||
if cErr := tx.Create(&mains[i]).Error; cErr != nil {
|
||
return cErr
|
||
}
|
||
}
|
||
mainIdx := 0
|
||
for _, im := range imgs {
|
||
if im.IsDetail == 1 {
|
||
continue
|
||
}
|
||
newIDByDraftID[im.ID] = mains[mainIdx].ID
|
||
mainIdx++
|
||
}
|
||
|
||
// 第二遍：插入副图，把父引用改写成新主图行 id。
|
||
details := make([]model.StreetSnapImage, 0, len(imgs)-len(mains))
|
||
for _, im := range imgs {
|
||
if im.IsDetail != 1 {
|
||
continue
|
||
}
|
||
newParent, ok := newIDByDraftID[im.ParentImageID]
|
||
row := model.StreetSnapImage{
|
||
SnapID: snapID,
|
||
Image: im.Image,
|
||
Name: im.Name,
|
||
SortOrder: uint32(len(mains) + len(details) + 1),
|
||
IsDetail: 1,
|
||
Phash: im.Phash,
|
||
IsDuplicate: im.IsDuplicate,
|
||
DupOf: im.DupOf,
|
||
CreatedAt: now,
|
||
UpdatedAt: now,
|
||
}
|
||
if ok {
|
||
row.ParentImageID = newParent
|
||
} else {
|
||
row.IsDetail = 0 // 父行缺失：按主图渲染，与读侧容错口径一致
|
||
}
|
||
details = append(details, row)
|
||
}
|
||
if len(details) > 0 {
|
||
if cErr := tx.Create(&details).Error; cErr != nil {
|
||
return cErr
|
||
}
|
||
}
|
||
}
|
||
return nil
|
||
})
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
return snapID, nil
|
||
}
|
||
|
||
// runwayApprovedSiblingImages 返回同实体（品牌+季节码+系列）且已 approved 的其他草稿的图片，
|
||
// 用于多来源聚合：当前正被审核的草稿强制纳入，其余仅合并已审核来源，避免未审图泄露进正式表。
|
||
func (r *reviewRepository) runwayApprovedSiblingImages(ctx context.Context, excludeDraftID, brandID uint32, seasonCode, collectionType string) ([]model.BrandRunwayDraftImage, error) {
|
||
var drafts []model.BrandRunwayDraft
|
||
if err := r.db.WithContext(ctx).
|
||
Where("brand_id = ? AND season_code = ? AND collection_type = ? AND status = ? AND is_deleted = 0 AND id <> ?", brandID, seasonCode, collectionType, model.DraftStatusApproved, excludeDraftID).
|
||
Find(&drafts).Error; err != nil {
|
||
return nil, err
|
||
}
|
||
out := make([]model.BrandRunwayDraftImage, 0)
|
||
for _, d := range drafts {
|
||
imgs, err := r.DraftImages(ctx, d.ID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
out = append(out, imgs...)
|
||
}
|
||
return out, nil
|
||
}
|
||
|
||
// unionRunwayDraftImages 按 image URL 去重合并（S4 key 已是 sha1，同图 URL 必相同）。当前草稿图优先。
|
||
func unionRunwayDraftImages(a, b []model.BrandRunwayDraftImage) []model.BrandRunwayDraftImage {
|
||
seen := make(map[string]struct{}, len(a)+len(b))
|
||
out := make([]model.BrandRunwayDraftImage, 0, len(a)+len(b))
|
||
for _, im := range a {
|
||
if _, ok := seen[im.Image]; ok {
|
||
continue
|
||
}
|
||
seen[im.Image] = struct{}{}
|
||
out = append(out, im)
|
||
}
|
||
for _, im := range b {
|
||
if _, ok := seen[im.Image]; ok {
|
||
continue
|
||
}
|
||
seen[im.Image] = struct{}{}
|
||
out = append(out, im)
|
||
}
|
||
return out
|
||
}
|
||
|
||
// streetApprovedSiblingImages 同 city+year 已 approved 的其他街拍草稿图（多来源聚合）。
|
||
func (r *reviewRepository) streetApprovedSiblingImages(ctx context.Context, excludeDraftID uint32, city string, year uint16) ([]model.StreetSnapDraftImage, error) {
|
||
var drafts []model.StreetSnapDraft
|
||
if err := r.db.WithContext(ctx).
|
||
Where("city = ? AND year = ? AND status = ? AND is_deleted = 0 AND id <> ?", city, year, model.DraftStatusApproved, excludeDraftID).
|
||
Find(&drafts).Error; err != nil {
|
||
return nil, err
|
||
}
|
||
out := make([]model.StreetSnapDraftImage, 0)
|
||
for _, d := range drafts {
|
||
imgs, err := r.StreetDraftImages(ctx, d.ID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
out = append(out, imgs...)
|
||
}
|
||
return out, nil
|
||
}
|
||
|
||
// unionStreetDraftImages 按 image URL 去重合并。当前草稿图优先。
|
||
func unionStreetDraftImages(a, b []model.StreetSnapDraftImage) []model.StreetSnapDraftImage {
|
||
seen := make(map[string]struct{}, len(a)+len(b))
|
||
out := make([]model.StreetSnapDraftImage, 0, len(a)+len(b))
|
||
for _, im := range a {
|
||
if _, ok := seen[im.Image]; ok {
|
||
continue
|
||
}
|
||
seen[im.Image] = struct{}{}
|
||
out = append(out, im)
|
||
}
|
||
for _, im := range b {
|
||
if _, ok := seen[im.Image]; ok {
|
||
continue
|
||
}
|
||
seen[im.Image] = struct{}{}
|
||
out = append(out, im)
|
||
}
|
||
return out
|
||
}
|
||
|
||
// ── 草稿图片删除 ────────────────────────────────────────────────────────────────
|
||
|
||
// SoftDeleteRunwayDraftImage 软删单张走秀草稿图片（审核阶段剔除坏图），并同步草稿 image_count。
|
||
// 不删S4文件（与正式表 DeleteRunwayImage 行为一致：仅 DB 标记，避免误删被引用文件）。
|
||
func (r *reviewRepository) SoftDeleteRunwayDraftImage(ctx context.Context, draftID, imgID uint32) error {
|
||
if err := r.db.WithContext(ctx).
|
||
Model(&model.BrandRunwayDraftImage{}).
|
||
Where("id = ? AND draft_id = ? AND is_deleted = 0", imgID, draftID).
|
||
Updates(map[string]any{"is_deleted": 1, "updated_at": uint32(time.Now().Unix())}).Error; err != nil {
|
||
return err
|
||
}
|
||
return r.db.WithContext(ctx).
|
||
Exec("UPDATE brand_runway_drafts SET image_count = GREATEST(image_count - 1, 0) WHERE id = ?", draftID).Error
|
||
}
|
||
|
||
// SoftDeleteStreetDraftImage 软删单张街拍草稿图片，并同步 image_count。
|
||
func (r *reviewRepository) SoftDeleteStreetDraftImage(ctx context.Context, draftID, imgID uint32) error {
|
||
if err := r.db.WithContext(ctx).
|
||
Model(&model.StreetSnapDraftImage{}).
|
||
Where("id = ? AND draft_id = ? AND is_deleted = 0", imgID, draftID).
|
||
Updates(map[string]any{"is_deleted": 1, "updated_at": uint32(time.Now().Unix())}).Error; err != nil {
|
||
return err
|
||
}
|
||
return r.db.WithContext(ctx).
|
||
Exec("UPDATE street_snap_drafts SET image_count = GREATEST(image_count - 1, 0) WHERE id = ?", draftID).Error
|
||
}
|
||
|
||
// GetDraftImageKey 取单张草稿图片的S4 key（删除前清理用）；不存在返回空串。
|
||
// kind 决定查哪张草稿图片表（runway / street）；未知 kind 返回 ErrNotFound。
|
||
func (r *reviewRepository) GetDraftImageKey(ctx context.Context, kind string, draftID, imgID uint32) (string, error) {
|
||
switch kind {
|
||
case dto.IngestKindRunway:
|
||
var im model.BrandRunwayDraftImage
|
||
err := r.db.WithContext(ctx).Where("id = ? AND draft_id = ?", imgID, draftID).First(&im).Error
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return "", nil
|
||
}
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
return im.Image, nil
|
||
case dto.IngestKindStreet:
|
||
var im model.StreetSnapDraftImage
|
||
err := r.db.WithContext(ctx).Where("id = ? AND draft_id = ?", imgID, draftID).First(&im).Error
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return "", nil
|
||
}
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
return im.Image, nil
|
||
default:
|
||
return "", ErrNotFound
|
||
}
|
||
}
|
||
|
||
// ── 街拍草稿图片：主副图分组 ──────────────────────────────────────────────
|
||
|
||
// AttachStreetDraftImages 把 imgIDs 变成 mainImgID 的副图（批量并入）。
|
||
//
|
||
// 事务内三步：1) 校验主图存在、未软删、本身是主图；2) 先「改挂孙辈」——被并入图原有的副图
|
||
// 改挂到新主图（避免副图的副图）；3) 再把被并入图自身标为副图。第 2 步必须在第 3 步之前。
|
||
func (r *reviewRepository) AttachStreetDraftImages(ctx context.Context, draftID, mainImgID uint32, imgIDs []uint32) error {
|
||
ids := make([]uint32, 0, len(imgIDs))
|
||
for _, id := range imgIDs {
|
||
if id != 0 && id != mainImgID {
|
||
ids = append(ids, id)
|
||
}
|
||
}
|
||
if mainImgID == 0 || len(imgIDs) == 0 {
|
||
// 主图无效，或传入集合本身就是空的（调用方连一张待并入图都没给）。
|
||
return ErrInvalidImageGroup
|
||
}
|
||
if len(ids) == 0 {
|
||
// 集合里只有主图自身：视为无操作成功（忽略），不报错也不改动主图。
|
||
return nil
|
||
}
|
||
|
||
return r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
|
||
var main model.StreetSnapDraftImage
|
||
if err := tx.Where("id = ? AND draft_id = ? AND is_deleted = 0 AND is_detail = 0", mainImgID, draftID).
|
||
First(&main).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return ErrInvalidImageGroup
|
||
}
|
||
return err
|
||
}
|
||
|
||
now := uint32(time.Now().Unix())
|
||
if err := tx.Model(&model.StreetSnapDraftImage{}).
|
||
Where("draft_id = ? AND is_deleted = 0 AND parent_image_id IN ?", draftID, ids).
|
||
Updates(map[string]any{"parent_image_id": mainImgID, "updated_at": now}).Error; err != nil {
|
||
return err
|
||
}
|
||
return tx.Model(&model.StreetSnapDraftImage{}).
|
||
Where("draft_id = ? AND is_deleted = 0 AND id IN ?", draftID, ids).
|
||
Updates(map[string]any{"is_detail": 1, "parent_image_id": mainImgID, "updated_at": now}).Error
|
||
})
|
||
}
|
||
|
||
// AttachPrevStreetDraftImage 把 imgID 并入「同草稿内 sort_order 紧邻的前一行所属组的主图」。
|
||
// 折回组主图是刻意的：连续「并入上一张」应链式归并到同一主图，而不是造出副图的副图。
|
||
func (r *reviewRepository) AttachPrevStreetDraftImage(ctx context.Context, draftID, imgID uint32) error {
|
||
var target uint32
|
||
err := r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
|
||
var cur model.StreetSnapDraftImage
|
||
if err := tx.Where("id = ? AND draft_id = ? AND is_deleted = 0", imgID, draftID).First(&cur).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return ErrNotFound
|
||
}
|
||
return err
|
||
}
|
||
|
||
var prev model.StreetSnapDraftImage
|
||
err := tx.Where("draft_id = ? AND is_deleted = 0 AND (sort_order < ? OR (sort_order = ? AND id < ?))",
|
||
draftID, cur.SortOrder, cur.SortOrder, cur.ID).
|
||
Order("sort_order DESC, id DESC").First(&prev).Error
|
||
if err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return ErrNotFound
|
||
}
|
||
return err
|
||
}
|
||
|
||
if prev.IsDetail == 1 && prev.ParentImageID != 0 {
|
||
target = prev.ParentImageID
|
||
} else {
|
||
target = prev.ID
|
||
}
|
||
if target == imgID {
|
||
return ErrInvalidImageGroup
|
||
}
|
||
|
||
now := uint32(time.Now().Unix())
|
||
if err := tx.Model(&model.StreetSnapDraftImage{}).
|
||
Where("draft_id = ? AND is_deleted = 0 AND parent_image_id = ?", draftID, imgID).
|
||
Updates(map[string]any{"parent_image_id": target, "updated_at": now}).Error; err != nil {
|
||
return err
|
||
}
|
||
return tx.Model(&model.StreetSnapDraftImage{}).
|
||
Where("id = ? AND draft_id = ?", imgID, draftID).
|
||
Updates(map[string]any{"is_detail": 1, "parent_image_id": target, "updated_at": now}).Error
|
||
})
|
||
return err
|
||
}
|
||
|
||
// DetachStreetDraftImage 把副图恢复为主图。对主图调用为幂等成功。
|
||
// 拆出的图若自身带着副图，那些副图保持挂在它下面——它重新成为主图，不变量依然成立。
|
||
func (r *reviewRepository) DetachStreetDraftImage(ctx context.Context, draftID, imgID uint32) error {
|
||
return r.db.WithContext(ctx).
|
||
Model(&model.StreetSnapDraftImage{}).
|
||
Where("id = ? AND draft_id = ? AND is_deleted = 0 AND is_detail = 1", imgID, draftID).
|
||
Updates(map[string]any{"is_detail": 0, "parent_image_id": 0, "updated_at": uint32(time.Now().Unix())}).
|
||
Error
|
||
}
|