mirror of https://github.com/rocboss/paopao-ce
113 lines
2.5 KiB
113 lines
2.5 KiB
2 years ago
|
// 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
|
||
2 years ago
|
|
||
|
import (
|
||
|
"time"
|
||
|
|
||
|
"gorm.io/gorm"
|
||
|
)
|
||
|
|
||
|
type Tag struct {
|
||
|
*Model
|
||
|
UserID int64 `json:"user_id"`
|
||
|
Tag string `json:"tag"`
|
||
|
QuoteNum int64 `json:"quote_num"`
|
||
|
}
|
||
2 years ago
|
|
||
|
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"`
|
||
|
ReserveA string `json:"-"`
|
||
|
ReserveB string `json:"-"`
|
||
|
}
|
||
|
|
||
2 years ago
|
type TagFormated struct {
|
||
2 years ago
|
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"`
|
||
2 years ago
|
IsTop int8 `json:"is_top"`
|
||
2 years ago
|
}
|
||
|
|
||
|
func (t *Tag) Format() *TagFormated {
|
||
|
if t.Model == nil {
|
||
|
return &TagFormated{}
|
||
|
}
|
||
|
|
||
|
return &TagFormated{
|
||
2 years ago
|
ID: t.ID,
|
||
|
UserID: t.UserID,
|
||
|
User: &UserFormated{},
|
||
|
Tag: t.Tag,
|
||
|
QuoteNum: t.QuoteNum,
|
||
2 years ago
|
IsFollowing: 0,
|
||
2 years ago
|
IsTop: 0,
|
||
2 years ago
|
}
|
||
|
}
|
||
|
|
||
|
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 {
|
||
2 years ago
|
return nil, err
|
||
2 years ago
|
}
|
||
|
|
||
|
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
|
||
|
}
|
||
|
|
||
2 years ago
|
func (t *Tag) List(db *gorm.DB, conditions *ConditionsT, offset, limit int) (tags []*Tag, err error) {
|
||
2 years ago
|
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)
|
||
|
}
|
||
|
}
|
||
2 years ago
|
err = db.Where("is_del = 0 and quote_num > 0").Find(&tags).Error
|
||
|
return
|
||
2 years ago
|
}
|
||
|
|
||
|
func (t *Tag) TagsFrom(db *gorm.DB, tags []string) (res []*Tag, err error) {
|
||
|
err = db.Where("tag IN ?", tags).Find(&res).Error
|
||
|
return
|
||
|
}
|