kimi
2020-05-18 c8aee7b9bfd79cfd741d7e5692520f4f51a31a86
src/main/java/com/highdatas/mdm/controller/SysMenuController.java
@@ -3,24 +3,26 @@
import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.baomidou.mybatisplus.plugins.Page;
import com.highdatas.mdm.entity.DeletedTableLog;
import com.highdatas.mdm.entity.MenuMapping;
import com.highdatas.mdm.entity.SysField;
import com.highdatas.mdm.entity.SysMenu;
import com.highdatas.mdm.entity.*;
import com.highdatas.mdm.pojo.ActivitiBusinessType;
import com.highdatas.mdm.pojo.ActivitiStatus;
import com.highdatas.mdm.pojo.CodeMsg;
import com.highdatas.mdm.pojo.Result;
import com.highdatas.mdm.service.IDeletedTableLogService;
import com.highdatas.mdm.service.IMenuMappingService;
import com.highdatas.mdm.service.ISysFieldService;
import com.highdatas.mdm.service.ISysMenuService;
import com.highdatas.mdm.service.*;
import com.highdatas.mdm.util.Constant;
import com.highdatas.mdm.util.DbUtils;
import com.highdatas.mdm.util.NoticeClient;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import javax.servlet.http.HttpServletRequest;
import java.util.*;
import javax.servlet.http.HttpSession;
import java.util.Date;
import java.util.HashSet;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.stream.Collectors;
/**
 * <p>
@@ -42,19 +44,85 @@
    ISysFieldService fieldService;
    @Autowired
    IDeletedTableLogService deletedTableLogService;
    @RequestMapping(value = "/all", method = RequestMethod.GET)
    @Autowired
    IFlowsService flowsService;
    @Autowired
    IMaintainService maintainService;
    @Autowired
    IMaintainFieldService maintainFieldService;
    @Autowired
    IMasterAuthorService masterAuthorService;
    @Autowired
    NoticeClient noticeClient;
    @RequestMapping(value = "/all")
    public Result<List<SysMenu>> getAll() {
        EntityWrapper<SysMenu> sysMenuEntityWrapper = new EntityWrapper<>();
        sysMenuEntityWrapper.orderBy(" parent_id, order_no");
        sysMenuEntityWrapper.orderBy("parent_id, order_no");
        return Result.success(menuService.selectList(sysMenuEntityWrapper)) ;
    }
    @RequestMapping(value = "/author/all", method = RequestMethod.GET)
    public Result<List<SysMenu>> getAllBak(HttpServletRequest request) {
        TUser user = DbUtils.getUser(request);
        List<SysMenu> menu = masterAuthorService.getMenu(user);
        return Result.success(menu) ;
    }
    @RequestMapping(value = "/audit", method = RequestMethod.GET)
    public Result<List<SysMenu>> audit(HttpServletRequest request) {
        HttpSession session = request.getSession();
        TUser user = (TUser) session.getAttribute("user");
        String userId = user.getUserId();
        List<Flows> flows = flowsService.selectList(new EntityWrapper<Flows>().ne("business_type", ActivitiBusinessType.exists).ne("status", ActivitiStatus.open).ne("status", ActivitiStatus.close));
        flows = flows.stream().filter(flow ->  flowsService.isNextAudit(flow, userId)).collect(Collectors.toList());
        HashSet<String> tableNameSet = new HashSet<>();
        List<String> maintainIds = flows.stream().filter(flow -> flow.getBusinessType().equals(ActivitiBusinessType.maintain)).map(Flows::getBusinessId).collect(Collectors.toList());
        List<String> maintainFieldIds = flows.stream().filter(flow -> flow.getBusinessType().equals(ActivitiBusinessType.field)).map(Flows::getBusinessId).collect(Collectors.toList());
        if (maintainIds.isEmpty() && maintainFieldIds.isEmpty()) {
            return  Result.success(null);
        }
        for (String maintainFieldId : maintainFieldIds) {
            MaintainField maintainField = maintainFieldService.selectById(maintainFieldId);
            if (maintainField == null) {
                continue;
            }
            String tableName = maintainField.getTableName();
            if (StringUtils.isEmpty(tableName)) {
                continue;
            }
            tableNameSet.add(tableName);
        }
        maintainIds = maintainIds.stream().filter(s -> masterAuthorService.checkMaintainAuthor(user,s)).collect(Collectors.toList());
        if (!maintainIds.isEmpty()) {
            List<Maintain> maintains = maintainService.selectBatchIds(maintainIds);
            List<String> tableNameList = maintains.stream().map(Maintain::getTableName).collect(Collectors.toList());
            tableNameSet.addAll(tableNameList);
        }
        if (tableNameSet.size() == 0) {
            return Result.success(null);
        }
        List<MenuMapping> menuMappingList = menuMappingService.selectList(new EntityWrapper<MenuMapping>().in("table_name", tableNameSet));
        List<String> menuIds = menuMappingList.stream().map(MenuMapping::getMenuId).collect(Collectors.toList());
        if (menuIds.isEmpty()) {
            return Result.success(null);
        }
        LinkedHashSet<String> parentIdSet = new LinkedHashSet<>(menuIds);
        List<SysMenu> parentList = menuService.getMenuByParentId(parentIdSet);
        if (parentList == null) {
            return Result.success(null);
        }
        return Result.success(parentList);
    }
    @RequestMapping(value = "/list", method = RequestMethod.GET)
    public Result<List<SysMenu>> getMenuList() {
        EntityWrapper<SysMenu> sysMenuEntityWrapper = new EntityWrapper<>();
        sysMenuEntityWrapper.eq("DataMenu", "DataMenu");
        sysMenuEntityWrapper.eq("menu_type", "DataMenu");
        sysMenuEntityWrapper.orderBy("parent_id, order_no");
        List<SysMenu> sysMenus = menuService.selectList(sysMenuEntityWrapper);
        LinkedHashSet<String> parentIdSet = new LinkedHashSet<>();
@@ -65,13 +133,12 @@
            }
            String tableName = menuMapping.getTableName();
            List<SysField> fieldByTable = fieldService.getFieldByTable(tableName);
            if (fieldByTable == null) {
            if (fieldByTable == null || fieldByTable.size() == 0) {
                continue;
            }
            parentIdSet.add(sysMenu.getId());
        }
        Set<String> byParentId = menuService.getByParentId(parentIdSet);
        List<SysMenu> parentList = menuService.selectBatchIds(byParentId);
        List<SysMenu> parentList = menuService.getMenuByParentId(parentIdSet);
        return Result.success(parentList) ;
    }
@@ -136,10 +203,10 @@
        menu.setId(uuid);
        menu.setParentId(parentId);
        menu.setOrderNo(orderno);
        menu.setMenuType(menuType);
        menu.setMenuType(menuType).setCreateTime(new Date());
        boolean inserted = menuService.insert(menu);
        if (inserted) {
            return  Result.success("插入成功", null);
            return  Result.success(menu);
        } else {
            return Result.error(CodeMsg.INSERT_ERROR);
        }
@@ -179,7 +246,7 @@
            }else {
                orderno = Integer.valueOf(ordernoStr);
            }
            menu = new SysMenu().setId(id).setParentId(parentId).setMenuType(menuType).setName(name).setOrderNo(orderno);
            menu = new SysMenu().setCreateTime(new Date()).setId(id).setParentId(parentId).setMenuType(menuType).setName(name).setOrderNo(orderno);
            menu.insert();
        }
@@ -192,15 +259,17 @@
//            }
//        }
        return Result.success(CodeMsg.SUCCESS);
        return Result.success(menu);
    }
    @RequestMapping(value = "/update", method = RequestMethod.GET)
    public Result<Object> update(@RequestParam  String id, HttpServletRequest request) {
        SysMenu menu = menuService.selectById(id);
        if (menu == null) {
            return Result.error(CodeMsg.SELECT_ERROR_NOTFOUND);
        }
        menu.setUpdateTime(new Date());
        String ordernoStr = request.getParameter("orderNo");
        String parentId = request.getParameter("parentId");
        String menuType = request.getParameter("menuType");
@@ -241,23 +310,33 @@
        return result;
    }
    @RequestMapping(value = "/delete/{menuid}", method = RequestMethod.GET)
    public Result<Object> delete(@PathVariable String menuid) throws Exception {
    public Result<Object> delete(@PathVariable String menuid, HttpServletRequest request) throws Exception {
        TUser user = DbUtils.getUser(request);
        Result<List<SysMenu>> all = getAll();
        List<SysMenu> data = all.getData();
        SysMenu parentMenu = menuService.selectById(menuid);
        HashSet<SysMenu> sysMenus = deleteMenuByParentId(data, menuid);
        for (SysMenu sysMenu : sysMenus) {
            List<MenuMapping> menuMappingList = menuMappingService.selectList(new EntityWrapper<MenuMapping>().eq("menu_id", sysMenu.getId()));
            for (MenuMapping menuMapping : menuMappingList) {
                //todo 添加待删除表的日志
                DeletedTableLog deletedTableLog = new DeletedTableLog();
                deletedTableLog.setId(DbUtils.getUUID()).setTableName(menuMapping.getTableName()).setCreateTime(new Date());
                deletedTableLog.insert();
                menuMapping.deleteById();
            }
            sysMenu.deleteById();
        }
        boolean delete = menuService.deleteById(menuid);
        if (parentMenu == null) {
            return Result.success(null);
        }
        String menuType = parentMenu.getMenuType();
        boolean delete = parentMenu.deleteById();
        if (delete) {
            return Result.success(null);
        } else {
@@ -289,6 +368,5 @@
        List<SysMenu> sysMenus = menuService.selectList(menuEntityWrapper);
        return Result.success(sysMenus);
    }
}