template_user.go 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. package user
  2. import (
  3. "eta/eta_mini_ht_api/models"
  4. "gorm.io/gorm"
  5. "time"
  6. )
  7. const (
  8. NotDeleted = 0
  9. Deleted = 1
  10. )
  11. type TemplateUser struct {
  12. Id int `gorm:"column:id;primaryKey;autoIncrement:'id'"`
  13. Username string `gorm:"column:username;type:varchar(20);comment:用户名"`
  14. Mobile string `gorm:"column:mobile;type:varchar(15);comment:手机号"`
  15. OpenId string `gorm:"column:open_id;type:varchar(50);comment:open_id"`
  16. GzhOpenId string `gorm:"column:gzh_open_id;type:varchar(255);comment:gzh_open_id"`
  17. UnionId string `gorm:"column:union_id;type:varchar(50);comment:union_id"`
  18. ReadCount int `gorm:"column:read_count;type:int(11);comment:阅读次数"`
  19. LastReadTime time.Time `gorm:"column:last_read_time;type:timestamps;comment:最后阅读时间"`
  20. IsDeleted int `gorm:"column:is_deleted;type:int(11);comment:是否删除"`
  21. CreatedTime time.Time `gorm:"column:created_time;type:timestamps;comment:创建时间"`
  22. UpdatedTime time.Time `gorm:"column:updated_time;type:timestamps;comment:更新时间"`
  23. }
  24. func (t *TemplateUser) BeforeCreate(tx *gorm.DB) (err error) {
  25. t.CreatedTime = time.Now()
  26. t.IsDeleted = NotDeleted
  27. return
  28. }
  29. func GetUserByMobile(mobile string) (user TemplateUser, err error) {
  30. return queryByColumn("mobile", mobile)
  31. }
  32. func GetUserByIdAndCountReadTimes(userId int) (err error) {
  33. return models.Main().Model(&TemplateUser{}).Where("id = ?", userId).Update("read_count", gorm.Expr("read_count + ?", 1)).Error
  34. }
  35. func GetUserByOpenId(openId string) (user TemplateUser, err error) {
  36. return queryByColumn("open_id", openId)
  37. }
  38. func GetUserByUnionId(unionId string) (user TemplateUser, err error) {
  39. return queryByColumn("union_id", unionId)
  40. }
  41. func queryByColumn(column string, value string) (user TemplateUser, err error) {
  42. err = models.Main().Unscoped().Where(column, value).First(&user).Error
  43. return
  44. }
  45. func RegisterTemplateUser(user *TemplateUser) (err error) {
  46. err = models.Main().Create(&user).Error
  47. return
  48. }
  49. func BindMobile(userId int, mobile string) (err error) {
  50. err = models.Main().Table("template_users").Where("id = ?", userId).Updates(map[string]interface{}{"mobile": mobile, "updated_time": time.Now()}).Error
  51. return
  52. }
  53. func BindXcxOpenId(userId int, xcxOpenId string) (err error) {
  54. err = models.Main().Table("template_users").Where("id = ?", userId).Updates(map[string]interface{}{"open_id": xcxOpenId, "updated_time": time.Now()}).Error
  55. return
  56. }
  57. func BindGzhOpenId(userId int, gzhOpenId string) (err error) {
  58. err = models.Main().Table("template_users").Where("id = ?", userId).Updates(map[string]interface{}{"gzh_open_id": gzhOpenId, "updated_time": time.Now()}).Error
  59. return
  60. }
  61. func GetUserByGzhOpenId(gzhOpenId string) (user TemplateUser, err error) {
  62. return queryByColumn("gzh_open_id", gzhOpenId)
  63. }