forked from gzzfw/backEnd/gz-dyh

xusd
2024-09-27 7c3418c45926b5288b22044d60dbb12fa2c982d0
dyh-service/dyh-utils/src/main/java/cn/huge/module/buildcode/service/BuildCodeService.java
@@ -63,7 +63,6 @@
     * @param packagePath 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableSchema 数据库
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
@@ -102,7 +101,6 @@
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableSchema 数据库
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
@@ -110,13 +108,16 @@
    public void buildAll(String basePackage, String packageName, String className, String tableSchema, String tableName, String author, String version) {
        Map tplData = this.getTplData(basePackage, packageName, className, tableSchema, tableName, author, version);
        try {
            // 必须
            this.buildPo(tplData);
            this.buildBo(tplData);
            this.buildDaoMapper(tplData);
            this.buildDaoMapperXml(tplData);
            this.buildService(tplData);
            this.buildController(tplData);
            this.buildJson(tplData);
            this.buildWebController(tplData);
            // 可选
//            this.buildWechatController(tplData);
//            this.buildJson(tplData);
//            this.buildJsForm(tplData);
        } catch (TemplateNotFoundException e) {
            log.error("TemplateNotFoundException");
@@ -136,7 +137,6 @@
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableSchema 数据库
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
@@ -163,7 +163,6 @@
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableSchema 数据库
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
@@ -190,7 +189,6 @@
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableSchema 数据库
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
@@ -217,7 +215,6 @@
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableSchema 数据库
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
@@ -244,7 +241,6 @@
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableSchema 数据库
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
@@ -267,19 +263,44 @@
    }
    /**
     * 生成Controller层的代码
     * 生成webController层的代码
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableSchema 数据库
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
     */
    public void buildController(String basePackage, String packageName, String className, String tableSchema, String tableName, String author, String version) {
    public void buildWebController(String basePackage, String packageName, String className, String tableSchema, String tableName, String author, String version) {
        Map tplData = this.getTplData(basePackage, packageName, className, tableSchema, tableName, author, version);
        try {
            this.buildController(tplData);
            this.buildWebController(tplData);
        } catch (TemplateNotFoundException e) {
            log.error("TemplateNotFoundException");
        } catch (TemplateException e) {
            log.error("TemplateException");
        } catch (MalformedTemplateNameException e) {
            log.error("MalformedTemplateNameException");
        } catch (ParseException e) {
            log.error("ParseException");
        } catch (IOException e) {
            log.error("IOException");
        }
    }
    /**
     * 生成wechatController层的代码
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
     */
    public void buildWechatController(String basePackage, String packageName, String className, String tableSchema, String tableName, String author, String version) {
        Map tplData = this.getTplData(basePackage, packageName, className, tableSchema, tableName, author, version);
        try {
            this.buildWechatController(tplData);
        } catch (TemplateNotFoundException e) {
            log.error("TemplateNotFoundException");
        } catch (TemplateException e) {
@@ -298,7 +319,6 @@
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableSchema 数据库
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
@@ -326,7 +346,6 @@
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableSchema 数据库
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
@@ -353,7 +372,6 @@
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableSchema 数据库
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
@@ -380,7 +398,7 @@
     * @param basePackage 包路径
     * @param packageName 包名
     * @param className 类名
     * @param tableSchema 数据库
     * @param tableName 表名
     * @param author 作者
     * @param version 版本
@@ -511,11 +529,11 @@
    }
    /**
     * 创建控制层
     * 创建web控制层
     *
     * @param tplData
     */
    private void buildController(Map tplData) throws TemplateException, IOException {
    private void buildWebController(Map tplData) throws TemplateException, IOException {
        tplData.put("packagePath", ((String) tplData.get("packageName")).toLowerCase());
        String className = (String) tplData.get("className");
        if (Character.isLowerCase(className.charAt(0))){
@@ -527,11 +545,37 @@
            tplData.put("lowerName", lowerName);
        }
        Template template = cfg.getTemplate("controller.ftl");
        Template template = cfg.getTemplate("webController.ftl");
        StringBuffer path = new StringBuffer(ROOT_MAIN_PATH)
                .append(tplData.get("basePackagePath"))
                .append(tplData.get("packageName")).append("/controller/")
                .append(tplData.get("className")).append("Controller")
                .append(tplData.get("packageName")).append("/controller/web/")
                .append(tplData.get("className")).append("WebController")
                .append(".java");
        this.buildFile(template, tplData, path.toString());
    }
    /**
     * 创建wechat控制层
     *
     * @param tplData
     */
    private void buildWechatController(Map tplData) throws TemplateException, IOException {
        tplData.put("packagePath", ((String) tplData.get("packageName")).toLowerCase());
        String className = (String) tplData.get("className");
        if (Character.isLowerCase(className.charAt(0))){
            tplData.put("classPath", className);
            tplData.put("lowerName", className);
        }else{
            String lowerName = (new StringBuilder()).append(Character.toLowerCase(className.charAt(0))).append(className.substring(1)).toString();
            tplData.put("classPath", lowerName);
            tplData.put("lowerName", lowerName);
        }
        Template template = cfg.getTemplate("wechatController.ftl");
        StringBuffer path = new StringBuffer(ROOT_MAIN_PATH)
                .append(tplData.get("basePackagePath"))
                .append(tplData.get("packageName")).append("/controller/wechat/")
                .append(tplData.get("className")).append("WechatController")
                .append(".java");
        this.buildFile(template, tplData, path.toString());
    }