fonchain-artistinfo/cmd/internal/dao/artistinfo_artwork.go

// Package dao -----------------------------
// @file      : artwork.go
// @author    : JJXu
// @contact   : wavingbear@163.com
// @time      : 2023/2/24 18:23
// -------------------------------------------
package dao

import (
	"errors"
	"github.com/fonchain/fonchain-artistinfo/cmd/model"
	"github.com/fonchain/fonchain-artistinfo/pb/artistInfoArtwork"
	db "github.com/fonchain/fonchain-artistinfo/pkg/db"
	"github.com/fonchain/fonchain-artistinfo/pkg/util/stime"
	"gorm.io/gorm"
	"time"
)

func CreateArtworkLockRecord(in *model.ArtworkLockRecord) error {
	var data = model.ArtworkLockRecord{
		ArtistUid:   in.ArtistUid,
		ArtworkUid:  in.ArtworkUid,
		Status:      in.Status,
		LockTime:    in.LockTime,
		AuditStatus: 1,
		//AuditMark:   in.AuditMark,
		//AuditMark2:  in.AuditMark2,
	}
	return db.DB.Create(&data).Error
}

func UpdateArtworkLockRecord(in *model.ArtworkLockRecord) error {
	var data = model.ArtworkLockRecord{
		ArtistUid:   in.ArtistUid,
		ArtworkUid:  in.ArtworkUid,
		Status:      in.Status,
		LockTime:    in.LockTime,
		AuditStatus: in.AuditStatus,
		AuditMark:   in.AuditMark,
		AuditMark2:  in.AuditMark2,
	}
	var thisData model.ArtworkLockRecord
	err := db.DB.Where("artist_uid = ?", in.ArtworkUid).First(&thisData).Error
	if err != nil {
		if err != gorm.ErrRecordNotFound {
			return errors.New("画作数据未找到")
		} else {
			return err
		}
	}
	data.Model = model.Model{
		ID:        thisData.ID,
		CreatedAt: thisData.CreatedAt,
	}
	return db.DB.Updates(&data).Error
}

func DeletedArtworkLockRecord(artworkUid ...string) error {
	if len(artworkUid) == 0 {
		return nil
	} else if len(artworkUid) == 1 {
		return db.DB.Where("artwork_uid = ? AND status != 2 ", artworkUid[0]).Delete(&model.ArtworkLockRecord{}).Error
	} else {
		return db.DB.Where("artwork_uid in ? AND status != 2 ", artworkUid).Delete(&model.ArtworkLockRecord{}).Error
	}
}

func BatchLockArtworks(artistUid string) error {
	var artworkUids []string
	db.DB.Model(model.ArtworkLockRecord{}).
		Where("status = 1 AND lock_time=''").Where(" artist_uid = ?", artistUid).
		Pluck("artwork_uid", &artworkUids)
	if artworkUids == nil {
		return nil
	}
	var updateMap = map[string]any{
		"status":    2,
		"lock_time": stime.TimeToString(time.Now(), stime.Format_Normal_YMDhms),
	}
	return db.DB.Model(model.ArtworkLockRecord{}).Where("artwork_uid in ?", artworkUids).Updates(&updateMap).Error
}

func BatchUnlockArtworks(artistUid string) error {
	//var artworkUids []string
	// 查询用户当前锁定的时间点
	//var nowLockTime string
	//err := db.DB.Model(model.ArtworkLockRecord{}).Select("MAX(lock_time)").Where("artist_uid = %s AND lock_time !='' AND STATUS =2", artistUid).Scan(&nowLockTime).Error
	//if err != nil {
	//	return err
	//}
	////查询此时间节点内的所有画作
	//db.DB.Model(model.ArtworkLockRecord{}).
	//	//Where("lock_time = (SELECT MAX(lock_time) FROM artwork_lock_record where artist_uid = %s AND lock_time !='' AND STATUS =2)", artistUid).
	//	//Joins("LEFT JOIN (lock_time = (select max(lock_time) form artwork_lock_record WHERE artist_uid = ?) lt", artistUid).
	//	//Where(fmt.Sprintf("lock_time = (select max(lock_time) form artwork_lock_record WHERE artist_uid = %s)", artistUid)).
	//	Where("status = 2").
	//	Pluck("artwork_uid", &artworkUids)
	//if artworkUids == nil {
	//	return nil
	//}
	return db.DB.Model(model.ArtworkLockRecord{}).Where("artist_uid = ? AND status =2 AND lock_time !=''", artistUid).Update("status", 3).Error
}

func GetArtworkLockDetail(artworkUid string) (data model.ArtworkLockRecord, err error) {
	err = db.DB.Where("artwork_uid = ?", artworkUid).First(&data).Error
	return
}
func GetArtworkLockRecords(req *artistInfoArtwork.GetArtworkLockRecordsRequest) (resp *artistInfoArtwork.ArtworkLockList, err error) {

	var (
		datas = []model.ArtworkLockRecord{}
		tx    = db.DB.Model(model.ArtworkLockRecord{}).Where("artist_uid = ?", req.ArtistUid).Order("lock_time desc")
	)
	resp = &artistInfoArtwork.ArtworkLockList{}

	switch req.QueryType {
	case artistInfoArtwork.ArtworkQueryMode_NowPreSaveArtwork: //当前暂存的画作
		tx = tx.Where("status = 1")
	case artistInfoArtwork.ArtworkQueryMode_NowLockedArtwork: //当前已锁定的画作
		tx = tx.Where("status = 2", req.ArtistUid)
	case artistInfoArtwork.ArtworkQueryMode_ArtistCanSee: //画家能看到的画作(暂存和锁定) //弃用,画家看到的是暂存的画作
		tx = tx.Where("status < 3")
	case artistInfoArtwork.ArtworkQueryMode_AllUnlockArtwork: //所有已解锁的画作(历史画作)
		tx = tx.Where("status = 3 OR status = 2")
		//case artistInfoArtwork.ArtworkQueryMode_AllHistoryArtwork: //所有历史画作
		//	tx = tx.Where("status > 1")
	}
	err = tx.Find(&datas).Error
	for _, v := range datas {
		resp.Data = append(resp.Data, &artistInfoArtwork.ArtistLockInfo{
			ArtistUid:   v.ArtistUid,
			ArtworkUid:  v.ArtworkUid,
			Status:      v.Status,
			LockTime:    v.LockTime,
			AuditStatus: v.AuditStatus,
			AuditMark:   v.AuditMark,
			AuditMark2:  v.AuditMark2,
			CreatedAt:   v.CreatedAt.Unix(),
			UpdatedAt:   v.UpdatedAt.Unix(),
			DeletedAt:   int64(v.DeletedAt),
		})
	}
	return
}
func HasBeenLocked(artistUid string) bool {
	//检查是否已经被锁定
	var hasLocked int64
	db.DB.Model(model.ArtworkLockRecord{}).Where("artist_uid = ? AND status = 2", artistUid).Count(&hasLocked)
	if hasLocked > 0 {
		return true
	}
	return false
}

func UpdateAuditStatus(artworkUid string, auditStatus int64, remark1 string, remark2 string) error {
	var thisData model.ArtworkLockRecord
	err := db.DB.Model(model.ArtworkLockRecord{}).Where("artwork_uid = ?", artworkUid).First(&thisData).Error
	if err != nil {
		if err != gorm.ErrRecordNotFound {
			return errors.New("画作数据未找到")
		} else {
			return err
		}
	}
	thisData.AuditStatus = auditStatus
	thisData.AuditMark = remark1
	thisData.AuditMark2 = remark2
	return db.DB.Save(&thisData).Error
}