package model import ( "errors" "message-pusher/common" "time" ) type Message struct { Id int `json:"id"` UserId int `json:"user_id" gorm:"index"` Title string `json:"title"` Description string `json:"description"` Desp string `json:"desp" gorm:"-:all"` // alias for description Content string `json:"content"` URL string `json:"url" gorm:"column:url"` Channel string `json:"channel"` Token string `json:"token" gorm:"-:all"` HTMLContent string `json:"html_content" gorm:"-:all"` Timestamp int64 `json:"timestamp" gorm:"type:int64"` Link string `json:"link" gorm:"unique;index"` } func GetMessageById(id int, userId int) (*Message, error) { if id == 0 || userId == 0 { return nil, errors.New("id 或 userId 为空!") } message := Message{Id: id, UserId: userId} err := DB.Where(message).First(&message).Error return &message, err } func GetMessageByLink(link string) (*Message, error) { if link == "" { return nil, errors.New("link 为空!") } message := Message{Link: link} err := DB.Where(message).First(&message).Error return &message, err } func GetMessagesByUserId(userId int, startIdx int, num int) (messages []*Message, err error) { err = DB.Where("user_id = ?", userId).Order("id desc").Limit(num).Offset(startIdx).Find(&messages).Error return messages, err } func DeleteMessageById(id int, userId int) (err error) { // Why we need userId here? In case user want to delete other's message. if id == 0 || userId == 0 { return errors.New("id 或 userId 为空!") } message := Message{Id: id, UserId: userId} err = DB.Where(message).First(&message).Error if err != nil { return err } return message.Delete() } func DeleteAllMessages() error { return DB.Exec("DELETE FROM messages").Error } func (message *Message) UpdateAndInsert(userId int) error { message.Link = common.GetUUID() message.Timestamp = time.Now().Unix() message.UserId = userId var err error err = DB.Create(message).Error return err } func (message *Message) Delete() error { err := DB.Delete(message).Error return err }