io.github.wujun728.generate.controller.CodeGeneratorController Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jun-code-generator-spring-boot-starter Show documentation
There is a newer version: 1.0.20
Show newest version
package io.github.wujun728.generate.controller;

import io.github.wujun728.common.Result;
import io.github.wujun728.common.generator.ClassInfo;
import io.github.wujun728.common.generator.util.FreemarkerTool;
import io.github.wujun728.common.generator.util.StringUtils;
import io.github.wujun728.common.generator.util.TableParseUtil;
import freemarker.template.TemplateException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;

import javax.annotation.Resource;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

@Controller
public class CodeGeneratorController {
    private static final Logger logger = LoggerFactory.getLogger(CodeGeneratorController.class);

    @Resource
    private FreemarkerTool freemarkerTool;

    @RequestMapping("/")
    public String index() {
        return "index";
    }
    @RequestMapping("/generate")
    public String generate() {
        return "generate";
    }

    @RequestMapping("/codeGenerate")
    @ResponseBody
    public Result codeGenerate(String tableSql) {

        try {

            if (tableSql==null || tableSql.trim().length()==0) {
                return new Result(500, "表结构信息不可为空");
            }

            // parse table
            ClassInfo classInfo = TableParseUtil.processTableIntoClassInfo(tableSql);

            // code genarete
            Map params = new HashMap();
            params.put("classInfo", classInfo);
         // genProcessStringWriter(datas, result);
			params.put("packageController", "io.github.wujun728.biz.controller");
			params.put("packageService", "io.github.wujun728.biz.service");
			params.put("packageServiceImpl", "io.github.wujun728.biz.service.impl");
			params.put("packageDao", "io.github.wujun728.biz.dao");
			params.put("packageMybatisXML", "io.github.wujun728.biz.model");
			params.put("packageModel", "io.github.wujun728.biz.model");

            // result
            Map result = new HashMap();

            result.put("controller_code", freemarkerTool.processString("code-generator/controller.ftl", params));
            result.put("service_code", freemarkerTool.processString("code-generator/service.ftl", params));
            result.put("service_impl_code", freemarkerTool.processString("code-generator/service_impl.ftl", params));

            result.put("dao_code", freemarkerTool.processString("code-generator/dao.ftl", params));
            result.put("mybatis_code", freemarkerTool.processString("code-generator/mybatis.ftl", params));
            result.put("model_code", freemarkerTool.processString("code-generator/model.ftl", params));

            // 计算,生成代码行数
            int lineNum = 0;
            for (Map.Entry item: result.entrySet()) {
                if (item.getValue() != null) {
                    lineNum += StringUtils.countMatches(item.getValue(), "\n");
                }
            }
            logger.info("生成代码行数：{}", lineNum);

            return  Result.ok(result);
        } catch (IOException | TemplateException e) {
            logger.error(e.getMessage(), e);
            return new Result(500, "表结构解析失败");
        }

    }

}