package com.xzx.gc.user.controller;
|
|
|
import cn.hutool.core.lang.Dict;
|
import com.xzx.gc.common.Result;
|
import com.xzx.gc.common.request.BaseController;
|
import com.xzx.gc.entity.UserAuth;
|
import com.xzx.gc.model.user.UserAuthDTO;
|
import com.xzx.gc.user.dto.AuthMenuDto;
|
import com.xzx.gc.user.service.UserAuthService;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.validation.annotation.Validated;
|
import org.springframework.web.bind.annotation.*;
|
|
import javax.servlet.http.HttpServletRequest;
|
import java.util.ArrayList;
|
import java.util.Comparator;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
@RestController
|
@RequestMapping("/auth/menu")
|
@Validated
|
@Slf4j
|
@Api(tags = "功能模块")
|
public class AuthController extends BaseController {
|
|
@Autowired
|
private UserAuthService userAuthService;
|
|
@ApiOperation(value = "根据角色查询所有功能")
|
@PostMapping("")
|
public Result<List<AuthMenuDto>> find(HttpServletRequest request,@RequestBody UserAuthDTO userAuthDTO) {
|
Result<List<AuthMenuDto>> result = new Result();
|
List<AuthMenuDto> authMenuDtos=new ArrayList<>();
|
List<UserAuth> menu = userAuthService.findMenu(getUserId(request),userAuthDTO.getHomePageType());
|
//分组排序
|
Dict dict=Dict.create();
|
for (UserAuth userAuth : menu) {
|
String categoryCode = userAuth.getCategoryCode();
|
String categoryName = userAuth.getCategoryName();
|
if(dict.containsKey(categoryCode)){
|
Dict dict1= (Dict) dict.get(categoryCode);
|
List<UserAuth> list= (List<UserAuth>) dict1.get("list");
|
list.add(userAuth);
|
dict1.set("list",list);
|
dict.set(categoryCode,dict1);
|
}else{
|
List<UserAuth> res=new ArrayList<>();
|
res.add(userAuth);
|
Dict dict1=Dict.create();
|
dict1.set("list",res);
|
dict1.set("name",categoryName);
|
dict.set(categoryCode,dict1);
|
}
|
}
|
|
if(!dict.isEmpty()){
|
for (String s : dict.keySet()) {
|
Dict dict1= (Dict) dict.get(s);
|
List<UserAuth> list= (List<UserAuth>) dict1.get("list");
|
String categoryName= dict1.getStr("name");
|
//排序
|
List<UserAuth> collect = list.stream().sorted(Comparator.comparing(UserAuth::getAuthOrder)).collect(Collectors.toList());
|
AuthMenuDto authMenuDto=new AuthMenuDto();
|
authMenuDto.setCategoryName(categoryName);
|
authMenuDto.setCategoryCode(s);
|
authMenuDto.setUserAuth(collect);
|
authMenuDtos.add(authMenuDto);
|
}
|
}
|
|
result.setData(authMenuDtos);
|
return result;
|
|
}
|
|
|
}
|
