user_message.go 2.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  1. package user
  2. import (
  3. logger "eta/eta_mini_ht_api/common/component/log"
  4. "eta/eta_mini_ht_api/models"
  5. "gorm.io/gorm"
  6. "time"
  7. )
  8. type MessageType string
  9. const (
  10. UnReadStatus StatusType = "UNREAD"
  11. ReadStatus StatusType = "READ"
  12. MaxBatchNum = 1000
  13. MyMessageColumns = "id,source_id,type,message"
  14. )
  15. // UserMessage 表示 user_message 表的模型
  16. type UserMessage struct {
  17. Id int `gorm:"primaryKey;autoIncrement;column:id"`
  18. AnalystId int `gorm:"column:analyst_id"`
  19. UserId int `gorm:"column:user_id"`
  20. SourceId int `gorm:"column:source_id"`
  21. Message string `gorm:"column:message"`
  22. Type SourceType `gorm:"column:type;type:enum('REPORT','VIDEO','AUDIO')"`
  23. Status StatusType `gorm:"column:status;type:enum('UNREAD','READ')"`
  24. CreatedTime time.Time `gorm:"column:created_time;type:timestamps;comment:创建时间"`
  25. UpdatedTime time.Time `gorm:"column:updated_time"`
  26. }
  27. func (u *UserMessage) BeforeCreate(_ *gorm.DB) (err error) {
  28. u.CreatedTime = time.Now()
  29. return
  30. }
  31. func CreateMessage(message UserMessage) (err error) {
  32. db := models.Main()
  33. return db.Create(&message).Error
  34. }
  35. func BatchInsertMessage(messages []UserMessage) bool {
  36. db := models.Main()
  37. //手动事务
  38. tx := db.Begin()
  39. err := db.CreateInBatches(messages, MaxBatchNum).Error
  40. if err != nil {
  41. logger.Error("批量插入消息失败:%v", err)
  42. tx.Rollback()
  43. return false
  44. }
  45. tx.Commit()
  46. return true
  47. }
  48. func NeedNotice(userId int, analystId int) bool {
  49. db := models.Main()
  50. var count int
  51. err := db.Model(&UserMessage{}).Select("count(*)").Where("user_id =? and analyst_id =? and status=?", userId, analystId, UnReadStatus).Scan(&count).Error
  52. if err != nil {
  53. logger.Error("统计未读消息失败:%v", err)
  54. return false
  55. }
  56. return count > 0
  57. }
  58. func GetUnReadMessageList(userId int) (messages []UserMessage, err error) {
  59. db := models.Main()
  60. err = db.Select(MyMessageColumns).Where("user_id=? and status=?", userId, UnReadStatus).Order("created_time desc").Find(&messages).Error
  61. return
  62. }
  63. func ReadMessage(userId int, messageId int) bool {
  64. db := models.Main()
  65. err := db.Model(&UserMessage{}).Where("id=? and user_id=?", messageId, userId).Update("status", ReadStatus).Error
  66. if err != nil {
  67. return false
  68. }
  69. return true
  70. }
  71. func ReadMessages(userId int, analystId int) bool {
  72. db := models.Main()
  73. err := db.Model(&UserMessage{}).Where("user_id=? and analyst_id =?", userId, analystId).Update("status", ReadStatus).Error
  74. if err != nil {
  75. return false
  76. }
  77. return true
  78. }