fix: 用户权限等问题 (#4)

Author: eryajf

Dockerfile

@@ -12,6 +12,6 @@ FROM centos:centos7
 RUN mkdir /app
 WORKDIR /app
 COPY --from=builder /app/ .
-RUN chmod +x wait xirang && yum -y install vim net-tools telnet wget curl && yum clean all
+RUN chmod +x wait xirang && yum -y install vim net-tools telnet wget curl sqlite && yum clean all
 
 CMD ./wait && ./xirang

controller/menu_controller.go

@@ -17,6 +17,14 @@ func (m *MenuController) GetTree(c *gin.Context) {
 	})
 }
 
+// GetUserMenuTreeByUserId 获取用户菜单树
+func (m *MenuController) GetAccessTree(c *gin.Context) {
+	req := new(request.MenuGetAccessTreeReq)
+	Run(c, req, func() (interface{}, interface{}) {
+		return logic.Menu.GetAccessTree(c, req)
+	})
+}
+
 // Add 新建
 func (m *MenuController) Add(c *gin.Context) {
 	req := new(request.MenuAddReq)

logic/menu_logic.go

@@ -178,3 +178,34 @@ func (l MenuLogic) GetTree(c *gin.Context, req interface{}) (data interface{}, r
 
 	return tree, nil
 }
+
+// GetAccessTree 获取用户菜单树
+func (l MenuLogic) GetAccessTree(c *gin.Context, req interface{}) (data interface{}, rspError interface{}) {
+	r, ok := req.(*request.MenuGetAccessTreeReq)
+	if !ok {
+		return nil, ReqAssertErr
+	}
+	_ = c
+	// 校验
+	filter := tools.H{"id": r.ID}
+	if !isql.User.Exist(filter) {
+		return nil, tools.NewValidatorError(fmt.Errorf("该用户不存在"))
+	}
+	user := new(model.User)
+	err := isql.User.Find(filter, user)
+	if err != nil {
+		return nil, tools.NewMySqlError(fmt.Errorf("在MySQL查询用户失败: " + err.Error()))
+	}
+	var roleIds []uint
+	for _, role := range user.Roles {
+		roleIds = append(roleIds, role.ID)
+	}
+	menus, err := isql.Menu.ListUserMenus(roleIds)
+	if err != nil {
+		return nil, tools.NewMySqlError(fmt.Errorf("获取资源列表失败: " + err.Error()))
+	}
+
+	tree := isql.GenMenuTree(0, menus)
+
+	return tree, nil
+}

model/request/menu_req.go

@@ -49,3 +49,8 @@ type MenuDeleteReq struct {
 // MenuGetTreeReq 获取菜单树结构体
 type MenuGetTreeReq struct {
 }
+
+// MenuGetAccessTreeReq 获取用户菜单树
+type MenuGetAccessTreeReq struct {
+	ID uint `json:"id" form:"id"`
+}

public/common/database.go

@@ -45,6 +45,12 @@ func ConnSqlite() *gorm.DB {
 	if err != nil {
 		Log.Panicf("failed to connect sqlite3: %v", err)
 	}
+	dbObj, err := db.DB()
+	if err != nil {
+		Log.Panicf("failed to get sqlite3 obj: %v", err)
+	}
+	// 参见： https://github.com/glebarez/sqlite/issues/52
+	dbObj.SetMaxOpenConns(1)
 	return db
 }
 

public/common/init_mysql_data.go

@@ -451,6 +451,13 @@ func InitData() {
 			Remark:   "获取菜单树",
 			Creator:  "系统",
 		},
+		{
+			Method:   "GET",
+			Path:     "/menu/access/tree",
+			Category: "menu",
+			Remark:   "获取用户菜单树",
+			Creator:  "系统",
+		},
 		{
 			Method:   "POST",
 			Path:     "/menu/add",
@@ -547,19 +554,8 @@ func InitData() {
 				"/base/changePwd",
 				"/base/dashboard",
 				"/user/info",
-				"/user/list",
 				"/user/changePwd",
-				"/group/list",
-				"/group/tree",
-				"/group/useringroup",
-				"/group/usernoingroup",
-				"/role/list",
-				"/role/getmenulist",
-				"/role/getapilist",
-				"/menu/tree",
-				"/menu/list",
-				"/api/list",
-				"/api/tree",
+				"/menu/access/tree",
 				"/log/operation/list",
 			}
 

routes/menu_routes.go

@@ -16,7 +16,7 @@ func InitMenuRoutes(r *gin.RouterGroup, authMiddleware *jwt.GinJWTMiddleware) gi
 	menu.Use(middleware.CasbinMiddleware())
 	{
 		menu.GET("/tree", controller.Menu.GetTree)
-		// menu.GET("/list", controller.Menu.List)
+		menu.GET("/access/tree", controller.Menu.GetAccessTree)
 		menu.POST("/add", controller.Menu.Add)
 		menu.POST("/update", controller.Menu.Update)
 		menu.POST("/delete", controller.Menu.Delete)