84 lines
2.7 KiB
Go
84 lines
2.7 KiB
Go
package model
|
|
|
|
import (
|
|
"SciencesServer/utils"
|
|
"gorm.io/gorm"
|
|
"time"
|
|
)
|
|
|
|
// UserInstance 账号信息
|
|
type UserInstance struct {
|
|
Model
|
|
UUID uint64 `gorm:"column:uuid;uniqueIndex:idx_tenant_user_uuid;type:int;default:0;comment:用户唯一UUID" json:"-"`
|
|
Source UserInstanceSource `gorm:"column:source;type:tinyint(1);default:1;comment:账号来源" json:"source"`
|
|
Avatar string `gorm:"column:avatar;type:varchar(255);default:'';comment:头像" json:"avatar"`
|
|
Name string `gorm:"column:name;type:varchar(20);default:'';comment:真实姓名" json:"name"`
|
|
Mobile string `gorm:"column:mobile;index:idx_user_instance_mobile;type:varchar(15);default:'';comment:联系方式" json:"mobile"`
|
|
Gender
|
|
Vip UserInstanceVipKind `gorm:"column:vip;type:tinyint(1);default:0;comment:会员状态/类型" json:"vip"`
|
|
VipDeadline time.Time `gorm:"column:vip_deadline;type:datetime;default:null;comment:会员过期时间" json:"vip_deadline"`
|
|
Identity int `gorm:"column:identity;type:int(8);default:0;comment:身份信息" json:"-"`
|
|
Password string `gorm:"column:password;type:varchar(100);default:'';comment:密码" json:"-"`
|
|
Salt string `gorm:"column:salt;type:varchar(10);default:'';comment:盐值" json:"-"`
|
|
AccountStatus
|
|
ModelDeleted
|
|
ModelAt
|
|
}
|
|
|
|
// UserInstanceSource 账号来源
|
|
type UserInstanceSource int
|
|
|
|
const (
|
|
// UserInstanceSourceForLocal 本地注册
|
|
UserInstanceSourceForLocal UserInstanceSource = iota + 1
|
|
// UserInstanceSourceForWechat 微信登陆
|
|
UserInstanceSourceForWechat
|
|
)
|
|
|
|
// UserInstanceVipKind 用户会员类型
|
|
type UserInstanceVipKind int
|
|
|
|
const (
|
|
// UserInstanceVipForOrdinary 普通VIP
|
|
UserInstanceVipForOrdinary UserInstanceVipKind = iota + 1
|
|
// UserInstanceVipForSenior 高级VIP
|
|
UserInstanceVipForSenior
|
|
)
|
|
|
|
func (m *UserInstance) TableName() string {
|
|
return "user_instance"
|
|
}
|
|
|
|
func (m *UserInstance) BeforeCreate(db *gorm.DB) error {
|
|
m.SetPasswordAttribute()
|
|
snowflake, _ := utils.NewSnowflake(1)
|
|
m.UUID = uint64(snowflake.GetID())
|
|
m.Status = AccountStatusForEnable
|
|
m.CreatedAt = time.Now()
|
|
return nil
|
|
}
|
|
|
|
func (m *UserInstance) GetAvatarAttribute(domain string) string {
|
|
if m.Avatar == "" {
|
|
return ""
|
|
}
|
|
compile := "^(http|https):"
|
|
if utils.ValidateCompile(m.Avatar, compile) {
|
|
return m.Avatar
|
|
}
|
|
return domain + m.Avatar
|
|
}
|
|
|
|
func (m *UserInstance) SetPasswordAttribute() {
|
|
m.Salt = utils.GetRandomString(8)
|
|
m.Password = utils.HashString([]byte(utils.Md5String(m.Password, m.Salt)))
|
|
}
|
|
|
|
func (m *UserInstance) VipStatus() bool {
|
|
return m.Vip != 0 && m.VipDeadline.After(time.Now())
|
|
}
|
|
|
|
func NewUserInstance() *UserInstance {
|
|
return &UserInstance{}
|
|
}
|