IOT/SERVER/IotAdmin/app/system/models/sys_user.go

package models

import (
	"IotAdmin/common/models"

	"golang.org/x/crypto/bcrypt"
	"gorm.io/gorm"
)

type SysUser struct {
	UserId   int       `gorm:"primaryKey;autoIncrement;comment:编码"  json:"userId"`
	Username string    `json:"userName" gorm:"size:64;comment:用户名"`
	Password string    `json:"-" gorm:"size:128;comment:密码"`
	NickName string    `json:"nickName" gorm:"size:128;comment:昵称"`
	Phone    string    `json:"phone" gorm:"size:11;comment:手机号"`
	RoleId   int       `json:"roleId" gorm:"size:20;comment:角色ID"`
	Salt     string    `json:"-" gorm:"size:255;comment:加盐"`
	Avatar   string    `json:"avatar" gorm:"size:255;comment:头像"`
	Sex      string    `json:"sex" gorm:"size:255;comment:性别"`
	Email    string    `json:"email" gorm:"size:128;comment:邮箱"`
	OrgId    int       `json:"orgId" gorm:"size:20;comment:部门"`
	PostId   int       `json:"postId" gorm:"size:20;comment:岗位"`
	Remark   string    `json:"remark" gorm:"size:255;comment:备注"`
	Status   string    `json:"status" gorm:"size:4;comment:状态"`
	OrgIds   []int     `json:"orgIds" gorm:"-"`
	PostIds  []int     `json:"postIds" gorm:"-"`
	RoleIds  []int     `json:"roleIds" gorm:"-"`
	Roles    []SysRole `json:"roles" gorm:"-"`
	Posts    []SysPost `json:"posts" gorm:"-"`
	Org      *SysOrg   `json:"org"`
	models.ControlBy
	models.ModelTime
}

func (*SysUser) TableName() string {
	return "sys_user"
}

func (e *SysUser) Generate() models.ActiveRecord {
	o := *e
	return &o
}

func (e *SysUser) GetId() interface{} {
	return e.UserId
}

// Encrypt 加密
func (e *SysUser) Encrypt() (err error) {
	if e.Password == "" {
		return
	}

	var hash []byte
	if hash, err = bcrypt.GenerateFromPassword([]byte(e.Password), bcrypt.DefaultCost); err != nil {
		return
	} else {
		e.Password = string(hash)
		return
	}
}

func (e *SysUser) BeforeCreate(_ *gorm.DB) error {
	return e.Encrypt()
}

func (e *SysUser) BeforeUpdate(_ *gorm.DB) error {
	var err error
	if e.Password != "" {
		err = e.Encrypt()
	}
	return err
}

func (e *SysUser) AfterFind(_ *gorm.DB) error {
	e.OrgIds = []int{e.OrgId}
	e.PostIds = []int{e.PostId}
	e.RoleIds = []int{e.RoleId}
	return nil
}