代码生成支持选择上级菜单

RuoYi

2020-07-30 8ae4f5e90b4d982ae43982dd5e9f832a28a9e792

 ruoyi-modules/ruoyi-gen/src/main/java/com/ruoyi/gen/service/GenTableServiceImpl.java

@@ -23,6 +23,7 @@
import com.ruoyi.common.core.constant.GenConstants;

import com.ruoyi.common.core.exception.CustomException;

import com.ruoyi.common.core.utils.StringUtils;

import com.ruoyi.common.security.utils.SecurityUtils;

import com.ruoyi.gen.domain.GenTable;

import com.ruoyi.gen.domain.GenTableColumn;

import com.ruoyi.gen.mapper.GenTableColumnMapper;

@@ -142,10 +143,10 @@
    @Transactional

    public void importGenTable(List<GenTable> tableList)

    {

        String operName = "";

        for (GenTable table : tableList)

        String operName = SecurityUtils.getUsername();

        try

        {

            try

            for (GenTable table : tableList)

            {

                String tableName = table.getTableName();

                GenUtils.initTable(table, operName);

@@ -161,10 +162,10 @@
                    }

                }

            }

            catch (Exception e)

            {

                log.error("表名 " + table.getTableName() + " 导入失败：", e);

            }

        }

        catch (Exception e)

        {

            throw new CustomException("导入失败：" + e.getMessage());

        }

    }



@@ -335,9 +336,14 @@
            String treeCode = paramsObj.getString(GenConstants.TREE_CODE);

            String treeParentCode = paramsObj.getString(GenConstants.TREE_PARENT_CODE);

            String treeName = paramsObj.getString(GenConstants.TREE_NAME);

            String parentMenuId = paramsObj.getString(GenConstants.PARENT_MENU_ID);

            String parentMenuName = paramsObj.getString(GenConstants.PARENT_MENU_NAME);

            
            genTable.setTreeCode(treeCode);

            genTable.setTreeParentCode(treeParentCode);

            genTable.setTreeName(treeName);

            genTable.setParentMenuId(parentMenuId);

            genTable.setParentMenuName(parentMenuName);

        }

    }

}