fix

songhuangfeng123

2022-09-20 f1f506172a3edff8d31e4db77b8940f115660919

 safePlatfrom-out-web/src/main/java/com/gkhy/safePlatform/accountController/MenuController.java

@@ -1,11 +1,14 @@
package com.gkhy.safePlatform.accountController;

import com.gkhy.safePlatform.account.rpc.apimodel.UserAccountService;
import com.alibaba.fastjson.JSONObject;
import com.gkhy.safePlatform.account.rpc.apimodel.AccountMenuService;
import com.gkhy.safePlatform.account.rpc.apimodel.model.req.MenuAddRPCReqDTO;
import com.gkhy.safePlatform.account.rpc.apimodel.model.resp.MenuModRPCReqDTO;
import com.gkhy.safePlatform.commons.co.ContextCacheUser;
import com.gkhy.safePlatform.commons.enums.ResultCodes;
import com.gkhy.safePlatform.commons.vo.ResultVO;
import org.apache.dubbo.config.annotation.DubboReference;
import org.springframework.security.core.Authentication;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
@@ -18,17 +21,16 @@
public class MenuController {

    @DubboReference(check = false)
    private UserAccountService userAccountService;
    private AccountMenuService accountMenuService;


    /**
     * @Description: 新增菜单
     */
    @RequestMapping(value = "/add",method = RequestMethod.POST)
    public ResultVO<String> addMenu(Principal principal, @RequestBody MenuAddRPCReqDTO menuAddDto) {
        String userId = principal.getName();
        userAccountService.addMenu(Long.valueOf(userId), menuAddDto);
        return new ResultVO<>(ResultCodes.OK);
    public ResultVO<String> addMenu(Authentication authentication, @RequestBody MenuAddRPCReqDTO menuAddDto) {
        ContextCacheUser currentUser = (ContextCacheUser) authentication.getPrincipal();
        return accountMenuService.addMenu(currentUser, menuAddDto);
    }


@@ -36,10 +38,20 @@
     * @Description: 新增菜单
     */
    @RequestMapping(value = "/mod",method = RequestMethod.POST)
    public ResultVO<String> addMenu(Principal principal, @RequestBody MenuModRPCReqDTO menuModDto) {
        String userId = principal.getName();
        userAccountService.modMenu(Long.valueOf(userId), menuModDto);
        return new ResultVO<>(ResultCodes.OK);
    public ResultVO<String> addMenu(Authentication authentication, @RequestBody MenuModRPCReqDTO menuModDto) {
        ContextCacheUser currentUser = (ContextCacheUser) authentication.getPrincipal();
        return accountMenuService.modMenu(currentUser, menuModDto);
    }


    /**
     * @Description: 新增菜单
     */
    @RequestMapping(value = "/del",method = RequestMethod.POST)
    public ResultVO<String> delMenu(Authentication authentication,@RequestBody JSONObject json ) {
        ContextCacheUser currentUser = (ContextCacheUser) authentication.getPrincipal();
        Long id = json.getLong("id");
        return accountMenuService.delMenu(currentUser, id);
    }

}