自动生成数据库表设计（二）之Freemarker的基本使用

最近项目快了验收，那么接下来就是写一些比较烦人的文档，在写数据库设计文档时，到了详细设计这一块有点尴尬了，每张表，没个字段都要写上去，由于前期没有整理，所以这个工作量还是很大，我查看了我们的数据库发现有353张表，这样写，得花多久的时间啊。。。于是想通过程序来自动完成，这就是这篇文章的核心。

系列文章：

自动生成数据库表设计（一）之获取JDBC获取元数据

自动生成数据库表设计（二）之Freemarker的基本使用

自动生成数据库表设计（三）之制作word模版

本篇主要内容：

1、Freemarker的使用

2、Freemarker的List的使用

3、Freemarker的List嵌套List的使用

Freemarker简单使用

1、引入freemarker依赖

<dependency>
<groupId>org.freemarker</groupId>
<artifactId>freemarker</artifactId>
<version>2.3.23</version>
</dependency>

2、引入Freemarker的工具类

public class FtUtil {
/**
* 获取模板
*
* @param templatesDir 例如"/templates"
* @return
*/
public Template getTemplate(String templatesDir, String name) {
try {
//通过Freemaker的Configuration读取相应的ftl
Configuration cfg = new Configuration();
//设定去哪里读取相应的ftl模板文件
cfg.setClassForTemplateLoading(this.getClass(), templatesDir);
//在模板文件目录中找到名称为name的文件
Template temp = cfg.getTemplate(name);
return temp;
} catch (IOException e) {
e.printStackTrace();
}
return null;
}

/**
* Description: 根据模版生成文件 <br/>
*/
public void generateFile(String templatesDir, String templateName, Map root, String outDir, String outFileName) {
FileWriter out = null;
try {
//通过一个文件输出流，就可以写到相应的文件中
out = new FileWriter(new File(outDir, outFileName));
Template temp = this.getTemplate(templatesDir, templateName);
temp.process(root, out);
} catch (IOException e) {
e.printStackTrace();
} catch (TemplateException e) {
e.printStackTrace();
} finally {
try {
if (out != null) out.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}

3、在resources下新建test.ftl文件

<#-- 这是Freemarker注释 -->
<#-- 获取简单值 -->
${table}

这里就是简单的取变量为table的值，下面我们就是用这个模版文件生成文件

4、测试 生成文件

public class TestFreemaker {
public static void main(String[] args) throws Exception {
Map map = new HashMap<>();
map.put("table", "123");

FtUtil ftUtil = new FtUtil();
ftUtil.generateFile("/", "test.ftl", map, "D:/", "test.txt");
}
}

这里给变量table设置了值123，生成文件将会显示123，就说明我们已经会使用Freemarker了，运行结果如下：



现在我们已经可以取单个变量了，下面我们看在模版里取集合的应用

Freemarker中List的使用

1、遍历List的用法如下：

<#-- List使用 -->
<#list table as t>
序号：${t_index}   表名：${t.NAME}
</#list>

其中table表示List,而t是List里的元素， t_index是表示t这个元素在list中的下标，t.NAME是表示取名为NAME的属性值

2、修改Main方法

public static void main(String[] args) throws Exception {

//模拟一张表
Map<String, Object> table = new HashMap<>();
table.put("NAME", "T_USER");

//模拟一个表集合
List<Map<String, Object>> tableList = new ArrayList<>();
tableList.add(table);

Map map = new HashMap<>();
//map.put("table", "123");
map.put("table", tableList);

FtUtil ftUtil = new FtUtil();
ftUtil.generateFile("/", "test.ftl", map, "D:/", "test.txt");
}

结果如下：

Freemarker中List嵌套List的使用

1、修改test.ftl如下：

<#-- List嵌套List的使用 -->
<#list table as t>
序号：${t_index}   表名：${t.NAME}
<#list t.COLUMNS as c>
序号：${c_index}   列名：${c.NAME}
</#list>
</#list>

其中t对象里多了一个属性COLUMNS，这个COLUMNS也是一个集合

2、修改Main方法

public static void main(String[] args) throws Exception {

//模拟2个列ID、AGE
Map<String, Object> column1 = new HashMap<>();
column1.put("NAME", "ID");
Map<String, Object> column2 = new HashMap<>();
column2.put("NAME", "AGE");

//模拟一个列集合
List<Map<String, Object>> columnList = new ArrayList<>();
columnList.add(column1);
columnList.add(column2);

//模拟一张表
Map<String, Object> table = new HashMap<>();
table.put("NAME", "T_USER");
table.put("COLUMNS", columnList);

//模拟一个表集合
List<Map<String, Object>> tableList = new ArrayList<>();
tableList.add(table);

Map map = new HashMap<>();
//map.put("table", "123");
map.put("table", tableList);

FtUtil ftUtil = new FtUtil();
ftUtil.generateFile("/", "test.ftl", map, "D:/", "test.txt");
}

运行结果如下：



有了如上知识，加上前面我们准备的表和列的数据，就可以根据任意模版生成对应的表数据和列数据了。下一篇我将介绍如何创建word模版。

文档下载