实现角色接口增删改查

This commit is contained in:
2025-06-03 16:04:00 +08:00
parent bcd274c750
commit fedcd288e8
35 changed files with 596 additions and 57 deletions

View File

@ -3,7 +3,7 @@ package v1
import "github.com/gogf/gf/v2/frame/g" import "github.com/gogf/gf/v2/frame/g"
type AdminInfoReq struct { type AdminInfoReq struct {
g.Meta `path:"/admin/info" method:"get" tags:"Admin" summary:"获取管理员信息"` g.Meta `path:"/admin/info" method:"get" tags:"Admin" summary:"(系统管理员)获取管理员信息"`
} }
type AdminInfoRes struct { type AdminInfoRes struct {
g.Meta `mime:"application/json"` g.Meta `mime:"application/json"`

View File

@ -3,7 +3,7 @@ package v1
import "github.com/gogf/gf/v2/frame/g" import "github.com/gogf/gf/v2/frame/g"
type AdminLoginReq struct { type AdminLoginReq struct {
g.Meta `path:"/admin/login" method:"post" tags:"Admin" summary:"管理员登录"` g.Meta `path:"/admin/login" method:"post" tags:"Admin" summary:"(系统管理员)管理员登录"`
Username string `json:"username" v:"required" dc:"用户名"` Username string `json:"username" v:"required" dc:"用户名"`
Password string `json:"password" v:"required" dc:"密码"` Password string `json:"password" v:"required" dc:"密码"`
} }
@ -13,7 +13,7 @@ type AdminLoginRes struct {
} }
type MerchantLoginReq struct { type MerchantLoginReq struct {
g.Meta `path:"/merchant/login" method:"post" tags:"Merchant" summary:"商户登录"` g.Meta `path:"/merchant/login" method:"post" tags:"Merchant" summary:"(商户管理员)商户登录"`
Usernaem string `json:"username" v:"required" dc:"用户名"` Usernaem string `json:"username" v:"required" dc:"用户名"`
Password string `json:"password" v:"required" dc:"密码"` Password string `json:"password" v:"required" dc:"密码"`
} }
@ -27,7 +27,7 @@ type MerchantRegisterRes struct {
} }
type StoreLoginReq struct { type StoreLoginReq struct {
g.Meta `path:"/store/login" method:"post" tags:"Store" summary:"门店登录"` g.Meta `path:"/store/login" method:"post" tags:"Store" summary:"(门店管理员)门店登录"`
Username string `json:"username" v:"required" dc:"用户名"` Username string `json:"username" v:"required" dc:"用户名"`
Password string `json:"password" v:"required" dc:"密码"` Password string `json:"password" v:"required" dc:"密码"`
} }

19
api/role/role.go Normal file
View File

@ -0,0 +1,19 @@
// =================================================================================
// Code generated and maintained by GoFrame CLI tool. DO NOT EDIT.
// =================================================================================
package role
import (
"context"
"server/api/role/v1"
)
type IRoleV1 interface {
List(ctx context.Context, req *v1.ListReq) (res *v1.ListRes, err error)
Create(ctx context.Context, req *v1.CreateReq) (res *v1.CreateRes, err error)
Update(ctx context.Context, req *v1.UpdateReq) (res *v1.UpdateRes, err error)
Delete(ctx context.Context, req *v1.DeleteReq) (res *v1.DeleteRes, err error)
BatchDelete(ctx context.Context, req *v1.BatchDeleteReq) (res *v1.BatchDeleteRes, err error)
}

67
api/role/v1/role.go Normal file
View File

@ -0,0 +1,67 @@
package v1
import (
"github.com/gogf/gf/v2/frame/g"
)
// ListReq 获取角色列表请求参数
type ListReq struct {
g.Meta `path:"/role" method:"get" tags:"Role" summary:"(系统管理员)获取角色列表"`
Page int `json:"page" dc:"页数"`
Size int `json:"size" dc:"每页数量"`
Status int `json:"status" dc:"状态1=启用2=禁用"`
}
// ListRes 获取角色列表响应参数
type ListRes struct {
List interface{} `json:"list" dc:"角色列表"`
Total int `json:"total" dc:"总数"`
}
// CreateReq 创建角色请求参数
type CreateReq struct {
g.Meta `path:"/role" method:"post" tags:"Role" summary:"(系统管理员)创建角色"`
Name string `json:"name" v:"required" dc:"角色名称"`
Code string `json:"code" v:"required" dc:"角色编码"`
Description string `json:"description" dc:"角色描述"`
Status int `json:"status" v:"required|in:1,2" dc:"状态1=启用2=禁用"`
}
// CreateRes 创建角色响应参数
type CreateRes struct {
Id int64 `json:"id" dc:"角色ID"`
}
// UpdateReq 更新角色请求参数
type UpdateReq struct {
g.Meta `path:"/role" method:"put" tags:"Role" summary:"(系统管理员)更新角色"`
Id int64 `json:"id" v:"required" dc:"角色ID"`
Name string `json:"name" v:"required" dc:"角色名称"`
Code string `json:"code" v:"required" dc:"角色编码"`
Description string `json:"description" dc:"角色描述"`
Status int `json:"status" v:"required|in:1,2" dc:"状态1=启用2=禁用"`
}
// UpdateRes 更新角色响应参数
type UpdateRes struct {
Success bool `json:"success" dc:"是否成功"`
}
// DeleteReq 删除角色请求参数
type DeleteReq struct {
g.Meta `path:"/role/{id}" method:"delete" tags:"Role" summary:"(系统管理员)删除角色"`
Id int64 `json:"id" in:"path" v:"required" dc:"角色ID"`
}
// DeleteRes 删除角色响应参数
type DeleteRes struct {
Success bool `json:"success" dc:"是否成功"`
}
type BatchDeleteReq struct {
g.Meta `path:"/role" method:"delete" tags:"Role" summary:"(系统管理员)批量删除角色"`
Ids []int `json:"ids" v:"required" dc:"角色ID"`
}
type BatchDeleteRes struct {
Success bool `json:"success" dc:"是否成功"`
}

View File

@ -3,7 +3,7 @@ package v1
import "github.com/gogf/gf/v2/frame/g" import "github.com/gogf/gf/v2/frame/g"
type WeChatLoginReq struct { type WeChatLoginReq struct {
g.Meta `path:"/wechat/login" method:"post" tags:"WeChat" summary:"获取微信二维码登录"` g.Meta `path:"/wechat/login" method:"post" tags:"WeChat" summary:"(用户)获取微信二维码登录"`
SceneId string `json:"sceneId" v:"required" dc:"场景ID,规则:[门店id]_[6位随机字符串]"` SceneId string `json:"sceneId" v:"required" dc:"场景ID,规则:[门店id]_[6位随机字符串]"`
} }
type WeChatLoginRes struct { type WeChatLoginRes struct {
@ -35,7 +35,7 @@ type WeChatVertifyRes struct {
} }
type WeChatPollingReq struct { type WeChatPollingReq struct {
g.Meta `path:"/wechat/polling" method:"post" tags:"WeChat" summary:"微信长轮询"` g.Meta `path:"/wechat/polling" method:"post" tags:"WeChat" summary:"(用户)微信长轮询"`
SceneId string `json:"sceneId" v:"required" dc:"场景ID"` SceneId string `json:"sceneId" v:"required" dc:"场景ID"`
} }
type WeChatPollingRes struct { type WeChatPollingRes struct {

View File

@ -7,6 +7,7 @@ import (
"github.com/gogf/gf/v2/os/gcmd" "github.com/gogf/gf/v2/os/gcmd"
"server/internal/controller/admin" "server/internal/controller/admin"
"server/internal/controller/auth" "server/internal/controller/auth"
"server/internal/controller/role"
"server/internal/controller/wx" "server/internal/controller/wx"
"server/internal/middleware" "server/internal/middleware"
) )
@ -30,6 +31,7 @@ var (
group.Middleware(middleware.Casbin) group.Middleware(middleware.Casbin)
group.Bind( group.Bind(
admin.NewV1(), admin.NewV1(),
role.NewV1(),
) )
}) })
}) })

View File

@ -1,13 +1,8 @@
package consts package consts
const ( const (
GuestPermission = "guest" UserRoleCode = "user"
UserPermission = "user" AdminRoleCode = "admin"
AdminPermission = "admin" MerchantRoleCode = "merchant"
MerchantPermission = "merchant" StoreRoleCode = "store"
StorePermission = "store"
)
const (
QRCodeExpireTime = 60
QRCodeLimitTime = 10
) )

View File

@ -0,0 +1,5 @@
// =================================================================================
// This is auto-generated by GoFrame CLI tool only once. Fill this file as you wish.
// =================================================================================
package role

View File

@ -0,0 +1,15 @@
// =================================================================================
// This is auto-generated by GoFrame CLI tool only once. Fill this file as you wish.
// =================================================================================
package role
import (
"server/api/role"
)
type ControllerV1 struct{}
func NewV1() role.IRoleV1 {
return &ControllerV1{}
}

View File

@ -0,0 +1,19 @@
package role
import (
"context"
"server/internal/model"
"server/internal/service"
"server/api/role/v1"
)
func (c *ControllerV1) BatchDelete(ctx context.Context, req *v1.BatchDeleteReq) (res *v1.BatchDeleteRes, err error) {
out, err := service.Role().BatchDelete(ctx, &model.BatchDeleteIn{Ids: req.Ids})
if err != nil {
return nil, err
}
return &v1.BatchDeleteRes{
Success: out.Success,
}, nil
}

View File

@ -0,0 +1,17 @@
package role
import (
"context"
"server/internal/model"
"server/internal/service"
"server/api/role/v1"
)
func (c *ControllerV1) Create(ctx context.Context, req *v1.CreateReq) (res *v1.CreateRes, err error) {
out, err := service.Role().Create(ctx, &model.RoleCreateInput{Name: req.Name, Code: req.Code, Description: req.Description, Status: req.Status})
if err != nil {
return nil, err
}
return &v1.CreateRes{Id: out.Id}, nil
}

View File

@ -0,0 +1,19 @@
package role
import (
"context"
"server/internal/model"
"server/internal/service"
"server/api/role/v1"
)
func (c *ControllerV1) Delete(ctx context.Context, req *v1.DeleteReq) (res *v1.DeleteRes, err error) {
out, err := service.Role().Delete(ctx, &model.RoleDeleteInput{Id: req.Id})
if err != nil {
return nil, err
}
return &v1.DeleteRes{
Success: out.Success,
}, nil
}

View File

@ -0,0 +1,20 @@
package role
import (
"context"
"server/internal/model"
"server/internal/service"
"server/api/role/v1"
)
func (c *ControllerV1) List(ctx context.Context, req *v1.ListReq) (res *v1.ListRes, err error) {
out, err := service.Role().GetRoleList(ctx, &model.RoleListInput{Page: req.Page, Size: req.Size, Status: req.Status})
if err != nil {
return nil, err
}
return &v1.ListRes{
List: out.List,
Total: out.Total,
}, nil
}

View File

@ -0,0 +1,19 @@
package role
import (
"context"
"server/internal/model"
"server/internal/service"
"server/api/role/v1"
)
func (c *ControllerV1) Update(ctx context.Context, req *v1.UpdateReq) (res *v1.UpdateRes, err error) {
out, err := service.Role().Update(ctx, &model.RoleUpdateInput{Id: req.Id, Name: req.Name, Code: req.Code, Status: req.Status})
if err != nil {
return nil, err
}
return &v1.UpdateRes{
Success: out.Success,
}, nil
}

View File

@ -21,6 +21,7 @@ type AdminsDao struct {
// AdminsColumns defines and stores column names for the table admins. // AdminsColumns defines and stores column names for the table admins.
type AdminsColumns struct { type AdminsColumns struct {
Id string // 管理员ID Id string // 管理员ID
RoleId string // 角色ID
Username string // 管理员用户名 Username string // 管理员用户名
PasswordHash string // 密码哈希 PasswordHash string // 密码哈希
RealName string // 真实姓名 RealName string // 真实姓名
@ -35,6 +36,7 @@ type AdminsColumns struct {
// adminsColumns holds the columns for the table admins. // adminsColumns holds the columns for the table admins.
var adminsColumns = AdminsColumns{ var adminsColumns = AdminsColumns{
Id: "id", Id: "id",
RoleId: "role_id",
Username: "username", Username: "username",
PasswordHash: "password_hash", PasswordHash: "password_hash",
RealName: "real_name", RealName: "real_name",

View File

@ -28,6 +28,7 @@ type RolesColumns struct {
CreatedAt string // 创建时间 CreatedAt string // 创建时间
UpdatedAt string // 更新时间 UpdatedAt string // 更新时间
DeletedAt string // 软删除时间戳 DeletedAt string // 软删除时间戳
IsDeletable string // 是否可删除0=不可删除1=可删除
} }
// rolesColumns holds the columns for the table roles. // rolesColumns holds the columns for the table roles.
@ -40,6 +41,7 @@ var rolesColumns = RolesColumns{
CreatedAt: "created_at", CreatedAt: "created_at",
UpdatedAt: "updated_at", UpdatedAt: "updated_at",
DeletedAt: "deleted_at", DeletedAt: "deleted_at",
IsDeletable: "is_deletable",
} }
// NewRolesDao creates and returns a new DAO object for table data access. // NewRolesDao creates and returns a new DAO object for table data access.

View File

@ -35,6 +35,7 @@ type UsersColumns struct {
CreatedAt string // 创建时间 CreatedAt string // 创建时间
UpdatedAt string // 更新时间 UpdatedAt string // 更新时间
DeletedAt string // 软删除时间 DeletedAt string // 软删除时间
RoleId string // 角色ID
} }
// usersColumns holds the columns for the table users. // usersColumns holds the columns for the table users.
@ -54,6 +55,7 @@ var usersColumns = UsersColumns{
CreatedAt: "created_at", CreatedAt: "created_at",
UpdatedAt: "updated_at", UpdatedAt: "updated_at",
DeletedAt: "deleted_at", DeletedAt: "deleted_at",
RoleId: "role_id",
} }
// NewUsersDao creates and returns a new DAO object for table data access. // NewUsersDao creates and returns a new DAO object for table data access.

View File

@ -2,6 +2,7 @@ package admin
import ( import (
"context" "context"
"github.com/gogf/gf/v2/database/gdb"
"github.com/gogf/gf/v2/os/glog" "github.com/gogf/gf/v2/os/glog"
"server/internal/consts" "server/internal/consts"
"server/internal/dao" "server/internal/dao"
@ -23,20 +24,53 @@ func New() service.IAdmin {
func checkAdmin() { func checkAdmin() {
ctx := context.Background() ctx := context.Background()
exist, err := dao.Admins.Ctx(ctx).Where(do.Admins{Username: "admin"}).Exist()
if err != nil { if err := dao.Roles.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
panic("初始化管理员失败") // 检查角色是否存在
} exist, err := tx.Model(dao.Roles.Table()).Where(do.Roles{Code: consts.AdminRoleCode}).Exist()
if !exist {
passwordHash, _ := utility.EncryptPassword("Aa123456")
_, err = dao.Admins.Ctx(ctx).Insert(do.Admins{
Username: "admin",
PasswordHash: passwordHash,
Status: 1,
})
if err != nil { if err != nil {
panic("初始化管理员失败") return err
} }
var roleId int64
if !exist {
roleId, err = tx.Model(dao.Roles.Table()).InsertAndGetId(do.Roles{
Name: "系统管理员",
Code: consts.AdminRoleCode,
Description: "管理员角色",
Status: 1,
IsDeletable: false,
})
} else {
value, err := tx.Model(dao.Roles.Table()).Where(do.Roles{Code: consts.AdminRoleCode}).Fields(dao.Roles.Columns().Id).Value()
if err != nil {
return err
}
roleId = value.Int64()
}
// 检查管理员是否存在
exist, err = tx.Model(dao.Admins.Table()).Where(do.Admins{Username: "admin"}).Exist()
if err != nil {
return err
}
if !exist {
password, err := utility.EncryptPassword("Aa123456")
if err != nil {
return err
}
_, err = tx.Model(dao.Admins.Table()).InsertAndGetId(do.Admins{
Username: "admin",
PasswordHash: password,
RoleId: roleId,
Status: 1,
})
if err != nil {
return err
}
}
return nil
}); err != nil {
panic("初始化系统失败")
} }
glog.Infof(ctx, "初始化管理员成功") glog.Infof(ctx, "初始化管理员成功")
} }
@ -60,7 +94,11 @@ func (s *sAdmin) Login(ctx context.Context, in *model.AdminLoginIn) (out *model.
if !utility.ComparePassword(admin.PasswordHash, in.Password) { if !utility.ComparePassword(admin.PasswordHash, in.Password) {
return nil, ecode.Auth return nil, ecode.Auth
} }
token, err := jwt.GenerateToken(&jwt.TokenIn{UserId: admin.Id, Permission: consts.AdminPermission}) value, err := dao.Roles.Ctx(ctx).WherePri(admin.RoleId).Fields(dao.Roles.Columns().Code).Value()
if err != nil {
return nil, ecode.Fail.Sub("查询角色失败")
}
token, err := jwt.GenerateToken(&jwt.TokenIn{UserId: admin.Id, Role: value.String()})
if err != nil { if err != nil {
return nil, ecode.Fail.Sub("生成token失败") return nil, ecode.Fail.Sub("生成token失败")
} }
@ -84,6 +122,7 @@ func (s *sAdmin) Info(ctx context.Context, in *model.AdminInfoIn) (out *model.Ad
return nil, ecode.Fail.Sub("查询管理员失败") return nil, ecode.Fail.Sub("查询管理员失败")
} }
out = &model.AdminInfoOut{ out = &model.AdminInfoOut{
Id: admin.Id,
Username: admin.Username, Username: admin.Username,
} }
return return

View File

@ -7,6 +7,7 @@ package logic
import ( import (
_ "server/internal/logic/admin" _ "server/internal/logic/admin"
_ "server/internal/logic/merchantAdmin" _ "server/internal/logic/merchantAdmin"
_ "server/internal/logic/role"
_ "server/internal/logic/storeAdmin" _ "server/internal/logic/storeAdmin"
_ "server/internal/logic/user" _ "server/internal/logic/user"
) )

162
internal/logic/role/role.go Normal file
View File

@ -0,0 +1,162 @@
package role
import (
"context"
"server/internal/dao"
"server/internal/model"
"server/internal/model/do"
"server/internal/service"
"server/utility/ecode"
"github.com/gogf/gf/v2/database/gdb"
)
type sRole struct {
}
func New() service.IRole {
return &sRole{}
}
func init() {
service.RegisterRole(New())
}
func (s *sRole) Create(ctx context.Context, in *model.RoleCreateInput) (out *model.CreateOut, err error) {
// 检查角色编码是否已存在
exist, err := dao.Roles.Ctx(ctx).Where(do.Roles{Code: in.Code}).Exist()
if err != nil {
return nil, ecode.Fail.Sub("新增角色查重出现异常")
}
if exist {
return nil, ecode.Params.Sub("角色编码已存在")
}
// 创建角色
id, err := dao.Roles.Ctx(ctx).InsertAndGetId(do.Roles{
Name: in.Name,
Code: in.Code,
Description: in.Description,
Status: in.Status,
IsDeletable: true,
})
if err != nil {
return nil, ecode.Fail.Sub("创建角色失败")
}
return &model.CreateOut{Id: id}, nil
}
func (s *sRole) Delete(ctx context.Context, in *model.RoleDeleteInput) (out *model.DeleteOut, err error) {
// 检查角色是否存在且可删除
role, err := dao.Roles.Ctx(ctx).WherePri(in.Id).One()
if err != nil {
return nil, ecode.Fail.Sub("查询角色失败")
}
if role.IsEmpty() {
return nil, ecode.Params.Sub("角色不存在")
}
if !role["is_deletable"].Bool() {
return nil, ecode.Params.Sub("该角色不可删除")
}
if err = dao.Roles.Transaction(ctx, func(ctx context.Context, tx gdb.TX) (err error) {
// 删除角色
if _, err = dao.Roles.Ctx(ctx).Where(do.Roles{Id: in.Id}).Delete(); err != nil {
err = ecode.Fail.Sub("删除角色失败")
}
// TODO : 删除角色后续需要处理的操作
return
}); err != nil {
return nil, err
}
return &model.DeleteOut{Success: true}, nil
}
func (s *sRole) BatchDelete(ctx context.Context, in *model.BatchDeleteIn) (out *model.DeleteOut, err error) {
// 检查角色是否存在且可删除
roles, err := dao.Roles.Ctx(ctx).WhereIn(dao.Roles.Columns().Id, in.Ids).All()
if err != nil {
return nil, ecode.Fail.Sub("查询角色失败")
}
if len(roles) == 0 {
return nil, ecode.Params.Sub("角色不存在")
}
// 检查是否有不可删除的角色
for _, role := range roles {
if !role["is_deletable"].Bool() {
return nil, ecode.Params.Sub("存在不可删除的角色")
}
}
// 开启事务
if err = dao.Roles.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
// 删除角色
if _, err = dao.Roles.Ctx(ctx).WhereIn(dao.Roles.Columns().Id, in.Ids).Delete(); err != nil {
return ecode.Fail.Sub("删除角色失败")
}
// TODO : 删除角色后续需要处理的操作
return nil
}); err != nil {
return nil, err
}
return &model.DeleteOut{Success: true}, nil
}
func (s *sRole) Update(ctx context.Context, in *model.RoleUpdateInput) (out *model.UpdateOut, err error) {
// 检查角色是否存在
exist, err := dao.Roles.Ctx(ctx).Where(do.Roles{Id: in.Id}).Exist()
if err != nil {
return nil, ecode.Fail.Sub("查询角色失败")
}
if !exist {
return nil, ecode.Params.Sub("角色不存在")
}
// 检查角色编码是否已存在(排除自身)
exist, err = dao.Roles.Ctx(ctx).
Where(do.Roles{Code: in.Code}).
WhereNot(dao.Roles.Columns().Id, in.Id).
Exist()
if err != nil {
return nil, ecode.Fail.Sub("查询角色编码失败")
}
if exist {
return nil, ecode.Params.Sub("角色编码已存在")
}
// 更新角色
if _, err = dao.Roles.Ctx(ctx).
Where(do.Roles{Id: in.Id}).
Update(do.Roles{
Name: in.Name,
Code: in.Code,
Description: in.Description,
Status: in.Status,
}); err != nil {
return nil, ecode.Fail.Sub("更新角色失败")
}
return &model.UpdateOut{Success: true}, nil
}
func (s *sRole) GetRoleList(ctx context.Context, in *model.RoleListInput) (out *model.RoleListOutput, err error) {
list := make([]model.Role, 0)
var total int
// 构建查询条件
orm := dao.Roles.Ctx(ctx)
if in.Status > 0 {
orm = orm.Where(dao.Roles.Columns().Status, in.Status)
}
// 获取分页数据
if err = orm.Page(in.Page, in.Size).ScanAndCount(&list, &total, false); err != nil {
return nil, ecode.Fail.Sub("查询角色列表失败")
}
return &model.RoleListOutput{
List: list,
Total: total,
}, nil
}

View File

@ -19,6 +19,27 @@ type sUser struct{}
func init() { func init() {
service.RegisterUser(New()) service.RegisterUser(New())
go checkUserRole()
}
func checkUserRole() {
ctx := context.Background()
exist, err := dao.Roles.Ctx(ctx).Where(do.Roles{Code: consts.UserRoleCode}).Exist()
if err != nil {
return
}
if !exist {
_, err := dao.Roles.Ctx(ctx).Data(do.Roles{
Name: "用户",
Code: consts.UserRoleCode,
Description: "用户角色",
Status: 1,
IsDeletable: false,
}).Insert()
if err != nil {
return
}
}
} }
func New() service.IUser { func New() service.IUser {
@ -26,12 +47,15 @@ func New() service.IUser {
} }
func (s *sUser) Login(ctx context.Context, in *model.UserLoginIn) (out *model.UserLoginOut, err error) { func (s *sUser) Login(ctx context.Context, in *model.UserLoginIn) (out *model.UserLoginOut, err error) {
value, err := dao.Roles.Ctx(ctx).Where(do.Roles{Code: consts.UserRoleCode}).Fields(dao.Roles.Columns().Code).Value()
if err != nil {
return nil, ecode.Fail.Sub("查找角色失败")
}
// 根据 OpenId 查找用户 // 根据 OpenId 查找用户
exist, err := dao.Users.Ctx(ctx).Where(do.Users{WxOpenId: in.OpenId}).Exist() exist, err := dao.Users.Ctx(ctx).Where(do.Users{WxOpenId: in.OpenId}).Exist()
if err != nil { if err != nil {
return nil, ecode.Fail.Sub("查找用户失败") return nil, ecode.Fail.Sub("查找用户失败")
} }
var userId int64 var userId int64
if !exist { if !exist {
// 用户不存在,创建新用户 // 用户不存在,创建新用户
@ -62,8 +86,8 @@ func (s *sUser) Login(ctx context.Context, in *model.UserLoginIn) (out *model.Us
// 生成 token // 生成 token
token, err := jwt.GenerateToken(&jwt.TokenIn{ token, err := jwt.GenerateToken(&jwt.TokenIn{
UserId: userId, UserId: userId,
Permission: consts.UserPermission, Role: value.String(),
}) })
if err != nil { if err != nil {
return nil, ecode.Fail.Sub("生成token失败") return nil, ecode.Fail.Sub("生成token失败")

View File

@ -41,7 +41,7 @@ func Auth(r *ghttp.Request) {
Exit(r, err) Exit(r, err)
} }
r.SetCtxVar("userId", tokenOut.UserId) r.SetCtxVar("userId", tokenOut.UserId)
r.SetCtxVar("permission", tokenOut.Permission) r.SetCtxVar("role", tokenOut.Role)
r.SetCtxVar("jti", tokenOut.JTI) r.SetCtxVar("jti", tokenOut.JTI)
} }
r.Middleware.Next() r.Middleware.Next()

View File

@ -19,8 +19,8 @@ import (
// - 如果权限验证未通过终止请求返回权限不足的错误ecode.Denied // - 如果权限验证未通过终止请求返回权限不足的错误ecode.Denied
// - 如果权限验证通过:继续执行后续中间件或处理逻辑。 // - 如果权限验证通过:继续执行后续中间件或处理逻辑。
func Casbin(r *ghttp.Request) { func Casbin(r *ghttp.Request) {
permission := r.GetCtxVar("permission").String() role := r.GetCtxVar("role").String()
if !myCasbin.GetMyCasbin().HasPermission(permission, r.URL.Path, r.Method) { if !myCasbin.GetMyCasbin().HasPermission(role, r.URL.Path, r.Method) {
Exit(r, ecode.Denied) Exit(r, ecode.Denied)
} }
r.Middleware.Next() r.Middleware.Next()

View File

@ -10,6 +10,7 @@ type (
Id int Id int
} }
AdminInfoOut struct { AdminInfoOut struct {
Id int64
Username string Username string
} }
) )

View File

@ -1,5 +1,8 @@
package model package model
type BatchDeleteIn struct {
Ids []int
}
type LoginOut struct { type LoginOut struct {
Token string Token string
} }

View File

@ -13,6 +13,7 @@ import (
type Admins struct { type Admins struct {
g.Meta `orm:"table:admins, do:true"` g.Meta `orm:"table:admins, do:true"`
Id interface{} // 管理员ID Id interface{} // 管理员ID
RoleId interface{} // 角色ID
Username interface{} // 管理员用户名 Username interface{} // 管理员用户名
PasswordHash interface{} // 密码哈希 PasswordHash interface{} // 密码哈希
RealName interface{} // 真实姓名 RealName interface{} // 真实姓名

View File

@ -20,4 +20,5 @@ type Roles struct {
CreatedAt *gtime.Time // 创建时间 CreatedAt *gtime.Time // 创建时间
UpdatedAt *gtime.Time // 更新时间 UpdatedAt *gtime.Time // 更新时间
DeletedAt *gtime.Time // 软删除时间戳 DeletedAt *gtime.Time // 软删除时间戳
IsDeletable interface{} // 是否可删除0=不可删除1=可删除
} }

View File

@ -27,4 +27,5 @@ type Users struct {
CreatedAt *gtime.Time // 创建时间 CreatedAt *gtime.Time // 创建时间
UpdatedAt *gtime.Time // 更新时间 UpdatedAt *gtime.Time // 更新时间
DeletedAt *gtime.Time // 软删除时间 DeletedAt *gtime.Time // 软删除时间
RoleId interface{} // 角色ID
} }

View File

@ -11,6 +11,7 @@ import (
// Admins is the golang structure for table admins. // Admins is the golang structure for table admins.
type Admins struct { type Admins struct {
Id int64 `json:"id" orm:"id" description:"管理员ID"` // 管理员ID Id int64 `json:"id" orm:"id" description:"管理员ID"` // 管理员ID
RoleId int64 `json:"roleId" orm:"role_id" description:"角色ID"` // 角色ID
Username string `json:"username" orm:"username" description:"管理员用户名"` // 管理员用户名 Username string `json:"username" orm:"username" description:"管理员用户名"` // 管理员用户名
PasswordHash string `json:"passwordHash" orm:"password_hash" description:"密码哈希"` // 密码哈希 PasswordHash string `json:"passwordHash" orm:"password_hash" description:"密码哈希"` // 密码哈希
RealName string `json:"realName" orm:"real_name" description:"真实姓名"` // 真实姓名 RealName string `json:"realName" orm:"real_name" description:"真实姓名"` // 真实姓名

View File

@ -10,12 +10,13 @@ import (
// Roles is the golang structure for table roles. // Roles is the golang structure for table roles.
type Roles struct { type Roles struct {
Id int64 `json:"id" orm:"id" description:"角色ID"` // 角色ID Id int64 `json:"id" orm:"id" description:"角色ID"` // 角色ID
Name string `json:"name" orm:"name" description:"角色名称"` // 角色名称 Name string `json:"name" orm:"name" description:"角色名称"` // 角色名称
Code string `json:"code" orm:"code" description:"角色编码"` // 角色编码 Code string `json:"code" orm:"code" description:"角色编码"` // 角色编码
Description string `json:"description" orm:"description" description:"角色描述"` // 角色描述 Description string `json:"description" orm:"description" description:"角色描述"` // 角色描述
Status int `json:"status" orm:"status" description:"状态1=启用2=禁用"` // 状态1=启用2=禁用 Status int `json:"status" orm:"status" description:"状态1=启用2=禁用"` // 状态1=启用2=禁用
CreatedAt *gtime.Time `json:"createdAt" orm:"created_at" description:"创建时间"` // 创建时间 CreatedAt *gtime.Time `json:"createdAt" orm:"created_at" description:"创建时间"` // 创建时间
UpdatedAt *gtime.Time `json:"updatedAt" orm:"updated_at" description:"更新时间"` // 更新时间 UpdatedAt *gtime.Time `json:"updatedAt" orm:"updated_at" description:"更新时间"` // 更新时间
DeletedAt *gtime.Time `json:"deletedAt" orm:"deleted_at" description:"软删除时间戳"` // 软删除时间戳 DeletedAt *gtime.Time `json:"deletedAt" orm:"deleted_at" description:"软删除时间戳"` // 软删除时间戳
IsDeletable bool `json:"isDeletable" orm:"is_deletable" description:"是否可删除0=不可删除1=可删除"` // 是否可删除0=不可删除1=可删除
} }

View File

@ -25,4 +25,5 @@ type Users struct {
CreatedAt *gtime.Time `json:"createdAt" orm:"created_at" description:"创建时间"` // 创建时间 CreatedAt *gtime.Time `json:"createdAt" orm:"created_at" description:"创建时间"` // 创建时间
UpdatedAt *gtime.Time `json:"updatedAt" orm:"updated_at" description:"更新时间"` // 更新时间 UpdatedAt *gtime.Time `json:"updatedAt" orm:"updated_at" description:"更新时间"` // 更新时间
DeletedAt *gtime.Time `json:"deletedAt" orm:"deleted_at" description:"软删除时间"` // 软删除时间 DeletedAt *gtime.Time `json:"deletedAt" orm:"deleted_at" description:"软删除时间"` // 软删除时间
RoleId int64 `json:"roleId" orm:"role_id" description:"角色ID"` // 角色ID
} }

54
internal/model/role.go Normal file
View File

@ -0,0 +1,54 @@
package model
import "github.com/gogf/gf/v2/frame/g"
// Role 角色信息
type Role struct {
g.Meta `orm:"table:roles"`
Id int64 `json:"id" dc:"角色ID" orm:"id"`
Name string `json:"name" dc:"角色名称" orm:"name"`
Code string `json:"code" dc:"角色编码" orm:"code"`
Description string `json:"description" dc:"角色描述" orm:"description"`
Status int `json:"status" dc:"状态1=启用2=禁用" orm:"status"`
IsDeletable bool `json:"is_deletable" dc:"是否可删除0=不可删除1=可删除" orm:"is_deletable"`
}
// RoleCreateInput 创建角色输入参数
type RoleCreateInput struct {
Name string
Code string
Description string
Status int
}
// RoleUpdateInput 更新角色输入参数
type RoleUpdateInput struct {
Id int64
Name string
Code string
Description string
Status int
}
// RoleListInput 获取角色列表输入参数
type RoleListInput struct {
Page int
Size int
Status int
}
// RoleListOutput 获取角色列表输出参数
type RoleListOutput struct {
List []Role
Total int
}
// RoleGetByIdInput 根据ID获取角色输入参数
type RoleGetByIdInput struct {
Id int64
}
// RoleDeleteInput 删除角色输入参数
type RoleDeleteInput struct {
Id int64
}

36
internal/service/role.go Normal file
View File

@ -0,0 +1,36 @@
// ================================================================================
// Code generated and maintained by GoFrame CLI tool. DO NOT EDIT.
// You can delete these comments if you wish manually maintain this interface file.
// ================================================================================
package service
import (
"context"
"server/internal/model"
)
type (
IRole interface {
Create(ctx context.Context, in *model.RoleCreateInput) (out *model.CreateOut, err error)
Delete(ctx context.Context, in *model.RoleDeleteInput) (out *model.DeleteOut, err error)
BatchDelete(ctx context.Context, in *model.BatchDeleteIn) (out *model.DeleteOut, err error)
Update(ctx context.Context, in *model.RoleUpdateInput) (out *model.UpdateOut, err error)
GetRoleList(ctx context.Context, in *model.RoleListInput) (out *model.RoleListOutput, err error)
}
)
var (
localRole IRole
)
func Role() IRole {
if localRole == nil {
panic("implement not found for interface IRole, forgot register?")
}
return localRole
}
func RegisterRole(i IRole) {
localRole = i
}

View File

@ -22,11 +22,11 @@ type (
// //
// 字段: // 字段:
// - UserId: 用户 ID // - UserId: 用户 ID
// - Permission: 权限标识; // - Role: 权限标识;
// - ExpireTime: token 过期时间(可选)。 // - ExpireTime: token 过期时间(可选)。
TokenIn struct { TokenIn struct {
UserId int64 // 用户 ID UserId int64 // 用户 ID
Permission string // 权限标识 Role string // 权限标识
ExpireTime time.Duration // 令牌有效期 ExpireTime time.Duration // 令牌有效期
} }
@ -34,19 +34,19 @@ type (
// //
// 字段: // 字段:
// - UserId: 用户 ID // - UserId: 用户 ID
// - Permission: 权限标识; // - Role: 权限标识;
// - JTI: JWT 的唯一标识。 // - JTI: JWT 的唯一标识。
TokenOut struct { TokenOut struct {
UserId int64 // 用户 ID UserId int64 // 用户 ID
Permission string // 权限标识 Role string // 权限标识
JTI string // JWT 唯一标识 JTI string // JWT 唯一标识
} }
// jwtClaims 自定义 JWT 的声明体结构,嵌入标准声明字段。 // jwtClaims 自定义 JWT 的声明体结构,嵌入标准声明字段。
jwtClaims struct { jwtClaims struct {
UserId int64 `json:"user_id"` // 用户 ID UserId int64 `json:"user_id"` // 用户 ID
Permission string `json:"permission"` // 权限标识 Role string `json:"Role"` // 权限标识
JTI string `json:"jti"` // 唯一标识 JTI string `json:"jti"` // 唯一标识
jwt.RegisteredClaims jwt.RegisteredClaims
} }
) )
@ -66,9 +66,9 @@ func GenerateToken(in *TokenIn) (string, error) {
} }
claims := jwtClaims{ claims := jwtClaims{
UserId: in.UserId, UserId: in.UserId,
Permission: in.Permission, Role: in.Role,
JTI: uuid.NewString(), JTI: uuid.NewString(),
RegisteredClaims: jwt.RegisteredClaims{ RegisteredClaims: jwt.RegisteredClaims{
ExpiresAt: jwt.NewNumericDate(time.Now().Add(expire)), ExpiresAt: jwt.NewNumericDate(time.Now().Add(expire)),
IssuedAt: jwt.NewNumericDate(time.Now()), IssuedAt: jwt.NewNumericDate(time.Now()),
@ -108,8 +108,8 @@ func ParseToken(tokenString string) (*TokenOut, error) {
} }
return &TokenOut{ return &TokenOut{
UserId: claims.UserId, UserId: claims.UserId,
Permission: claims.Permission, Role: claims.Role,
JTI: claims.JTI, JTI: claims.JTI,
}, nil }, nil
} }

View File

@ -33,8 +33,18 @@ func init() {
} }
enforcer.LoadPolicy() enforcer.LoadPolicy()
// 管理员
{ {
// admin
enforcer.AddPolicy("admin", "/x/admin/info", "GET", "获取管理员用户信息") enforcer.AddPolicy("admin", "/x/admin/info", "GET", "获取管理员用户信息")
// role
enforcer.AddPolicy("admin", "/x/role", "GET", "管理员获取角色列表")
enforcer.AddPolicy("admin", "/x/role", "POST", "管理员添加角色")
enforcer.AddPolicy("admin", "/x/role", "PUT", "管理员更新角色")
enforcer.AddPolicy("admin", "/x/role", "DELETE", "管理员给角色分配权限")
enforcer.AddPolicy("admin", "/x/role/*", "DELETE", "管理员删除单个角色")
} }
instance = &myCasbin{Enforcer: enforcer} instance = &myCasbin{Enforcer: enforcer}