eta_mini_ht_api/models/user/user_bookmark.go

package user

import (
	"eta/eta_mini_ht_api/models"
	"gorm.io/gorm/clause"
	"time"
)

type Status string
type SourceType string

const (
	Marked Status     = "marked"
	Unmark Status     = "unmark"
	Report SourceType = "report"
	Chart  SourceType = "chart"
)

type UserBookmark struct {
	ID          int        `gorm:"primaryKey;column:id;type:int(11);not null;comment:主键"`
	UserID      int        `gorm:"column:user_id;type:int(11);not null;comment:用户Id"`
	SourceID    int        `gorm:"column:source_id;type:int(11);not null;comment:资源Id"`
	SourceType  SourceType `gorm:"column:source_type;type:enum('report','chart');not null;comment:资源类型：report-研报 chart-图表"`
	Status      Status     `gorm:"column:status;type:enum('marked','unmark');not null;comment:是否收藏 :marked-收藏 ,unmark-取消收藏"`
	CreatedTime time.Time  `gorm:"column:created_time;type:datetime;comment:创建时间"`
	UpdatedTime time.Time  `gorm:"column:updated_time;type:datetime;default:null;onUpdate:CURRENT_TIMESTAMP;comment:更新时间"`
}

func (UserBookmark) TableName() string {
	return "user_bookmarks"
}

func BookMark(templateUserId int, sourceId int, sourceType SourceType) (err error) {
	db := models.Main()
	OnConflictFunc := clause.OnConflict{
		Columns:   []clause.Column{{Name: "source_id"}, {Name: "source_type"}},
		DoUpdates: clause.AssignmentColumns([]string{"status"}),
	}
	err = db.Clauses(OnConflictFunc).Create(&UserBookmark{
		UserID:      templateUserId,
		SourceID:    sourceId,
		SourceType:  sourceType,
		Status:      Marked,
		CreatedTime: time.Now(),
	}).Error
	return
}

func UnBookMark(templateUserId int, sourceId int, sourceType SourceType) (err error) {
	db := models.Main()
	err = db.Model(&UserBookmark{}).Where("user_id = ? AND source_id = ? AND source_type = ?", templateUserId, sourceId, sourceType).Update("status", Unmark).Error
	return
}

func CheckBookMarkStatus(templateUserId int, sourceId int, sourceType SourceType) (bookmark UserBookmark, err error) {
	db := models.Main()
	err = db.Model(&UserBookmark{}).Select("*").Where("user_id = ? AND source_id = ? AND source_type = ?", templateUserId, sourceId, sourceType).First(&bookmark).Error
	return
}