更新结构
This commit is contained in:
parent
e830896d47
commit
106abeeff6
@ -7,6 +7,7 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"server/models"
|
"server/models"
|
||||||
|
"server/services"
|
||||||
|
|
||||||
beego "github.com/beego/beego/v2/server/web"
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
)
|
)
|
||||||
@ -60,7 +61,7 @@ func toAdminUserDTO(u models.AdminUser) adminUserDTO {
|
|||||||
// GetAllUsers 获取全部平台管理员用户
|
// GetAllUsers 获取全部平台管理员用户
|
||||||
// GET /platform/getAllUsers
|
// GET /platform/getAllUsers
|
||||||
func (c *PlatformAdminUserController) GetAllUsers() {
|
func (c *PlatformAdminUserController) GetAllUsers() {
|
||||||
rows, total, err := models.ListAdminUsers()
|
rows, total, err := services.ListAdminUsers()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "查询失败"}
|
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "查询失败"}
|
||||||
_ = c.ServeJSON()
|
_ = c.ServeJSON()
|
||||||
@ -88,7 +89,7 @@ func (c *PlatformAdminUserController) GetUserInfo() {
|
|||||||
_ = c.ServeJSON()
|
_ = c.ServeJSON()
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
u, err := models.GetAdminUserByID(id)
|
u, err := services.GetAdminUserByID(id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.Data["json"] = map[string]interface{}{"code": 404, "msg": "用户不存在"}
|
c.Data["json"] = map[string]interface{}{"code": 404, "msg": "用户不存在"}
|
||||||
_ = c.ServeJSON()
|
_ = c.ServeJSON()
|
||||||
@ -151,7 +152,7 @@ func (c *PlatformAdminUserController) AddUser() {
|
|||||||
roleID = *p.Rid
|
roleID = *p.Rid
|
||||||
}
|
}
|
||||||
|
|
||||||
id, err := models.CreateAdminUser(p.Account, p.Password, p.Name, p.Phone, p.Email, p.Qq, p.Avatar, sex, roleID, status)
|
id, err := services.CreateAdminUser(p.Account, p.Password, p.Name, p.Phone, p.Email, p.Qq, p.Avatar, sex, roleID, status)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "添加失败"}
|
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "添加失败"}
|
||||||
_ = c.ServeJSON()
|
_ = c.ServeJSON()
|
||||||
@ -230,14 +231,14 @@ func (c *PlatformAdminUserController) EditUser() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if len(fields) > 0 {
|
if len(fields) > 0 {
|
||||||
if err := models.UpdateAdminUser(id, fields); err != nil {
|
if err := services.UpdateAdminUser(id, fields); err != nil {
|
||||||
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "编辑失败"}
|
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "编辑失败"}
|
||||||
_ = c.ServeJSON()
|
_ = c.ServeJSON()
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if p.Password != nil && strings.TrimSpace(*p.Password) != "" {
|
if p.Password != nil && strings.TrimSpace(*p.Password) != "" {
|
||||||
if err := models.ChangeAdminUserPassword(id, *p.Password); err != nil {
|
if err := services.ChangeAdminUserPassword(id, *p.Password); err != nil {
|
||||||
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "密码修改失败"}
|
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "密码修改失败"}
|
||||||
_ = c.ServeJSON()
|
_ = c.ServeJSON()
|
||||||
return
|
return
|
||||||
@ -258,7 +259,7 @@ func (c *PlatformAdminUserController) DeleteUser() {
|
|||||||
_ = c.ServeJSON()
|
_ = c.ServeJSON()
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if err := models.DeleteAdminUser(id); err != nil {
|
if err := services.DeleteAdminUser(id); err != nil {
|
||||||
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "删除失败"}
|
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "删除失败"}
|
||||||
_ = c.ServeJSON()
|
_ = c.ServeJSON()
|
||||||
return
|
return
|
||||||
@ -292,7 +293,7 @@ func (c *PlatformAdminUserController) ChangePassword() {
|
|||||||
_ = c.ServeJSON()
|
_ = c.ServeJSON()
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if err := models.ChangeAdminUserPassword(p.ID, p.Password); err != nil {
|
if err := services.ChangeAdminUserPassword(p.ID, p.Password); err != nil {
|
||||||
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "修改失败"}
|
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "修改失败"}
|
||||||
_ = c.ServeJSON()
|
_ = c.ServeJSON()
|
||||||
return
|
return
|
||||||
|
|||||||
@ -12,6 +12,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type platformLoginRequest struct {
|
type platformLoginRequest struct {
|
||||||
|
TenantName string `json:"tenant_name"`
|
||||||
Account string `json:"account"`
|
Account string `json:"account"`
|
||||||
Password string `json:"password"`
|
Password string `json:"password"`
|
||||||
}
|
}
|
||||||
@ -45,17 +46,17 @@ func (c *PlatformAuthController) Login() {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if req.Account == "" || req.Password == "" {
|
if req.TenantName == "" || req.Account == "" || req.Password == "" {
|
||||||
c.Data["json"] = map[string]interface{}{
|
c.Data["json"] = map[string]interface{}{
|
||||||
"code": 400,
|
"code": 400,
|
||||||
"msg": "用户名或密码不能为空",
|
"msg": "租户名称、用户名或密码不能为空",
|
||||||
}
|
}
|
||||||
_ = c.ServeJSON()
|
_ = c.ServeJSON()
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 控制器只做 HTTP 解析与响应编排,业务逻辑放 services 层
|
// 控制器只做 HTTP 解析与响应编排,业务逻辑放 services 层
|
||||||
token, loginUser, err := services.PlatformLogin(req.Account, req.Password)
|
token, loginUser, err := services.PlatformLogin(req.TenantName, req.Account, req.Password)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.Data["json"] = map[string]interface{}{
|
c.Data["json"] = map[string]interface{}{
|
||||||
"code": 401,
|
"code": 401,
|
||||||
@ -74,6 +75,7 @@ func (c *PlatformAuthController) Login() {
|
|||||||
"id": loginUser.ID,
|
"id": loginUser.ID,
|
||||||
"account": loginUser.Account,
|
"account": loginUser.Account,
|
||||||
"name": loginUser.Name,
|
"name": loginUser.Name,
|
||||||
|
"tid": loginUser.Tid,
|
||||||
"rid": loginUser.Rid,
|
"rid": loginUser.Rid,
|
||||||
"avatar": loginUser.Avatar,
|
"avatar": loginUser.Avatar,
|
||||||
"role_name": loginUser.RoleName,
|
"role_name": loginUser.RoleName,
|
||||||
@ -190,6 +192,24 @@ func (c *PlatformAuthController) GetOpenVerify() {
|
|||||||
_ = c.ServeJSON()
|
_ = c.ServeJSON()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Register 注册(占位实现)
|
||||||
|
func (c *PlatformAuthController) Register() {
|
||||||
|
c.Data["json"] = map[string]interface{}{
|
||||||
|
"code": 501,
|
||||||
|
"msg": "注册暂未实现",
|
||||||
|
}
|
||||||
|
_ = c.ServeJSON()
|
||||||
|
}
|
||||||
|
|
||||||
|
// SendRegisterCode 发送注册验证码(占位实现)
|
||||||
|
func (c *PlatformAuthController) SendRegisterCode() {
|
||||||
|
c.Data["json"] = map[string]interface{}{
|
||||||
|
"code": 501,
|
||||||
|
"msg": "发送注册验证码暂未实现",
|
||||||
|
}
|
||||||
|
_ = c.ServeJSON()
|
||||||
|
}
|
||||||
|
|
||||||
// ResetPassword 忘记密码重置(占位实现)
|
// ResetPassword 忘记密码重置(占位实现)
|
||||||
func (c *PlatformAuthController) ResetPassword() {
|
func (c *PlatformAuthController) ResetPassword() {
|
||||||
c.Data["json"] = map[string]interface{}{
|
c.Data["json"] = map[string]interface{}{
|
||||||
|
|||||||
@ -10,6 +10,8 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
|
|
||||||
"server/models"
|
"server/models"
|
||||||
|
"server/pkg/passwordutil"
|
||||||
|
"server/services"
|
||||||
|
|
||||||
"github.com/beego/beego/v2/client/orm"
|
"github.com/beego/beego/v2/client/orm"
|
||||||
beego "github.com/beego/beego/v2/server/web"
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
@ -166,6 +168,13 @@ func (c *PlatformTenantUserController) CreateTenantUser() {
|
|||||||
_ = c.ServeJSON()
|
_ = c.ServeJSON()
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
hashed, err := passwordutil.Hash(*p.Password)
|
||||||
|
if err != nil {
|
||||||
|
c.Data["json"] = map[string]interface{}{"code": 400, "msg": err.Error()}
|
||||||
|
_ = c.ServeJSON()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
p.Password = &hashed
|
||||||
if p.Uid == 0 {
|
if p.Uid == 0 {
|
||||||
uid, err := generateTenantUID(p.Tid)
|
uid, err := generateTenantUID(p.Tid)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -185,14 +194,14 @@ func (c *PlatformTenantUserController) CreateTenantUser() {
|
|||||||
status = *p.Status
|
status = *p.Status
|
||||||
}
|
}
|
||||||
|
|
||||||
id, err := models.BindTenantUser(p.Tid, p.Uid, p.Account, p.Name, p.Phone, p.Email, p.Password, isDefault, status, p.Remark)
|
id, err := services.BindTenantUser(p.Tid, p.Uid, p.Account, p.Name, p.Phone, p.Email, p.Password, isDefault, status, p.Remark)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "创建失败: " + err.Error()}
|
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "创建失败: " + err.Error()}
|
||||||
_ = c.ServeJSON()
|
_ = c.ServeJSON()
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if isDefault == 1 {
|
if isDefault == 1 {
|
||||||
_ = models.SetDefaultTenant(p.Uid, p.Tid)
|
_ = services.SetDefaultTenant(p.Uid, p.Tid)
|
||||||
}
|
}
|
||||||
|
|
||||||
c.Data["json"] = map[string]interface{}{"code": 200, "msg": "success", "data": map[string]interface{}{"id": id}}
|
c.Data["json"] = map[string]interface{}{"code": 200, "msg": "success", "data": map[string]interface{}{"id": id}}
|
||||||
@ -234,7 +243,13 @@ func (c *PlatformTenantUserController) EditTenantUser() {
|
|||||||
update["email"] = p.Email
|
update["email"] = p.Email
|
||||||
}
|
}
|
||||||
if p.Password != nil {
|
if p.Password != nil {
|
||||||
update["password"] = p.Password
|
hashed, err := passwordutil.Hash(*p.Password)
|
||||||
|
if err != nil {
|
||||||
|
c.Data["json"] = map[string]interface{}{"code": 400, "msg": err.Error()}
|
||||||
|
_ = c.ServeJSON()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
update["password"] = hashed
|
||||||
}
|
}
|
||||||
if p.IsDefault != nil {
|
if p.IsDefault != nil {
|
||||||
update["is_default"] = *p.IsDefault
|
update["is_default"] = *p.IsDefault
|
||||||
@ -260,7 +275,7 @@ func (c *PlatformTenantUserController) EditTenantUser() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if p.IsDefault != nil && *p.IsDefault == 1 && p.Uid > 0 && p.Tid > 0 {
|
if p.IsDefault != nil && *p.IsDefault == 1 && p.Uid > 0 && p.Tid > 0 {
|
||||||
_ = models.SetDefaultTenant(p.Uid, p.Tid)
|
_ = services.SetDefaultTenant(p.Uid, p.Tid)
|
||||||
}
|
}
|
||||||
|
|
||||||
c.Data["json"] = map[string]interface{}{"code": 200, "msg": "success"}
|
c.Data["json"] = map[string]interface{}{"code": 200, "msg": "success"}
|
||||||
@ -277,7 +292,7 @@ func (c *PlatformTenantUserController) DeleteTenantUser() {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := models.UnbindTenantUser(id); err != nil {
|
if err := services.UnbindTenantUser(id); err != nil {
|
||||||
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "删除失败: " + err.Error()}
|
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "删除失败: " + err.Error()}
|
||||||
_ = c.ServeJSON()
|
_ = c.ServeJSON()
|
||||||
return
|
return
|
||||||
|
|||||||
@ -7,7 +7,8 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"server/models"
|
"server/pkg/passwordutil"
|
||||||
|
"server/services"
|
||||||
|
|
||||||
beego "github.com/beego/beego/v2/server/web"
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
)
|
)
|
||||||
@ -62,6 +63,12 @@ func (c *PlatformUserController) AddUser() {
|
|||||||
_ = c.ServeJSON()
|
_ = c.ServeJSON()
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
hashed, err := passwordutil.Hash(p.Password)
|
||||||
|
if err != nil {
|
||||||
|
c.Data["json"] = map[string]interface{}{"code": 400, "msg": err.Error()}
|
||||||
|
_ = c.ServeJSON()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
status := int8(1)
|
status := int8(1)
|
||||||
if p.Status != nil {
|
if p.Status != nil {
|
||||||
@ -78,9 +85,10 @@ func (c *PlatformUserController) AddUser() {
|
|||||||
name := &p.Name
|
name := &p.Name
|
||||||
phone := &p.Phone
|
phone := &p.Phone
|
||||||
email := &p.Email
|
email := &p.Email
|
||||||
password := &p.Password
|
hashedPwd := hashed
|
||||||
|
password := &hashedPwd
|
||||||
|
|
||||||
_, err := models.BindTenantUser(p.Tid, uid, account, name, phone, email, password, 0, status, p.Remark)
|
_, err := services.BindTenantUser(p.Tid, uid, account, name, phone, email, password, 0, status, p.Remark)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
c.Data["json"] = map[string]interface{}{
|
c.Data["json"] = map[string]interface{}{
|
||||||
"code": 200,
|
"code": 200,
|
||||||
|
|||||||
@ -1,7 +1,7 @@
|
|||||||
package middleware
|
package middleware
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"server/models"
|
"server/services"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/beego/beego/v2/server/web/context"
|
"github.com/beego/beego/v2/server/web/context"
|
||||||
@ -62,7 +62,7 @@ func PermissionMiddleware() func(ctx *context.Context) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 检查用户是否拥有该权限
|
// 检查用户是否拥有该权限
|
||||||
hasPermission, err := models.CheckUserPermission(userId, permission)
|
hasPermission, err := services.CheckUserPermission(userId, permission)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ctx.Output.JSON(map[string]interface{}{
|
ctx.Output.JSON(map[string]interface{}{
|
||||||
"success": false,
|
"success": false,
|
||||||
|
|||||||
@ -1,17 +1,12 @@
|
|||||||
package models
|
package models
|
||||||
|
|
||||||
import (
|
import "time"
|
||||||
"crypto/md5"
|
|
||||||
"encoding/hex"
|
|
||||||
"strings"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
// AdminUser 平台管理员信息表 yz_system_admin_user
|
// AdminUser 平台管理员信息表 yz_system_admin_user
|
||||||
type AdminUser struct {
|
type AdminUser struct {
|
||||||
ID uint64 `orm:"column(id);pk;auto" json:"id"`
|
ID uint64 `orm:"column(id);pk;auto" json:"id"`
|
||||||
Account string `orm:"column(account);size(64)" json:"account"`
|
Account string `orm:"column(account);size(64)" json:"account"`
|
||||||
Password string `orm:"column(password);size(32)" json:"-"`
|
Password string `orm:"column(password);size(255)" json:"-"`
|
||||||
Name *string `orm:"column(name);size(32);null" json:"name"`
|
Name *string `orm:"column(name);size(32);null" json:"name"`
|
||||||
Phone *string `orm:"column(phone);size(18);null" json:"phone"`
|
Phone *string `orm:"column(phone);size(18);null" json:"phone"`
|
||||||
Email *string `orm:"column(email);size(255);null" json:"email"`
|
Email *string `orm:"column(email);size(255);null" json:"email"`
|
||||||
@ -30,66 +25,3 @@ type AdminUser struct {
|
|||||||
func (m *AdminUser) TableName() string {
|
func (m *AdminUser) TableName() string {
|
||||||
return "yz_system_admin_user"
|
return "yz_system_admin_user"
|
||||||
}
|
}
|
||||||
|
|
||||||
func md5Hex(s string) string {
|
|
||||||
sum := md5.Sum([]byte(s))
|
|
||||||
return hex.EncodeToString(sum[:])
|
|
||||||
}
|
|
||||||
|
|
||||||
func NormalizeAccount(s string) string {
|
|
||||||
return strings.TrimSpace(s)
|
|
||||||
}
|
|
||||||
|
|
||||||
// CreateAdminUser 创建平台管理员用户(password 会被 md5)
|
|
||||||
func CreateAdminUser(account, password string, name, phone, email, qq, avatar *string, sex uint8, roleID uint64, status uint8) (uint64, error) {
|
|
||||||
u := &AdminUser{
|
|
||||||
Account: NormalizeAccount(account),
|
|
||||||
Password: md5Hex(strings.TrimSpace(password)),
|
|
||||||
Name: name,
|
|
||||||
Phone: phone,
|
|
||||||
Email: email,
|
|
||||||
Qq: qq,
|
|
||||||
Avatar: avatar,
|
|
||||||
Sex: sex,
|
|
||||||
RoleID: roleID,
|
|
||||||
Status: status,
|
|
||||||
}
|
|
||||||
id, err := Orm.Insert(u)
|
|
||||||
return uint64(id), err
|
|
||||||
}
|
|
||||||
|
|
||||||
func GetAdminUserByID(id uint64) (*AdminUser, error) {
|
|
||||||
u := &AdminUser{ID: id}
|
|
||||||
if err := Orm.Read(u); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return u, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// UpdateAdminUser 更新用户基础信息(不含 password)
|
|
||||||
func UpdateAdminUser(id uint64, fields map[string]interface{}) error {
|
|
||||||
_, err := Orm.QueryTable(new(AdminUser)).Filter("id", id).Update(fields)
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
func DeleteAdminUser(id uint64) error {
|
|
||||||
_, err := Orm.QueryTable(new(AdminUser)).Filter("id", id).Delete()
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
func ChangeAdminUserPassword(id uint64, newPassword string) error {
|
|
||||||
_, err := Orm.QueryTable(new(AdminUser)).Filter("id", id).Update(map[string]interface{}{
|
|
||||||
"password": md5Hex(strings.TrimSpace(newPassword)),
|
|
||||||
})
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
func ListAdminUsers() ([]AdminUser, int64, error) {
|
|
||||||
var rows []AdminUser
|
|
||||||
total, err := Orm.QueryTable(new(AdminUser)).Count()
|
|
||||||
if err != nil {
|
|
||||||
return nil, 0, err
|
|
||||||
}
|
|
||||||
_, err = Orm.QueryTable(new(AdminUser)).OrderBy("-id").All(&rows)
|
|
||||||
return rows, total, err
|
|
||||||
}
|
|
||||||
|
|||||||
@ -24,82 +24,3 @@ type TenantUser struct {
|
|||||||
func (m *TenantUser) TableName() string {
|
func (m *TenantUser) TableName() string {
|
||||||
return "yz_system_tenant_user"
|
return "yz_system_tenant_user"
|
||||||
}
|
}
|
||||||
|
|
||||||
// BindTenantUser 绑定用户到租户(若已存在则更新状态/默认值)
|
|
||||||
func BindTenantUser(tid, uid uint64, account, name, phone, email, password *string, isDefault, status int8, remark *string) (uint64, error) {
|
|
||||||
var existed TenantUser
|
|
||||||
err := Orm.QueryTable(new(TenantUser)).
|
|
||||||
Filter("tid", tid).
|
|
||||||
Filter("uid", uid).
|
|
||||||
One(&existed)
|
|
||||||
if err == nil {
|
|
||||||
update := map[string]interface{}{
|
|
||||||
"account": account,
|
|
||||||
"name": name,
|
|
||||||
"phone": phone,
|
|
||||||
"email": email,
|
|
||||||
"password": password,
|
|
||||||
"status": status,
|
|
||||||
"is_default": isDefault,
|
|
||||||
"remark": remark,
|
|
||||||
}
|
|
||||||
_, uErr := Orm.QueryTable(new(TenantUser)).Filter("id", existed.ID).Update(update)
|
|
||||||
return existed.ID, uErr
|
|
||||||
}
|
|
||||||
|
|
||||||
m := &TenantUser{
|
|
||||||
Tid: tid,
|
|
||||||
Uid: uid,
|
|
||||||
Account: account,
|
|
||||||
Name: name,
|
|
||||||
Phone: phone,
|
|
||||||
Email: email,
|
|
||||||
Password: password,
|
|
||||||
IsDefault: isDefault,
|
|
||||||
Status: status,
|
|
||||||
Remark: remark,
|
|
||||||
}
|
|
||||||
id, iErr := Orm.Insert(m)
|
|
||||||
return uint64(id), iErr
|
|
||||||
}
|
|
||||||
|
|
||||||
// UnbindTenantUser 删除绑定关系
|
|
||||||
func UnbindTenantUser(id uint64) error {
|
|
||||||
_, err := Orm.QueryTable(new(TenantUser)).Filter("id", id).Delete()
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
// ListTenantUsersByTid 根据租户ID查询绑定关系
|
|
||||||
func ListTenantUsersByTid(tid uint64) ([]TenantUser, error) {
|
|
||||||
var rows []TenantUser
|
|
||||||
_, err := Orm.QueryTable(new(TenantUser)).
|
|
||||||
Filter("tid", tid).
|
|
||||||
OrderBy("-is_default", "-id").
|
|
||||||
All(&rows)
|
|
||||||
return rows, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// ListTenantBindingsByUid 根据用户ID查询绑定关系
|
|
||||||
func ListTenantBindingsByUid(uid uint64) ([]TenantUser, error) {
|
|
||||||
var rows []TenantUser
|
|
||||||
_, err := Orm.QueryTable(new(TenantUser)).
|
|
||||||
Filter("uid", uid).
|
|
||||||
OrderBy("-is_default", "-id").
|
|
||||||
All(&rows)
|
|
||||||
return rows, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// SetDefaultTenant 设置用户默认租户(同一用户仅一个默认)
|
|
||||||
func SetDefaultTenant(uid, tid uint64) error {
|
|
||||||
_, err := Orm.QueryTable(new(TenantUser)).Filter("uid", uid).Update(map[string]interface{}{
|
|
||||||
"is_default": 0,
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
_, err = Orm.QueryTable(new(TenantUser)).
|
|
||||||
Filter("uid", uid).
|
|
||||||
Filter("tid", tid).
|
|
||||||
Update(map[string]interface{}{"is_default": 1})
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|||||||
55
pkg/passwordutil/password.go
Normal file
55
pkg/passwordutil/password.go
Normal file
@ -0,0 +1,55 @@
|
|||||||
|
package passwordutil
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/rand"
|
||||||
|
"crypto/sha256"
|
||||||
|
"encoding/hex"
|
||||||
|
"errors"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
saltBytes = 16
|
||||||
|
separator = "$"
|
||||||
|
hashLength = 64 // sha256 hex length
|
||||||
|
)
|
||||||
|
|
||||||
|
// Hash 生成 salt+hash 的存储串,格式:salt$hash(均为 hex)
|
||||||
|
func Hash(plain string) (string, error) {
|
||||||
|
plain = strings.TrimSpace(plain)
|
||||||
|
if plain == "" {
|
||||||
|
return "", errors.New("password 不能为空")
|
||||||
|
}
|
||||||
|
salt := make([]byte, saltBytes)
|
||||||
|
if _, err := rand.Read(salt); err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
saltHex := hex.EncodeToString(salt)
|
||||||
|
hashHex := hashHex(saltHex, plain)
|
||||||
|
return saltHex + separator + hashHex, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify 校验存储串(salt$hash)是否匹配输入明文密码。
|
||||||
|
func Verify(stored, plain string) bool {
|
||||||
|
stored = strings.TrimSpace(stored)
|
||||||
|
plain = strings.TrimSpace(plain)
|
||||||
|
if stored == "" || plain == "" {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
parts := strings.Split(stored, separator)
|
||||||
|
if len(parts) != 2 {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
saltHex := strings.TrimSpace(parts[0])
|
||||||
|
hashHexStored := strings.TrimSpace(parts[1])
|
||||||
|
if saltHex == "" || len(hashHexStored) != hashLength {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return hashHex(saltHex, plain) == strings.ToLower(hashHexStored)
|
||||||
|
}
|
||||||
|
|
||||||
|
func hashHex(saltHex, plain string) string {
|
||||||
|
sum := sha256.Sum256([]byte(saltHex + plain))
|
||||||
|
return hex.EncodeToString(sum[:])
|
||||||
|
}
|
||||||
|
|
||||||
@ -1,7 +1,33 @@
|
|||||||
package backend
|
package backend
|
||||||
|
|
||||||
|
import (
|
||||||
|
"server/controllers"
|
||||||
|
|
||||||
|
beego "github.com/beego/beego/v2/server/web"
|
||||||
|
)
|
||||||
|
|
||||||
// Register 注册租户端(backend)路由。
|
// Register 注册租户端(backend)路由。
|
||||||
// 该端不包含平台菜单配置接口。
|
// 该端不包含平台菜单配置接口。
|
||||||
func Register() {
|
func Register() {
|
||||||
|
RegisterAuthRoutes()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// RegisterAuthRoutes 注册 backend 认证相关路由。
|
||||||
|
func RegisterAuthRoutes() {
|
||||||
|
// backend 登录相关(统一走 /backend/*)
|
||||||
|
beego.Router("/backend/login", &controllers.PlatformAuthController{}, "post:Login")
|
||||||
|
beego.Router("/backend/sendLoginCode", &controllers.PlatformAuthController{}, "post:SendLoginCode")
|
||||||
|
beego.Router("/backend/loginBySms", &controllers.PlatformAuthController{}, "post:LoginBySms")
|
||||||
|
beego.Router("/backend/logout", &controllers.PlatformAuthController{}, "post:Logout")
|
||||||
|
|
||||||
|
// 极验与登录验证配置
|
||||||
|
beego.Router("/backend/login/getGeetest3Infos", &controllers.PlatformAuthController{}, "get:GetGeetest3Infos")
|
||||||
|
beego.Router("/backend/login/getGeetest4Infos", &controllers.PlatformAuthController{}, "get:GetGeetest4Infos")
|
||||||
|
beego.Router("/backend/login/getOpenVerify", &controllers.PlatformAuthController{}, "get:GetOpenVerify")
|
||||||
|
|
||||||
|
// 注册与找回密码
|
||||||
|
beego.Router("/backend/register", &controllers.PlatformAuthController{}, "post:Register")
|
||||||
|
beego.Router("/backend/sendRegisterCode", &controllers.PlatformAuthController{}, "post:SendRegisterCode")
|
||||||
|
beego.Router("/backend/resetPassword", &controllers.PlatformAuthController{}, "post:ResetPassword")
|
||||||
|
beego.Router("/backend/sendResetCode", &controllers.PlatformAuthController{}, "post:SendResetCode")
|
||||||
|
}
|
||||||
|
|||||||
@ -48,6 +48,8 @@ func init() {
|
|||||||
switch mode {
|
switch mode {
|
||||||
case "platform":
|
case "platform":
|
||||||
platform.Register()
|
platform.Register()
|
||||||
|
// 在 platform 模式下,仍保留 backend 登录相关路由,避免后台登录 404
|
||||||
|
backend.RegisterAuthRoutes()
|
||||||
case "backend":
|
case "backend":
|
||||||
backend.Register()
|
backend.Register()
|
||||||
case "index":
|
case "index":
|
||||||
|
|||||||
73
services/admin_user.go
Normal file
73
services/admin_user.go
Normal file
@ -0,0 +1,73 @@
|
|||||||
|
package services
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"server/models"
|
||||||
|
"server/pkg/passwordutil"
|
||||||
|
)
|
||||||
|
|
||||||
|
func NormalizeAccount(s string) string {
|
||||||
|
return strings.TrimSpace(s)
|
||||||
|
}
|
||||||
|
|
||||||
|
func CreateAdminUser(account, password string, name, phone, email, qq, avatar *string, sex uint8, roleID uint64, status uint8) (uint64, error) {
|
||||||
|
hashed, err := passwordutil.Hash(password)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
u := &models.AdminUser{
|
||||||
|
Account: NormalizeAccount(account),
|
||||||
|
Password: hashed,
|
||||||
|
Name: name,
|
||||||
|
Phone: phone,
|
||||||
|
Email: email,
|
||||||
|
Qq: qq,
|
||||||
|
Avatar: avatar,
|
||||||
|
Sex: sex,
|
||||||
|
RoleID: roleID,
|
||||||
|
Status: status,
|
||||||
|
}
|
||||||
|
id, err := models.Orm.Insert(u)
|
||||||
|
return uint64(id), err
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetAdminUserByID(id uint64) (*models.AdminUser, error) {
|
||||||
|
u := &models.AdminUser{ID: id}
|
||||||
|
if err := models.Orm.Read(u); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return u, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func UpdateAdminUser(id uint64, fields map[string]interface{}) error {
|
||||||
|
_, err := models.Orm.QueryTable(new(models.AdminUser)).Filter("id", id).Update(fields)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func DeleteAdminUser(id uint64) error {
|
||||||
|
_, err := models.Orm.QueryTable(new(models.AdminUser)).Filter("id", id).Delete()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func ChangeAdminUserPassword(id uint64, newPassword string) error {
|
||||||
|
hashed, err := passwordutil.Hash(newPassword)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
_, err = models.Orm.QueryTable(new(models.AdminUser)).Filter("id", id).Update(map[string]interface{}{
|
||||||
|
"password": hashed,
|
||||||
|
})
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func ListAdminUsers() ([]models.AdminUser, int64, error) {
|
||||||
|
var rows []models.AdminUser
|
||||||
|
total, err := models.Orm.QueryTable(new(models.AdminUser)).Count()
|
||||||
|
if err != nil {
|
||||||
|
return nil, 0, err
|
||||||
|
}
|
||||||
|
_, err = models.Orm.QueryTable(new(models.AdminUser)).OrderBy("-id").All(&rows)
|
||||||
|
return rows, total, err
|
||||||
|
}
|
||||||
|
|
||||||
@ -1,8 +1,10 @@
|
|||||||
package models
|
package services
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
"server/models"
|
||||||
)
|
)
|
||||||
|
|
||||||
// CheckUserPermission 校验用户是否拥有指定权限标识。
|
// CheckUserPermission 校验用户是否拥有指定权限标识。
|
||||||
@ -12,13 +14,13 @@ func CheckUserPermission(userID int, permission string) (bool, error) {
|
|||||||
return true, nil
|
return true, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
var user AdminUser
|
var user models.AdminUser
|
||||||
if err := Orm.QueryTable(new(AdminUser)).Filter("id", userID).One(&user); err != nil {
|
if err := models.Orm.QueryTable(new(models.AdminUser)).Filter("id", userID).One(&user); err != nil {
|
||||||
return false, err
|
return false, err
|
||||||
}
|
}
|
||||||
|
|
||||||
var role AdminRole
|
var role models.AdminRole
|
||||||
if err := Orm.QueryTable(new(AdminRole)).Filter("id", user.RoleID).One(&role); err != nil {
|
if err := models.Orm.QueryTable(new(models.AdminRole)).Filter("id", user.RoleID).One(&role); err != nil {
|
||||||
return false, err
|
return false, err
|
||||||
}
|
}
|
||||||
if role.Rights == nil || strings.TrimSpace(*role.Rights) == "" {
|
if role.Rights == nil || strings.TrimSpace(*role.Rights) == "" {
|
||||||
@ -46,3 +48,4 @@ func CheckUserPermission(userID int, permission string) (bool, error) {
|
|||||||
|
|
||||||
return false, nil
|
return false, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1,19 +1,19 @@
|
|||||||
package services
|
package services
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"crypto/md5"
|
|
||||||
"encoding/hex"
|
|
||||||
"errors"
|
"errors"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"server/models"
|
"server/models"
|
||||||
"server/pkg/jwtutil"
|
"server/pkg/jwtutil"
|
||||||
|
"server/pkg/passwordutil"
|
||||||
)
|
)
|
||||||
|
|
||||||
type PlatformLoginUser struct {
|
type PlatformLoginUser struct {
|
||||||
ID uint64
|
ID uint64
|
||||||
Account string
|
Account string
|
||||||
Name string
|
Name string
|
||||||
|
Tid uint64
|
||||||
Rid uint64
|
Rid uint64
|
||||||
Avatar string
|
Avatar string
|
||||||
RoleName string
|
RoleName string
|
||||||
@ -44,53 +44,77 @@ func toPlatformLoginUser(user *models.AdminUser) *PlatformLoginUser {
|
|||||||
ID: user.ID,
|
ID: user.ID,
|
||||||
Account: user.Account,
|
Account: user.Account,
|
||||||
Name: name,
|
Name: name,
|
||||||
|
Tid: 0,
|
||||||
Rid: user.RoleID,
|
Rid: user.RoleID,
|
||||||
Avatar: avatar,
|
Avatar: avatar,
|
||||||
RoleName: adminRoleNameByID(user.RoleID),
|
RoleName: adminRoleNameByID(user.RoleID),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func md5Hex(s string) string {
|
// PlatformLogin 登录业务:先校验租户,再校验租户下用户
|
||||||
sum := md5.Sum([]byte(s))
|
func PlatformLogin(tenantName, account, password string) (string, *PlatformLoginUser, error) {
|
||||||
return hex.EncodeToString(sum[:])
|
tenantName = strings.TrimSpace(tenantName)
|
||||||
}
|
|
||||||
|
|
||||||
// PlatformLogin 平台登录业务(仅允许平台用户 yz_system_admin_user 登录)
|
|
||||||
func PlatformLogin(account, password string) (string, *PlatformLoginUser, error) {
|
|
||||||
account = strings.TrimSpace(account)
|
account = strings.TrimSpace(account)
|
||||||
password = strings.TrimSpace(password)
|
password = strings.TrimSpace(password)
|
||||||
if account == "" || password == "" {
|
if tenantName == "" || account == "" || password == "" {
|
||||||
return "", nil, errors.New("用户名或密码不能为空")
|
return "", nil, errors.New("租户名称、用户名或密码不能为空")
|
||||||
}
|
}
|
||||||
|
|
||||||
var user models.AdminUser
|
// 1) 校验租户名称
|
||||||
err := models.Orm.QueryTable(new(models.AdminUser)).
|
var tenant models.Tenant
|
||||||
|
err := models.Orm.QueryTable(new(models.Tenant)).
|
||||||
|
Filter("tenant_name", tenantName).
|
||||||
|
One(&tenant)
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, errors.New("租户不存在")
|
||||||
|
}
|
||||||
|
if tenant.Status != 1 {
|
||||||
|
return "", nil, errors.New("租户已停用")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2) 在 tid 下校验租户用户账号和密码
|
||||||
|
var tenantUser models.TenantUser
|
||||||
|
err = models.Orm.QueryTable(new(models.TenantUser)).
|
||||||
|
Filter("tid", tenant.ID).
|
||||||
Filter("account", account).
|
Filter("account", account).
|
||||||
One(&user)
|
One(&tenantUser)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", nil, errors.New("用户名或密码错误")
|
return "", nil, errors.New("用户名或密码错误")
|
||||||
}
|
}
|
||||||
if user.Password != md5Hex(password) {
|
if tenantUser.Status == 0 {
|
||||||
|
return "", nil, errors.New("账号已禁用")
|
||||||
|
}
|
||||||
|
if tenantUser.Password == nil || !passwordutil.Verify(*tenantUser.Password, password) {
|
||||||
return "", nil, errors.New("用户名或密码错误")
|
return "", nil, errors.New("用户名或密码错误")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 3) 读取用户主档用于返回资料与角色信息
|
||||||
|
var user models.AdminUser
|
||||||
|
err = models.Orm.QueryTable(new(models.AdminUser)).
|
||||||
|
Filter("id", tenantUser.Uid).
|
||||||
|
One(&user)
|
||||||
|
if err != nil {
|
||||||
|
return "", nil, errors.New("用户不存在")
|
||||||
|
}
|
||||||
if user.Status == 0 {
|
if user.Status == 0 {
|
||||||
return "", nil, errors.New("账号已禁用")
|
return "", nil, errors.New("账号已禁用")
|
||||||
}
|
}
|
||||||
const fakeTenantID = 0
|
|
||||||
|
tenantID := int(tenant.ID)
|
||||||
const userType = "platform"
|
const userType = "platform"
|
||||||
token, err := jwtutil.GenerateToken(int(user.ID), user.Account, fakeTenantID, userType)
|
token, err := jwtutil.GenerateToken(int(user.ID), user.Account, tenantID, userType)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", nil, err
|
return "", nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
loginUser := toPlatformLoginUser(&user)
|
loginUser := toPlatformLoginUser(&user)
|
||||||
|
loginUser.Tid = tenant.ID
|
||||||
return token, loginUser, nil
|
return token, loginUser, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// PlatformGetCurrentUser 根据平台管理员用户 ID 返回登录用户信息(含角色名称)
|
// PlatformGetCurrentUser 根据平台管理员用户 ID 返回登录用户信息(含角色名称)
|
||||||
func PlatformGetCurrentUser(uid uint64) (*PlatformLoginUser, error) {
|
func PlatformGetCurrentUser(uid uint64) (*PlatformLoginUser, error) {
|
||||||
u, err := models.GetAdminUserByID(uid)
|
u, err := GetAdminUserByID(uid)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, errors.New("用户不存在")
|
return nil, errors.New("用户不存在")
|
||||||
}
|
}
|
||||||
|
|||||||
83
services/tenant_user.go
Normal file
83
services/tenant_user.go
Normal file
@ -0,0 +1,83 @@
|
|||||||
|
package services
|
||||||
|
|
||||||
|
import "server/models"
|
||||||
|
|
||||||
|
// BindTenantUser 绑定用户到租户(若已存在则更新状态/默认值)
|
||||||
|
func BindTenantUser(tid, uid uint64, account, name, phone, email, password *string, isDefault, status int8, remark *string) (uint64, error) {
|
||||||
|
var existed models.TenantUser
|
||||||
|
err := models.Orm.QueryTable(new(models.TenantUser)).
|
||||||
|
Filter("tid", tid).
|
||||||
|
Filter("uid", uid).
|
||||||
|
One(&existed)
|
||||||
|
if err == nil {
|
||||||
|
update := map[string]interface{}{
|
||||||
|
"account": account,
|
||||||
|
"name": name,
|
||||||
|
"phone": phone,
|
||||||
|
"email": email,
|
||||||
|
"password": password,
|
||||||
|
"status": status,
|
||||||
|
"is_default": isDefault,
|
||||||
|
"remark": remark,
|
||||||
|
}
|
||||||
|
_, uErr := models.Orm.QueryTable(new(models.TenantUser)).Filter("id", existed.ID).Update(update)
|
||||||
|
return existed.ID, uErr
|
||||||
|
}
|
||||||
|
|
||||||
|
m := &models.TenantUser{
|
||||||
|
Tid: tid,
|
||||||
|
Uid: uid,
|
||||||
|
Account: account,
|
||||||
|
Name: name,
|
||||||
|
Phone: phone,
|
||||||
|
Email: email,
|
||||||
|
Password: password,
|
||||||
|
IsDefault: isDefault,
|
||||||
|
Status: status,
|
||||||
|
Remark: remark,
|
||||||
|
}
|
||||||
|
id, iErr := models.Orm.Insert(m)
|
||||||
|
return uint64(id), iErr
|
||||||
|
}
|
||||||
|
|
||||||
|
// UnbindTenantUser 删除绑定关系
|
||||||
|
func UnbindTenantUser(id uint64) error {
|
||||||
|
_, err := models.Orm.QueryTable(new(models.TenantUser)).Filter("id", id).Delete()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListTenantUsersByTid 根据租户ID查询绑定关系
|
||||||
|
func ListTenantUsersByTid(tid uint64) ([]models.TenantUser, error) {
|
||||||
|
var rows []models.TenantUser
|
||||||
|
_, err := models.Orm.QueryTable(new(models.TenantUser)).
|
||||||
|
Filter("tid", tid).
|
||||||
|
OrderBy("-is_default", "-id").
|
||||||
|
All(&rows)
|
||||||
|
return rows, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListTenantBindingsByUid 根据用户ID查询绑定关系
|
||||||
|
func ListTenantBindingsByUid(uid uint64) ([]models.TenantUser, error) {
|
||||||
|
var rows []models.TenantUser
|
||||||
|
_, err := models.Orm.QueryTable(new(models.TenantUser)).
|
||||||
|
Filter("uid", uid).
|
||||||
|
OrderBy("-is_default", "-id").
|
||||||
|
All(&rows)
|
||||||
|
return rows, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetDefaultTenant 设置用户默认租户(同一用户仅一个默认)
|
||||||
|
func SetDefaultTenant(uid, tid uint64) error {
|
||||||
|
_, err := models.Orm.QueryTable(new(models.TenantUser)).Filter("uid", uid).Update(map[string]interface{}{
|
||||||
|
"is_default": 0,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
_, err = models.Orm.QueryTable(new(models.TenantUser)).
|
||||||
|
Filter("uid", uid).
|
||||||
|
Filter("tid", tid).
|
||||||
|
Update(map[string]interface{}{"is_default": 1})
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
Loading…
Reference in New Issue
Block a user