yvan/Study
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

动态menu和角色关联以及基础menu构建

Browse Source
This commit is contained in:
pixelqm
2019-09-19 23:48:45 +08:00
parent dc4700c557
commit 968b8c21cd
11 changed files with 516 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files
QMPlusServer/controller/api/menu.go
+107 -1
View File
@@ -6,6 +6,7 @@ import (
"main/controller/servers"
"main/middleware"
"main/model/dbModel"
"main/model/modelInterface"
)
// @Tags Menu
@@ -15,7 +16,6 @@ import (
// @Param data body api.RegistAndLoginStuct true "可以什么都不填"
// @Success 200 {string} json "{"success":true,"data":{},"msg":"返回成功"}"
// @Router /menu/getMenu [post]
func GetMenu(c *gin.Context) {
claims, _ := c.Get("claims")
waitUse := claims.(*middleware.CustomClaims)
@@ -26,3 +26,109 @@ func GetMenu(c *gin.Context) {
servers.ReportFormat(c, true, "获取成功", gin.H{"menus": menus})
}
}
// @Tags authority
// @Summary 分页获取基础menu列表
// @Security ApiKeyAuth
// @accept application/json
// @Produce application/json
// @Param data body modelInterface.PageInfo true "分页获取基础menu列表"
// @Success 200 {string} json "{"success":true,"data":{},"msg":"获取成功"}"
// @Router /menu/getMenuList [post]
func GetMenuList(c *gin.Context) {
var pageInfo modelInterface.PageInfo
_ = c.BindJSON(&pageInfo)
err, menuList, total := new(dbModel.BaseMenu).GetInfoList(pageInfo)
if err != nil {
servers.ReportFormat(c, false, fmt.Sprintf("获取数据失败，%v", err), gin.H{})
} else {
servers.ReportFormat(c, true, "获取数据成功", gin.H{
"menuList": menuList,
"total": total,
"page": pageInfo.Page,
"pageSize": pageInfo.PageSize,
})
}
}
// @Tags authority
// @Summary 分页获取基础menu列表
// @Security ApiKeyAuth
// @accept application/json
// @Produce application/json
// @Param data body modelInterface.PageInfo true "分页获取基础menu列表"
// @Success 200 {string} json "{"success":true,"data":{},"msg":"获取成功"}"
// @Router /menu/addBaseMenu [post]
func AddBaseMenu(c *gin.Context) {
var addMenu dbModel.BaseMenu
_ = c.BindJSON(&addMenu)
err := addMenu.AddBaseMenu()
if err != nil {
servers.ReportFormat(c, false, fmt.Sprintf("添加失败，%v", err), gin.H{})
} else {
servers.ReportFormat(c, true, fmt.Sprintf("添加成功，%v", err), gin.H{})
}
}
// @Tags Menu
// @Summary 获取用户动态路由
// @Security ApiKeyAuth
// @Produce application/json
// @Param data body api.RegistAndLoginStuct true "可以什么都不填"
// @Success 200 {string} json "{"success":true,"data":{},"msg":"返回成功"}"
// @Router /menu/getBaseMenuTree [post]
func GetBaseMenuTree(c *gin.Context) {
err, menus := new(dbModel.BaseMenu).GetBaseMenuTree()
if err != nil {
servers.ReportFormat(c, false, fmt.Sprintf("获取失败：%v", err), gin.H{"menus": menus})
} else {
servers.ReportFormat(c, true, "获取成功", gin.H{"menus": menus})
}
}
type AddMenuAuthorityInfo struct {
Menus []dbModel.BaseMenu
AuthorityId string
}
// @Tags authority
// @Summary 增加menu和角色关联关系
// @Security ApiKeyAuth
// @accept application/json
// @Produce application/json
// @Param data body modelInterface.AddMenuAuthorityInfo true "增加menu和角色关联关系"
// @Success 200 {string} json "{"success":true,"data":{},"msg":"获取成功"}"
// @Router /menu/addMenuAuthority [post]
func AddMenuAuthority(c *gin.Context) {
var addMenuAuthorityInfo AddMenuAuthorityInfo
_ = c.BindJSON(&addMenuAuthorityInfo)
err := new(dbModel.Menu).AddMenuAuthority(addMenuAuthorityInfo.Menus, addMenuAuthorityInfo.AuthorityId)
if err != nil {
servers.ReportFormat(c, false, fmt.Sprintf("添加失败，%v", err), gin.H{})
} else {
servers.ReportFormat(c, true, fmt.Sprintf("添加成功，%v", err), gin.H{})
}
}
type AuthorityIdInfo struct {
AuthorityId string
}
// @Tags authority
// @Summary 获取指定角色menu
// @Security ApiKeyAuth
// @accept application/json
// @Produce application/json
// @Param data body modelInterface.AuthorityIdInfo true "增加menu和角色关联关系"
// @Success 200 {string} json "{"success":true,"data":{},"msg":"获取成功"}"
// @Router /menu/addMenuAuthority [post]
func GetMenuAuthority(c *gin.Context) {
var authorityIdInfo AuthorityIdInfo
_ = c.BindJSON(&authorityIdInfo)
err, menus := new(dbModel.Menu).GetMenuAuthority(authorityIdInfo.AuthorityId)
if err != nil {
servers.ReportFormat(c, false, fmt.Sprintf("获取失败：%v", err), gin.H{"menus": menus})
} else {
servers.ReportFormat(c, true, "获取成功", gin.H{"menus": menus})
}
}
QMPlusServer/init/registTable/registTable.go
+1 -1
View File
@@ -7,5 +7,5 @@ import (
//注册数据库表专用
func RegistTable(db *gorm.DB) {
db.AutoMigrate(dbModel.User{}, dbModel.Authority{}, dbModel.Menu{}, dbModel.Api{})
db.AutoMigrate(dbModel.User{}, dbModel.Authority{}, dbModel.Menu{}, dbModel.Api{}, dbModel.ApiAuthority{}, dbModel.BaseMenu{})
}
QMPlusServer/model/dbModel/api.go
+8 -4
View File
@@ -9,23 +9,27 @@ import (
type Api struct {
gorm.Model `json:"-"`
AuthorityId uint `json:"-"`
Path string `json:"path"`
Description string `json:"description"`
}
func (a *Api) CreateApi() (err error) {
err = qmsql.DEFAULTDB.Create(a).Error
findOne := qmsql.DEFAULTDB.Where("path = ?", a.Path).Find(&Menu{}).Error
if findOne != nil {
} else {
err = qmsql.DEFAULTDB.Create(a).Error
}
return err
}
func (a *Api) DeleteApi() (err error) {
err = qmsql.DEFAULTDB.Where("id = ?", a.AuthorityId).Delete(a).Error
err = qmsql.DEFAULTDB.Where("path = ?", a.Path).Delete(a).Delete(&ApiAuthority{}).Error
return err
}
func (a *Api) EditApi() (err error) {
err = qmsql.DEFAULTDB.Update(a).Error
err = qmsql.DEFAULTDB.Update(a).Update(&Authority{}).Error
return err
}
QMPlusServer/model/dbModel/api_authority.go
+6
View File
@@ -0,0 +1,6 @@
package dbModel
type ApiAuthority struct {
AuthorityId string `json:"-"`
Api
}
QMPlusServer/model/dbModel/baseMenu.go
+55
View File
@@ -0,0 +1,55 @@
package dbModel
import (
"github.com/jinzhu/gorm"
"github.com/pkg/errors"
"main/controller/servers"
"main/init/qmsql"
"main/model/modelInterface"
)
type BaseMenu struct {
gorm.Model
MenuLevel uint `json:"-"`
ParentId string `json:"parentId"`
Path string `json:"path"`
Name string `json:"name"`
Hidden bool `json:"hidden"`
Component string `json:"component"`
Meta `json:"meta"`
NickName string `json:"nickName"`
Children []BaseMenu `json:"children"`
}
func (b *BaseMenu) AddBaseMenu() (err error) {
findOne := qmsql.DEFAULTDB.Where("name = ?", b.Name).Find(&BaseMenu{}).Error
if findOne != nil {
b.NickName = b.Title
err = qmsql.DEFAULTDB.Create(b).Error
} else {
err = errors.New("存在重复name，请修改name")
}
return err
}
func (b *BaseMenu) DeleteBaseMenu(name string) (err error) {
err = qmsql.DEFAULTDB.Where("name = ?", name).Delete(&b).Delete(&Menu{}).Error
return err
}
func (b *BaseMenu) UpdataBaseMenu(name string) (err error) {
err = qmsql.DEFAULTDB.Where("name = ?", name).Update(&b).Update(&Menu{}).Error
return err
}
func (b *BaseMenu) GetInfoList(info modelInterface.PageInfo) (err error, list interface{}, total int) {
// 封装分页方法 调用即可 传入 当前的结构体和分页信息
err, db, total := servers.PagingServer(b, info)
if err != nil {
return
} else {
var menuList []BaseMenu
err = db.Find(&menuList).Error
return err, menuList, total
}
}
QMPlusServer/model/dbModel/menu.go
-41
View File
@@ -1,41 +0,0 @@
package dbModel
import (
"github.com/jinzhu/gorm"
"main/init/qmsql"
)
type Menu struct {
gorm.Model `json:"-"`
MenuLevel uint `json:"-"`
AuthorityId uint `json:"-"`
ParentId uint `json:"parentId"`
Path string `json:"path"`
Name string `json:"name"`
Hidden bool `json:"hidden"`
Component string `json:"component"`
Meta `json:"meta"`
Children []Menu `json:"children"`
}
type Meta struct {
Title string `json:"title"`
Icon string `json:"icon"`
}
//获取动态路由树
func (m *Menu) GetMenuTree(authorityId float64) (err error, menus []Menu) {
err = qmsql.DEFAULTDB.Where("authority_id = ? AND parent_id = ?", authorityId, 0).Find(&menus).Error
for i := 0; i < len(menus); i++ {
err = getChildrenList(&menus[i])
}
return err, menus
}
func getChildrenList(menu *Menu) (err error) {
err = qmsql.DEFAULTDB.Where("authority_id = ? AND parent_id = ?", menu.AuthorityId, menu.ID).Find(&menu.Children).Error
for i := 0; i < len(menu.Children); i++ {
err = getChildrenList(&menu.Children[i])
}
return err
}
QMPlusServer/model/dbModel/menu_authority.go
+75
View File
@@ -0,0 +1,75 @@
package dbModel
import (
"fmt"
"main/init/qmsql"
)
type Menu struct {
BaseMenu
MenuId string `json:"menuId"`
AuthorityId string `json:"-"`
Children []Menu `json:"children"`
}
type Meta struct {
Title string `json:"title"`
Icon string `json:"icon"`
}
// 为角色增加menu树
func (m *Menu) AddMenuAuthority(menus []BaseMenu, authorityId string) (err error) {
var menu Menu
qmsql.DEFAULTDB.Where("authority_id = ? ", authorityId).Unscoped().Delete(&Menu{})
for _, v := range menus {
menu.BaseMenu = v
menu.AuthorityId = authorityId
menu.MenuId = fmt.Sprintf("%v", v.ID)
menu.ID = 0
err = qmsql.DEFAULTDB.Create(&menu).Error
if err != nil {
return err
}
}
return nil
}
// 查看当前角色树
func (m *Menu) GetMenuAuthority(authorityId string) (err error, menus []Menu) {
err = qmsql.DEFAULTDB.Where("authority_id = ?", authorityId).Find(&menus).Error
return err, menus
}
//获取动态路由树
func (m *Menu) GetMenuTree(authorityId float64) (err error, menus []Menu) {
err = qmsql.DEFAULTDB.Where("authority_id = ? AND parent_id = ?", authorityId, 0).Find(&menus).Error
for i := 0; i < len(menus); i++ {
err = getChildrenList(&menus[i])
}
return err, menus
}
func getChildrenList(menu *Menu) (err error) {
err = qmsql.DEFAULTDB.Where("authority_id = ? AND parent_id = ?", menu.AuthorityId, menu.MenuId).Find(&menu.Children).Error
for i := 0; i < len(menu.Children); i++ {
err = getChildrenList(&menu.Children[i])
}
return err
}
//获取基础路由树
func (m *BaseMenu) GetBaseMenuTree() (err error, menus []BaseMenu) {
err = qmsql.DEFAULTDB.Where(" parent_id = ?", 0).Find(&menus).Error
for i := 0; i < len(menus); i++ {
err = getBaseChildrenList(&menus[i])
}
return err, menus
}
func getBaseChildrenList(menu *BaseMenu) (err error) {
err = qmsql.DEFAULTDB.Where("parent_id = ?", menu.ID).Find(&menu.Children).Error
for i := 0; i < len(menu.Children); i++ {
err = getBaseChildrenList(&menu.Children[i])
}
return err
}
QMPlusServer/router/menu.go
+5
View File
@@ -10,5 +10,10 @@ func InitMenuRouter(Router *gin.Engine) {
MenuRouter := Router.Group("menu").Use(middleware.JWTAuth())
{
MenuRouter.POST("getMenu", api.GetMenu)
MenuRouter.POST("getMenuList", api.GetMenuList)
MenuRouter.POST("addBaseMenu", api.AddBaseMenu)
MenuRouter.POST("getBaseMenuTree", api.GetBaseMenuTree)
MenuRouter.POST("addMenuAuthority", api.AddMenuAuthority)
MenuRouter.POST("getMenuAuthority", api.GetMenuAuthority)
}
}