iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mpk/common/utils/GenUtils.java

@@ -21,81 +21,6 @@

@Slf4j
public class GenUtils {
    public static List<String> getTemplates(){
        List<String> templates = new ArrayList<String>();
        templates.add("abstract.java.vm");
        templates.add("impl.java.vm");
        templates.add("cpp.vm");
        templates.add("Jni.cpp.vm");
        templates.add("h.vm");
        templates.add("Jni.h.vm");
        return templates;
    }
    /**
     * 生成代码
     */
    public static void generatorCode(MpkFileDTO entity, ZipOutputStream zip){
        //设置velocity资源加载器
        Properties prop = new Properties();
        prop.put("file.resource.loader.class", "org.apache.velocity.runtime.resource.loader.ClasspathResourceLoader");
        Velocity.init(prop);

        //封装模板数据
        Map<String, Object> map = new HashMap<>();
        map.put("pkgName",entity.getPkgName());
        map.put("modelMethods",entity.getModelMethods());
        map.put("pyName",entity.getPyName());
        map.put("pyModule",entity.getPyModule());

        VelocityContext context = new VelocityContext(map);

        //获取模板列表
        List<String> templates = getTemplates();
        for(String template : templates){
            //渲染模板
            StringWriter sw = drawTemplate(template,context);

            try {
                //添加到zip
                zip.putNextEntry(new ZipEntry(getFileName(template, entity.getPyName(),entity.getPkgName(),entity.getPyModule())));
                IOUtils.write(sw.toString(), zip, "UTF-8");
                IOUtils.closeQuietly(sw);
                zip.closeEntry();
            } catch (IOException e) {
                log.error("渲染模板失败，模型名称：" + entity.getPyName(), e);
                throw new RuntimeException("渲染模板失败，模型名称：" + entity.getPyName(), e);
            }
        }
    }

    /**
     * 获取文件名
     */
    public static String getFileName(String template, String moduleName, String pkgName, String pyModule) {
        // java
        if (template.equals("abstract.java.vm")) {
            return pkgName.replace(".", File.separator) + File.separator + moduleName + ".java";
        }
        if (template.equals("impl.java.vm")) {
            return pkgName.replace(".", File.separator) + File.separator + MdkConstant.IMPL + File.separator + moduleName + "Impl.java";
        }
        // c++
        if (template.equals("cpp.vm")) {
            return pyModule.replace(".", File.separator) + File.separator + moduleName + ".cpp";
        }
        if (template.equals("h.vm")) {
            return pyModule.replace(".", File.separator) + File.separator + moduleName + ".h";
        }
        // Jni c++
        if (template.equals("Jni.cpp.vm")) {
            return pyModule.replace(".", File.separator) + File.separator + MdkConstant.JNI + File.separator + moduleName + "Jni.cpp";
        }
        if (template.equals("Jni.h.vm")) {
            return pyModule.replace(".", File.separator) + File.separator + MdkConstant.JNI + File.separator + moduleName + "Jni.h";
        }
        return null;
    }

    /**
     * 渲染模板
     **/
@@ -116,21 +41,25 @@
        return writer;
    }

    public static void drawTemplate(String template,Map<String, Object> map,File toFile) throws IOException {
    public static void drawTemplate(String template,Map<String, Object> map,File toFile) {
        VelocityContext context = new VelocityContext(map);
        drawTemplate(template,context,toFile);
    }

    public static void drawTemplate(String template,VelocityContext context,File toFile) throws IOException {
        StringWriter writer = drawTemplate(template,context);
    public static void drawTemplate(String template,VelocityContext context,File toFile) {
        try {
            StringWriter writer = drawTemplate(template,context);

        FileUtil.mkParentDirs(toFile);
            FileUtil.mkParentDirs(toFile);

        if (!toFile.exists()) {
            toFile.createNewFile();
            if (!toFile.exists()) {
                toFile.createNewFile();
            }

            FileUtil.writeUtf8String(writer.toString(),toFile);
            IOUtils.closeQuietly(writer);
        } catch (IOException e) {
            throw new RuntimeException("模板渲染异常");
        }

        FileUtil.writeUtf8String(writer.toString(),toFile);
        IOUtils.closeQuietly(writer);
    }
}