template_user.go 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  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. UnionId string `gorm:"column:union_id;type:varchar(50);comment:union_id"`
  17. IsDeleted int `gorm:"column:is_deleted;type:int(11);comment:是否删除"`
  18. CreatedTime time.Time `gorm:"column:created_time;type:timestamps;comment:创建时间"`
  19. UpdatedTime time.Time `gorm:"column:updated_time;type:timestamps;comment:更新时间"`
  20. }
  21. func (t *TemplateUser) BeforeCreate(tx *gorm.DB) (err error) {
  22. t.CreatedTime = time.Now()
  23. t.IsDeleted = NotDeleted
  24. return
  25. }
  26. func GetUserByMobile(mobile string) (user TemplateUser, err error) {
  27. return queryByColumn("mobile", mobile)
  28. }
  29. func GetUserByOpenId(openId string) (user TemplateUser, err error) {
  30. return queryByColumn("open_id", openId)
  31. }
  32. func queryByColumn(column string, value string) (user TemplateUser, err error) {
  33. err = models.Main().Unscoped().Where(column, value).First(&user).Error
  34. return
  35. }
  36. func RegisterTemplateUser(user *TemplateUser) (err error) {
  37. err = models.Main().Create(&user).Error
  38. return
  39. }
  40. func BindMobile(userId int, mobile string) (err error) {
  41. err = models.Main().Where("id = ?", userId).Updates(map[string]interface{}{"mobile": mobile, "updated_time": time.Now()}).Error
  42. return
  43. }