package model import ( "strings" "time" "gorm.io/gorm" ) type Post struct { *Model UserID int64 `json:"user_id"` CommentCount int64 `json:"comment_count"` CollectionCount int64 `json:"collection_count"` UpvoteCount int64 `json:"upvote_count"` IsTop int `json:"is_top"` IsEssence int `json:"is_essence"` IsLock int `json:"is_lock"` LatestRepliedOn int64 `json:"latest_replied_on"` Tags string `json:"tags"` AttachmentPrice int64 `json:"attachment_price"` IP string `json:"ip"` IPLoc string `json:"ip_loc"` } type PostFormated struct { ID int64 `json:"id"` UserID int64 `json:"user_id"` User *UserFormated `json:"user"` Contents []*PostContentFormated `json:"contents"` CommentCount int64 `json:"comment_count"` CollectionCount int64 `json:"collection_count"` UpvoteCount int64 `json:"upvote_count"` IsTop int `json:"is_top"` IsEssence int `json:"is_essence"` IsLock int `json:"is_lock"` LatestRepliedOn int64 `json:"latest_replied_on"` CreatedOn int64 `json:"created_on"` ModifiedOn int64 `json:"modified_on"` Tags map[string]int8 `json:"tags"` AttachmentPrice int64 `json:"attachment_price"` IPLoc string `json:"ip_loc"` } func (p *Post) Format() *PostFormated { if p.Model != nil { tagsMap := map[string]int8{} for _, tag := range strings.Split(p.Tags, ",") { tagsMap[tag] = 1 } return &PostFormated{ ID: p.ID, UserID: p.UserID, User: &UserFormated{}, Contents: []*PostContentFormated{}, CommentCount: p.CommentCount, CollectionCount: p.CollectionCount, UpvoteCount: p.UpvoteCount, IsTop: p.IsTop, IsEssence: p.IsEssence, IsLock: p.IsLock, LatestRepliedOn: p.LatestRepliedOn, CreatedOn: p.CreatedOn, ModifiedOn: p.ModifiedOn, AttachmentPrice: p.AttachmentPrice, Tags: tagsMap, IPLoc: p.IPLoc, } } return nil } func (p *Post) Create(db *gorm.DB) (*Post, error) { err := db.Create(&p).Error return p, err } func (s *Post) Delete(db *gorm.DB) error { return db.Model(&Post{}).Where("id = ? AND is_del = ?", s.Model.ID, 0).Updates(map[string]interface{}{ "deleted_on": time.Now().Unix(), "is_del": 1, }).Error } func (p *Post) Get(db *gorm.DB) (*Post, error) { var post Post if p.Model != nil && p.ID > 0 { db = db.Where("id = ? AND is_del = ?", p.ID, 0) } else { return nil, gorm.ErrRecordNotFound } err := db.First(&post).Error if err != nil { return &post, err } return &post, nil } func (p *Post) List(db *gorm.DB, conditions *ConditionsT, offset, limit int) ([]*Post, error) { var posts []*Post var err error if offset >= 0 && limit > 0 { db = db.Offset(offset).Limit(limit) } if p.UserID > 0 { db = db.Where("user_id = ?", p.UserID) } for k, v := range *conditions { if k == "ORDER" { db = db.Order(v) } else { db = db.Where(k, v) } } if err = db.Where("is_del = ?", 0).Find(&posts).Error; err != nil { return nil, err } return posts, nil } func (p *Post) Count(db *gorm.DB, conditions *ConditionsT) (int64, error) { var count int64 if p.UserID > 0 { db = db.Where("user_id = ?", p.UserID) } for k, v := range *conditions { if k != "ORDER" { db = db.Where(k, v) } } if err := db.Model(p).Count(&count).Error; err != nil { return 0, err } return count, nil } func (p *Post) Update(db *gorm.DB) error { return db.Model(&Post{}).Where("id = ? AND is_del = ?", p.Model.ID, 0).Save(p).Error }