jzg/platform/src/main/java/com/jzg/controller/SysMenuController.java

package com.jzg.controller;

import com.jzg.commons.core.base.BaseController;
import com.jzg.commons.core.page.HttpResult;
import com.jzg.commons.entity.dto.SysCommonFunctionDto;
import com.jzg.commons.entity.po.SysMenu;
import com.jzg.commons.entity.vo.SystemMenuVo;
import com.jzg.commons.util.StringUtils;
import com.jzg.service.SysMenuService;
import io.swagger.v3.oas.annotations.Operation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.List;
import java.util.stream.Collectors;

/**
 * 菜单控制器
 *
 * @author Yasepix
 * @date Oct 29, 2018
 */
@RestController
@RequestMapping("menu")
public class SysMenuController extends BaseController {

    @Autowired
    private SysMenuService sysMenuService;


    @PostMapping(value = "/save")
    public HttpResult save(@RequestBody SysMenu record) {
        record.setSysType("1");
        if (StringUtils.isNullOrEmpty(record.getId()) || record.getId().equalsIgnoreCase("0")) {
            return HttpResult.ok(sysMenuService.save(record));
        } else {
            return HttpResult.ok(sysMenuService.updateById(record));
        }
    }


    @PostMapping(value = "/delete")
    public HttpResult delete(@RequestBody List<SysMenu> records) {
        List<String> ids = records.stream().map(SysMenu::getId).collect(Collectors.toList());
        return HttpResult.ok(sysMenuService.removeBatchByIds(ids));
    }


    @GetMapping(value = "/findNavTree")
    public HttpResult findNavTree(@RequestParam String userName) {
        return HttpResult.ok(sysMenuService.findTree(userName, 1));
    }


    @GetMapping(value = "/findMenuTree")
    public HttpResult findMenuTree() {
        String systemCode = getSystemCode();
        return HttpResult.ok(sysMenuService.findTree(systemCode, 0));
    }

    @GetMapping(value = "/getRoleMenuTree")
    public HttpResult getRoleMenuTree(){
        String userId = getUserId();
        String systemCode = getSystemCode();
        List<SysMenu> roleMenuTree = sysMenuService.getRoleMenuTree(systemCode, userId);
        return HttpResult.ok(roleMenuTree);
    }

    @GetMapping("/common/functions")
    @Operation(summary = "常用功能")
    public HttpResult<List<SysCommonFunctionDto>> getCommonFunction() {
        String userId = getUserId();
        String systemCode = getSystemCode();
        List<SysCommonFunctionDto> sysCommonFunctionDto = sysMenuService.getCommonFunction(systemCode, userId);
        return HttpResult.ok(sysCommonFunctionDto);
    }


    @GetMapping(value = "/getAllButtons")
    public HttpResult getAllButtons(){
        String userId = getUserId();
        String systemCode = getSystemCode();
        return HttpResult.ok(sysMenuService.getAllButtons(userId,systemCode));
    }

    @GetMapping(value = "/getSystemMenu")
    public HttpResult getSystemMenu(String systemCode){
        return HttpResult.ok(sysMenuService.getSystemMenu(systemCode));
    }

    @PostMapping(value = "/saveSystemMenu")
    public HttpResult saveSystemMenu(@RequestBody SystemMenuVo systemMenuVo){
        return HttpResult.ok(sysMenuService.saveSystemMenu(systemMenuVo));
    }
}