GenCodeGauss

liulin

2024-09-03 bcb3de6093b6638f80bcf9bfceb08865fa8e255b

 src/test/java/com/lunhan/xxx/GenCodeGauss.java

@@ -77,7 +77,7 @@
    /**
     * po中getter/setter生成方式 传统=1,lombok注解=2
     */
    private static final Integer SET_PO_Getter = 1;
    private static final Integer SET_PO_Getter = 2;
    //endregion

    //region 类头部文字设置
@@ -119,18 +119,18 @@

    //region vo类 import 内容设置
    private static final String[] SET_VO_IMPORT = new String[]{
            "import com.lunhan.xxx.common.util.LocalDateTimeUtil;",
            "import com.lunhan.xxx.common.util.NumericUtil;",
            "import com.lunhan.farming.common.util.LocalDateTimeUtil;",
            "import com.lunhan.farming.common.util.NumericUtil;",
            "import java.util.Objects;"
    };
    //endregion

    //region 实体属性映射类 import 内容设置
    private static final String[] SET_CONVERT_IMPORT = new String[]{
            "import org.mapstruct.Mapper;" +
                    "import org.mapstruct.factory.Mappers;" +
                    "" +
                    "import java.util.List;"
            "import org.mapstruct.Mapper;",
            "import org.mapstruct.factory.Mappers;",
            "",
            "import java.util.List;"
    };
    //endregion

@@ -138,12 +138,12 @@
    private static final String[] SET_MAPPER_IMPL_IMPORT = new String[]{
            "import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;",
            "import com.baomidou.mybatisplus.extension.plugins.pagination.Page;",
            "import com.lunhan.xxx.common.PagerResult;",
            "import com.lunhan.xxx.common.enums.EYesOrNo;",
            "import com.lunhan.xxx.common.util.*;",
            "import com.lunhan.xxx.entity.dto.*;",
            "import com.lunhan.xxx.entity.enums.*;",
            "import com.lunhan.xxx.repository.BasicMapperImpl;",
            "import com.lunhan.farming.common.PagerResult;",
            "import com.lunhan.farming.common.enums.EYesOrNo;",
            "import com.lunhan.farming.common.util.*;",
            "import com.lunhan.farming.entity.dto.*;",
            "import com.lunhan.farming.entity.enums.*;",
            "import com.lunhan.farming.repository.BasicMapperImpl;",
            "import org.springframework.stereotype.Repository;",
            "",
            "import java.util.List;"
@@ -152,10 +152,10 @@

    //region service类 import 内容设置
    private static final String[] SET_Service_IMPORT = new String[]{
            "import com.lunhan.xxx.common.*;",
            "import com.lunhan.xxx.common.enums.*;",
            "import com.lunhan.xxx.common.model.Tuple;",
            "import com.lunhan.xxx.common.util.*;",
            "import com.lunhan.farming.common.*;",
            "import com.lunhan.farming.common.enums.*;",
            "import com.lunhan.farming.common.model.Tuple;",
            "import com.lunhan.farming.common.util.*;",
            "import org.apache.commons.lang3.BooleanUtils;",
            "import org.springframework.beans.factory.annotation.Autowired;",
            "import org.springframework.stereotype.Service;",
@@ -166,13 +166,13 @@

    //region Controller类 import 内容设置
    private static final String[] SET_Controller_IMPORT = new String[]{
            "import com.lunhan.xxx.common.ExecutedResult;",
            "import com.lunhan.xxx.common.PagerResult;",
            "import com.lunhan.xxx.common.util.ParameterUtil;",
            "import com.lunhan.xxx.common.validator.ParameterValidateResult;",
            "import com.lunhan.xxx.common.validator.ParameterValidator;",
            "import com.lunhan.xxx.entity.request.ReqListId;",
            "import com.lunhan.xxx.host.BasicController;",
            "import com.lunhan.farming.common.ExecutedResult;",
            "import com.lunhan.farming.common.PagerResult;",
            "import com.lunhan.farming.common.util.ParameterUtil;",
            "import com.lunhan.farming.common.validator.ParameterValidateResult;",
            "import com.lunhan.farming.common.validator.ParameterValidator;",
            "import com.lunhan.farming.entity.request.ReqListId;",
            "import com.lunhan.farming.host.BasicController;",
            "import org.springframework.beans.factory.annotation.Autowired;",
            "import org.springframework.web.bind.annotation.*;",
            "import java.util.List;"
@@ -181,7 +181,7 @@

    //region search类 import 内容设置
    private static final String[] SET_SEARCH_IMPORT = new String[]{
            "import " + GenCodeGauss.PackageSet.SEARCH.replaceAll("\\.search", "") + ".SearchBasicDTO;",
            "import " + GenCodeGauss.PackageSet.SEARCH.replaceAll("\\.search", "") + ".dto.SearchBasicDTO;",
    };
    //endregion

@@ -192,13 +192,11 @@
            "/**\n" +
                    " * {#=desc}\n" +
                    " * @author {#=author}\n" +
                    " * @description {#=desc}\n" +
                    " */";
    private static final String SET_METHOD_DESC_TPL =
            "    /**\n" +
                    "     * {#=desc}\n" +
                    "     * @author {#=author}\n" +
                    "     * @description {#=desc}\n" +
                    "     */";
    private static final String SET_PRO_DESC_TPL =
            "\t/**\n" +
@@ -334,8 +332,12 @@
                    listColumn.add("(type = IdType.ASSIGN_ID)");
                }
            }
            if ("is_delete".equals(name)) {
                String columnAnnotation = "\t@TableLogic";
                listColumn.add(columnAnnotation);
            }
            if (!"id".equals(name)) {
                String columnAnnotation = "\t@TableField(\"" + name + "\"";
                String columnAnnotation = "\t@TableField(value = \"" + name + "\"";
                //忽略最后修改时间这一列的新增和编辑
//            if (UPDATE_TIME_COLUMNS.stream().anyMatch(c -> c.equalsIgnoreCase(name))) {
//                columnAnnotation += ", insertable = false, updatable = false";
@@ -659,7 +661,7 @@
                "\t\t\tqueryWrapper.orderByDesc(" + poName + "::getId);\n" +
                "\t\t}\n" +
                "\t\tPage<" + poName + "> pageResult = super.selectPage(new Page<>(search.getPage(), search.getLimit()), queryWrapper);\n" +
                "\t\treturn new PagerResult<>(pageResult.getSize(), pageResult.getPages(), pageResult.getTotal(), pageResult.getRecords());\n" +
                "\t\treturn new PagerResult<>(pageResult.getSize(), pageResult.getCurrent(), pageResult.getTotal(), pageResult.getRecords());\n" +
                "\t}\n" +
                "\n" +
                "\tpublic Boolean add(" + poName + " item) {\n" +
@@ -969,6 +971,7 @@
        String modifyDTOName = GenCodeGauss.SuffixSet.Modify + modelName;
        String responseDTOName = modelName + "VO";
        String searchName = "Search" + modelName;
        String convertMapperName = modelName + SuffixSet.CONVERT_MAPPER;
        String tableNamePackage = tableName.toLowerCase().replaceAll("_", "");

        StringBuilder content = new StringBuilder();
@@ -985,29 +988,13 @@
                content.append(ConstantFactory.STR_NEWLINE);
            }
        }
        content.append("import ");
        content.append(GenCodeGauss.PackageSet.MAPPER_IMPL);
        content.append(".");
        content.append(daoName + ";\n");
        content.append("import ");
        content.append(GenCodeGauss.PackageSet.PO);
        content.append(".");
        content.append(poName + ";\n");
        content.append("import ");
        content.append(GenCodeGauss.PackageSet.Request);
        content.append("." + tableNamePackage + ".");
        content.append(createDTOName + ";\n");
        content.append("import ");
        content.append(GenCodeGauss.PackageSet.Request);
        content.append("." + tableNamePackage + ".");
        content.append(modifyDTOName + ";\n");
        content.append("import ");
        content.append(GenCodeGauss.PackageSet.SEARCH + ".");
        content.append(searchName + ";\n");
        content.append("import ");
        content.append(GenCodeGauss.PackageSet.VO);
        content.append(".");
        content.append(responseDTOName + ";\n");
        content.append(String.format("import %s.%s;\n", GenCodeGauss.PackageSet.MAPPER_IMPL, daoName));
        content.append(String.format("import %s.%s;\n", GenCodeGauss.PackageSet.PO, poName));
        content.append(String.format("import %s.%s.%s;\n", GenCodeGauss.PackageSet.Request, tableNamePackage, createDTOName));
        content.append(String.format("import %s.%s.%s;\n", GenCodeGauss.PackageSet.Request, tableNamePackage, modifyDTOName));
        content.append(String.format("import %s.%s;\n", GenCodeGauss.PackageSet.SEARCH, searchName));
        content.append(String.format("import %s.%s;\n", GenCodeGauss.PackageSet.VO, responseDTOName));
        content.append(String.format("import %s.%s;\n", GenCodeGauss.PackageSet.CONVERT_MAPPER, convertMapperName));
        content.append(ConstantFactory.STR_NEWLINE);
        content.append(SET_CLASS_DESC_TPL
                .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
@@ -1024,7 +1011,7 @@
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("    public ExecutedResult<Long> create(" + createDTOName + " request) {\n" +
                "        // 转换po\n" +
                "        " + poName + " item = CopierUtil.mapTo(request, " + poName + ".class);\n" +
                "        " + poName + " item = " + convertMapperName + ".INSTANCE.toCreate(request);\n" +
                "        // 设置状态\n" +
                "        //item.setStatus(EState.NORMAL.getValue());\n" +
                "        // 设置记录创建时间\n" +
@@ -1046,7 +1033,7 @@
                "            return ExecutedResult.failed(checkExists.getMsg());\n" +
                "        }\n" +
                "        // 转换po\n" +
                "        " + poName + " item = CopierUtil.mapTo(request, " + poName + ".class);\n" +
                "        " + poName + " item = " + convertMapperName + ".INSTANCE.toModify(request);\n" +
                "\n" +
                "        int rowCount = mapper.updateById(item);\n" +
                "        if (rowCount != 1) {\n" +
@@ -1060,7 +1047,7 @@
                "\n" +
                "        " + poName + " find = mapper.get(id);\n" +
                "        if (null != find) {\n" +
                "            result = CopierUtil.mapTo(find, " + responseDTOName + ".class);\n" +
                "            result = " + convertMapperName + ".INSTANCE.toVo(find);\n" +
                "        }\n" +
                "        return ExecutedResult.success(result);\n" +
                "    }\n" +
@@ -1092,7 +1079,7 @@
                "//        item.setId(id);\n" +
                "//        item.setStatus(EState.NORMAL.getValue());\n" +
                "//\n" +
                "//        int rowCount = mapper.updateById(item);\\n\" +\n" +
                "//        int rowCount = mapper.updateById(item);\n" +
                "//        if (rowCount != 1) {\n" +
                "//            return ExecutedResult.failed(\"启用[" + tableDesc + "]失败。\");\n" +
                "//        }\n" +
@@ -1109,7 +1096,7 @@
                "//        item.setId(request.getId());\n" +
                "//        item.setSort(request.getSort());\n" +
                "//\n" +
                "//        int rowCount = mapper.updateById(item);\\n\" +\n" +
                "//        int rowCount = mapper.updateById(item);\n" +
                "//        if (rowCount != 1) {\n" +
                "//            return ExecutedResult.failed(\"设置[" + tableDesc + "]排序值失败。\");\n" +
                "//        }\n" +
@@ -1161,7 +1148,7 @@
                "\n" +
                "        List<" + poName + "> list = mapper.getList(listId);\n" +
                "        if (ListUtil.isNotNullOrEmpty(list)) {\n" +
                "            result = CopierUtil.mapTo(list, " + responseDTOName + ".class);\n" +
                "            result = " + convertMapperName + ".INSTANCE.toVo(list);\n" +
                "        }\n" +
                "        return ExecutedResult.success(result);\n" +
                "    }\n" +
@@ -1182,7 +1169,7 @@
                "        if (ListUtil.isNotNullOrEmpty(list)) {\n" +
                "            pageList.setLastId(list.get(list.size() - 1).getId());\n" +
                "            // 转换vo\n" +
                "            listVo = CopierUtil.mapTo(list, " + responseDTOName + ".class);\n" +
                "            listVo = TroopsInfoConvert.INSTANCE.toVo(list);\n" +
                "        }\n" +
                "        PagerResult<" + responseDTOName + "> result = new PagerResult<>(pageList.getLimit(), pageList.getPage(), pageList.getTotal(), listVo);\n" +
                "        result.setLastId(pageList.getLastId());\n" +
@@ -1233,7 +1220,7 @@
        String modifyDTOName = GenCodeGauss.SuffixSet.Modify + modelName;
        String responseDTOName = modelName + "VO";
        String searchName = "Search" + modelName;
        String tableNamePackage = tableName.toLowerCase().replaceAll("_", "");
        String tableNamePackage = tableName.toLowerCase().replaceAll("\\_", "");

        StringBuilder content = new StringBuilder();
        content.append(SET_BEFORE_TEXT);
@@ -1270,8 +1257,8 @@
        content.append(responseDTOName + ";\n");
        content.append(ConstantFactory.STR_NEWLINE);
        content.append(SET_CLASS_DESC_TPL
                .replaceAll("\\{\\#\\=desc\\}", tableDesc == null ? modelName : tableDesc)
                .replaceAll("\\{\\#\\=author\\}", AUTHOR)
                .replaceAll("\\{\\#\\=desc\\}", "9000." + (tableDesc == null ? modelName : tableDesc))
                .replaceAll("\\{\\#\\=author\\}", AUTHOR + "\n * @order 9000")
                .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
        );
        content.append(ConstantFactory.STR_NEWLINE);
@@ -1303,7 +1290,7 @@
                "            return failed(result.getErrorMsg());\n" +
                "        }\n" +
                "        //#endregion\n" +
                "        return this.service.create(request);\n" +
                "        return service.create(request);\n" +
                "    }\n" +
                "\n");
        content.append(SET_METHOD_DESC_TPL
@@ -1328,7 +1315,7 @@
                "            return failed(result.getErrorMsg());\n" +
                "        }\n" +
                "        //#endregion\n" +
                "        return this.service.modify(request);\n" +
                "        return service.modify(request);\n" +
                "    }\n\n");
        content.append(SET_METHOD_DESC_TPL
                .replaceAll("\\{\\#\\=desc\\}", "获取[" + tableDesc + "]")
@@ -1336,19 +1323,19 @@
                .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
        );
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("    @GetMapping(value = \"get\")\n" +
                "    public ExecutedResult<" + voName + "> get(@RequestParam Long id) {\n" +
                "        return this.service.get(id);\n" +
        content.append("    @GetMapping(value = \"get/{id}\")\n" +
                "    public ExecutedResult<" + voName + "> get(@PathVariable Long id) {\n" +
                "        return service.get(id);\n" +
                "    }\n\n");
        content.append(SET_METHOD_DESC_TPL
                .replaceAll("\\{\\#\\=desc\\}", "根据id批量获取[" + tableDesc + "]")
                .replaceAll("\\{\\#\\=desc\\}", "查询[" + tableDesc + "]")
                .replaceAll("\\{\\#\\=author\\}", AUTHOR)
                .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
        );
        content.append(ConstantFactory.STR_NEWLINE);
        content.append("    @PostMapping(value = \"search\")\n" +
                "    public ExecutedResult<PagerResult<" + voName + ">> search(@RequestBody " + searchName + " request) {\n" +
                "        return this.service.search(request);\n" +
                "        return service.search(request);\n" +
                "    }\n");
        content.append("}");