java根据模板.vm生成文件

最新推荐文章于 2023-07-25 08:39:16 发布

Jcl-_

最新推荐文章于 2023-07-25 08:39:16 发布

阅读量750

点赞数

分类专栏：实习文章标签： java 数据库 mysql

本文链接：https://blog.csdn.net/qq_45841851/article/details/128956899

版权

实习专栏收录该内容

31 篇文章 0 订阅

订阅专栏

 @Test
    void templateDataImport() throws IOException {//传递的参数都是在DbVmConfig，DataConfigFieldVmConfig（columnName，name）中
        // 设置velocity资源加载器
        Properties prop = new Properties();
        prop.put("resource.loader.file.class", "org.apache.velocity.runtime.resource.loader.ClasspathResourceLoader");
        Velocity.init(prop);
        Velocity.init();
        Map<String, Object> map = new HashMap<>();

        //DbVmConfig中的参数
        String dirver = "com.mysql.cj.jdbc.Driver";
        String url = "";
        String user = "";
        String password = "";
        String database = "";
        String deltaQueryProperties = "DATE_FORMAT(update_time,'%Y-%m-%d %H:%i:%S')";

        String table = "custom_user";
        String primaryKey = "id";

        map.put("driver",dirver);
        map.put("url", url);
        map.put("user", user);
        map.put("password", password);
        map.put("table", table);
        map.put("primaryKey", primaryKey);
        map.put("database", database);
        map.put("deltaQueryProperties", deltaQueryProperties);
//        map.put("convertType", convertType);
//        map.put("updateTimeKey", updateTimeKey);
//        map.put("timeType", timeType);

        List<FieldVmConfig> dataConfigFields = new ArrayList<>();
        dataConfigFields.add(new FieldVmConfig("id","id"));
        dataConfigFields.add(new FieldVmConfig("username","username"));
        dataConfigFields.add(new FieldVmConfig("password","password"));
        dataConfigFields.add(new FieldVmConfig("gender","gender"));
        dataConfigFields.add(new FieldVmConfig("birth","birth"));
        dataConfigFields.add(new FieldVmConfig("email","email"));
        dataConfigFields.add(new FieldVmConfig("phone","phone"));
        dataConfigFields.add(new FieldVmConfig("remark","remark"));
        dataConfigFields.add(new FieldVmConfig("createTime","create_time"));
        dataConfigFields.add(new FieldVmConfig("updateTime","update_time"));
        map.put("dataConfigFields", dataConfigFields);

        VelocityContext context = new VelocityContext(map);
        // 函数库
        context.put("math", new MathTool());
        context.put("dateTool", new DateTool());

        // 获取模板列表
        List<String> templates = Collections.singletonList("template/dataconfig.xml.vm");
        Map<String, String> resultMap = new LinkedHashMap<>(8);

        for (String template : templates) {
            // 渲染模板
            StringWriter sw = new StringWriter();
            Template tpl = Velocity.getTemplate(template, CharsetUtil.UTF_8.toString());
            tpl.merge(context, sw);
            FileUtil.writeString(sw.toString(), "dataimport", CharsetUtil.UTF_8);
            FileWriter fw = new FileWriter("dataimport.xml");
            fw.write(sw.toString());
            fw.close();
            resultMap.put(template, sw.toString());
        }
    }

.vm模板文件

<?xml version="1.0" encoding="UTF-8"?>
<dataConfig>
    <dataSource type="JdbcDataSource"
                driver="${driver}"
                url="${url}"
                user="${user}"
                password="${password}"/>
    <document>
        <entity name="${table}" pk="id"
                query="select ${primaryKey} as id${columns} from ${database}.${table}  where deleted ='N' "

                deltaImportQuery="select ${primaryKey} as id${columns} from ${database}.${table} where ${primaryKey} = '${dih.delta.id}'"

                deltaQuery="select ${primaryKey} from ${database}.${table} where ${convertType}(${updateTimeKey},${timeType})  &gt; '${dih.last_index_time}' and deleted ='N' "

                deletedPkQuery="select ${primaryKey} from ${database}.${table}  where deleted ='Y'"
        >
            #foreach ($field in $dataConfigFields)
                <field column="$field.columnName" name="$field.name"/>
            #end

        </entity>
    </document>
</dataConfig>