user_group.go•4.46 kB
package user
import (
"github.com/duke-git/lancet/v2/slice"
"github.com/gin-gonic/gin"
"github.com/weibaohui/k8m/internal/dao"
"github.com/weibaohui/k8m/pkg/comm/utils/amis"
"github.com/weibaohui/k8m/pkg/models"
"github.com/weibaohui/k8m/pkg/service"
"gorm.io/gorm"
)
type AdminUserGroupController struct {
}
// AdminUserGroupController 用于用户组相关接口
// 路由注册函数
func RegisterAdminUserGroupRoutes(admin *gin.RouterGroup) {
ctrl := AdminUserGroupController{}
admin.GET("/user_group/list", ctrl.ListUserGroup)
admin.POST("/user_group/save", ctrl.SaveUserGroup)
admin.POST("/user_group/save_menu", ctrl.SaveUserGroupMenu)
admin.POST("/user_group/delete/:ids", ctrl.DeleteUserGroup)
admin.GET("/user_group/option_list", ctrl.GroupOptionList)
}
// @Summary 获取用户组列表
// @Description 获取所有用户组信息
// @Security BearerAuth
// @Success 200 {object} []models.UserGroup
// @Router /admin/user_group/list [get]
func (a *AdminUserGroupController) ListUserGroup(c *gin.Context) {
params := dao.BuildParams(c)
m := &models.UserGroup{}
items, total, err := m.List(params)
if err != nil {
amis.WriteJsonError(c, err)
return
}
amis.WriteJsonListWithTotal(c, total, items)
}
// @Summary 保存用户组
// @Description 新增或更新用户组信息
// @Security BearerAuth
// @Accept json
// @Param data body models.UserGroup true "用户组信息"
// @Success 200 {object} map[string]any
// @Router /admin/user_group/save [post]
func (a *AdminUserGroupController) SaveUserGroup(c *gin.Context) {
params := dao.BuildParams(c)
m := models.UserGroup{}
err := c.ShouldBindJSON(&m)
if err != nil {
amis.WriteJsonError(c, err)
return
}
err = m.Save(params)
if err != nil {
amis.WriteJsonError(c, err)
return
}
// 清除用户组的缓存
service.UserService().ClearCacheByKey(m.GroupName)
amis.WriteJsonData(c, gin.H{
"id": m.ID,
})
}
// @Summary 保存用户组菜单配置
// @Description 保存用户组的菜单配置数据
// @Security BearerAuth
// @Accept json
// @Param data body map[string]any true "菜单配置信息"
// @Success 200 {object} map[string]any
// @Router /admin/user_group/save_menu [post]
func (a *AdminUserGroupController) SaveUserGroupMenu(c *gin.Context) {
var requestData struct {
ID uint `json:"id" binding:"required"`
MenuData string `json:"menu_data"`
}
err := c.ShouldBindJSON(&requestData)
if err != nil {
amis.WriteJsonError(c, err)
return
}
params := dao.BuildParams(c)
m := &models.UserGroup{}
// 先查找用户组
userGroup, err := m.GetOne(params, func(db *gorm.DB) *gorm.DB {
return db.Where("id = ?", requestData.ID)
})
if err != nil {
amis.WriteJsonError(c, err)
return
}
// 更新菜单数据
userGroup.MenuData = requestData.MenuData
err = userGroup.Save(params)
if err != nil {
amis.WriteJsonError(c, err)
return
}
// 清除用户组的缓存
service.UserService().ClearCacheByKey(userGroup.GroupName)
amis.WriteJsonOK(c)
}
// @Summary 删除用户组
// @Description 根据ID批量删除用户组
// @Security BearerAuth
// @Param ids path string true "用户组ID，多个用逗号分隔"
// @Success 200 {object} string
// @Router /admin/user_group/delete/{ids} [post]
func (a *AdminUserGroupController) DeleteUserGroup(c *gin.Context) {
ids := c.Param("ids")
params := dao.BuildParams(c)
m := &models.UserGroup{}
err := m.Delete(params, ids)
if err != nil {
amis.WriteJsonError(c, err)
return
}
// 清除用户组的缓存
service.UserService().ClearCacheByKey(m.GroupName)
amis.WriteJsonOK(c)
}
// @Summary 用户组选项列表
// @Description 获取所有用户组的选项（仅ID和名称）
// @Security BearerAuth
// @Success 200 {object} []map[string]string
// @Router /admin/user_group/option_list [get]
func (a *AdminUserGroupController) GroupOptionList(c *gin.Context) {
params := dao.BuildParams(c)
m := &models.UserGroup{}
items, _, err := m.List(params, func(db *gorm.DB) *gorm.DB {
return db.Distinct("id,group_name")
})
if err != nil {
amis.WriteJsonData(c, gin.H{
"options": make([]map[string]string, 0),
})
return
}
var names []map[string]string
for _, n := range items {
names = append(names, map[string]string{
"label": n.GroupName,
"value": n.GroupName,
})
}
slice.SortBy(names, func(a, b map[string]string) bool {
return a["label"] < b["label"]
})
amis.WriteJsonData(c, gin.H{
"options": names,
})
}