105 lines
3.2 KiB
Go
105 lines
3.2 KiB
Go
package model
|
|
|
|
import (
|
|
"SciencesServer/app/common/model"
|
|
"SciencesServer/serve/orm"
|
|
"SciencesServer/utils"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// SysUser 用户信息
|
|
type SysUser struct {
|
|
*model.SysUser
|
|
}
|
|
|
|
type (
|
|
// SysUserInfo 用户信息
|
|
SysUserInfo struct {
|
|
*model.SysUser
|
|
DepartmentID uint64 `json:"-"`
|
|
DepartmentName string `json:"-"`
|
|
RoleIDs string `json:"-"`
|
|
RoleNames string `json:"-"`
|
|
}
|
|
)
|
|
|
|
func (m *SysUser) ValidatePassword(password string) bool {
|
|
return utils.HashCompare([]byte(m.Password), []byte(utils.Md5String(password, m.Salt)))
|
|
}
|
|
|
|
func (m *SysUser) IsAdminUser() bool {
|
|
return m.IsAdmin == model.SysUserAdministratorForAdmin
|
|
}
|
|
|
|
func (m *SysUser) GetByAccountOrMobile(account string, tenantID uint64) (bool, error) {
|
|
db := orm.GetDB().Table(m.TableName()).
|
|
Select("id", "uuid", "name", "mobile", "password", "salt", "is_admin", "status").
|
|
//Where("tenant_id = ?", tenantID).
|
|
//Where("(email = ? OR mobile = ?)", account, account).
|
|
Where("mobile = ?", account).
|
|
Where("is_deleted = ?", model.DeleteStatusForNot)
|
|
|
|
if err := db.First(m.SysUser).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return false, nil
|
|
}
|
|
return false, err
|
|
}
|
|
return true, nil
|
|
}
|
|
|
|
func (m *SysUser) CheckByAccountOrMobile(account string, tenantID uint64) (bool, error) {
|
|
db := orm.GetDB().Table(m.TableName()).
|
|
Where("tenant_id = ?", tenantID).
|
|
Where("(account = ? OR mobile = ?)", account, account).
|
|
Where("is_deleted = ?", model.DeleteStatusForNot)
|
|
|
|
var count int64
|
|
|
|
if err := db.Count(&count).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return false, nil
|
|
}
|
|
return false, err
|
|
}
|
|
return true, nil
|
|
}
|
|
|
|
// Users 用户信息
|
|
func (m *SysUser) Users(page, pageSize int, count *int64, where ...*model.ModelWhere) ([]*SysUserInfo, error) {
|
|
db := orm.GetDB().Table(m.TableName()+" AS u").
|
|
Select("u.id", "u.uuid", "u.account", "u.avatar", "u.name", "u.mobile", "u.gender",
|
|
"u.is_admin", "u.status", "u.created_at", "d.id AS department_id", "d.name AS department_name",
|
|
"u_r.role_ids", "u_r.role_names").
|
|
Joins(fmt.Sprintf("LEFT JOIN %s AS u_d ON u.uuid = u_d.uid AND u_d.is_deleted = %d",
|
|
model.NewSysUserDepartment().TableName(), model.DeleteStatusForNot)).
|
|
Joins(fmt.Sprintf("LEFT JOIN %s AS d ON u_d.department_id = d.id",
|
|
model.NewSysDepartment().TableName())).
|
|
Joins(fmt.Sprintf("LEFT JOIN (SELECT uid, GROUP_CONCAT(r.id SEPARATOR '&&') AS role_ids, GROUP_CONCAT(r.name SEPARATOR '&&') AS role_names "+
|
|
"FROM %s AS u LEFT JOIN %s AS r ON u.role_id = r.id WHERE u.is_deleted = %d GROUP BY u.uid) u_r ON u.uuid = u_r.uid",
|
|
model.NewSysUserRole().TableName(), model.NewSysRole().TableName(), model.DeleteStatusForNot)).
|
|
Where("u.is_deleted = ?", model.DeleteStatusForNot)
|
|
|
|
if len(where) > 0 {
|
|
for _, wo := range where {
|
|
db = db.Where(wo.Condition, wo.Value)
|
|
}
|
|
}
|
|
out := make([]*SysUserInfo, 0)
|
|
|
|
if err := db.Count(count).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
if err := db.Order("u.id " + model.OrderModeToDesc).Offset((page - 1) * pageSize).Limit(pageSize).Scan(&out).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
func NewSysUser() *SysUser {
|
|
return &SysUser{SysUser: model.NewSysUser()}
|
|
}
|