// Copyright 2022 ROC. All rights reserved. // Use of this source code is governed by a MIT style // license that can be found in the LICENSE file. package dbr import ( "time" "gorm.io/gorm" ) type Tag struct { *Model UserID int64 `json:"user_id"` Tag string `json:"tag"` QuoteNum int64 `json:"quote_num"` } type TopicUser struct { *Model UserID int64 `json:"user_id"` TopicID int64 `json:"topic_id"` AliasName string `json:"-"` Remark string `json:"-"` QuoteNum int64 `json:"quote_num"` IsTop int8 `json:"is_top"` IsPin int8 `json:"is_pin"` ReserveA string `json:"-"` ReserveB string `json:"-"` } type TagFormated struct { ID int64 `json:"id"` UserID int64 `json:"user_id"` User *UserFormated `json:"user"` Tag string `json:"tag"` QuoteNum int64 `json:"quote_num"` IsFollowing int8 `json:"is_following"` IsTop int8 `json:"is_top"` } func (t *Tag) Format() *TagFormated { if t.Model == nil { return &TagFormated{} } return &TagFormated{ ID: t.ID, UserID: t.UserID, User: &UserFormated{}, Tag: t.Tag, QuoteNum: t.QuoteNum, IsFollowing: 0, IsTop: 0, } } func (t *Tag) Get(db *gorm.DB) (*Tag, error) { var tag Tag if t.Model != nil && t.Model.ID > 0 { db = db.Where("id= ? AND is_del = ?", t.Model.ID, 0) } else { db = db.Where("tag = ? AND is_del = ?", t.Tag, 0) } err := db.First(&tag).Error if err != nil { return nil, err } return &tag, nil } func (t *Tag) Create(db *gorm.DB) (*Tag, error) { err := db.Create(&t).Error return t, err } func (t *Tag) Update(db *gorm.DB) error { return db.Model(&Tag{}).Where("id = ? AND is_del = ?", t.Model.ID, 0).Save(t).Error } func (t *Tag) Delete(db *gorm.DB) error { return db.Model(t).Where("id = ?", t.Model.ID).Updates(map[string]any{ "deleted_on": time.Now().Unix(), "is_del": 1, }).Error } func (t *Tag) List(db *gorm.DB, conditions *ConditionsT, offset, limit int) (tags []*Tag, err error) { if offset >= 0 && limit > 0 { db = db.Offset(offset).Limit(limit) } if t.UserID > 0 { db = db.Where("user_id = ?", t.UserID) } for k, v := range *conditions { if k == "ORDER" { db = db.Order(v) } else { db = db.Where(k, v) } } err = db.Where("is_del = 0 and quote_num > 0").Find(&tags).Error return } func (t *Tag) TagsFrom(db *gorm.DB, tags []string) (res []*Tag, err error) { err = db.Where("tag IN ?", tags).Find(&res).Error return }