Freemarker模板自动生成java代码

前言：对于初次使用代码生成器的人来说，最先要了解的是思路，下面我们具体来看如何开始构建自己的代码生成工具。

一，了解Freemarker.

这里说的了解，只是简单了解，能够使用就行。

我们就当做Freemarker是一个模板工具，引入相关jar包，我们就可以直接使用了。

使用方式：

1，写一个模板页面，数据用${name}占位符的方式显示。

2，后台将实际数据name=testName加入到Freemarker实体工具中

3，指定模板存放的目录

4，运行，生成的页面中，${name}会自动变成实际数据【testName】

下面来看实例：

新建一个普通maven工程，先看目录：

接下来在pom.xml中引入依赖包：

<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
  <modelVersion>4.0.0</modelVersion>
  <groupId>test</groupId>
  <artifactId>test</artifactId>
  <version>0.0.1-SNAPSHOT</version>
  
	<dependencies>
		<!-- freemarker -->
		<dependency>
			<groupId>org.freemarker</groupId>
			<artifactId>freemarker</artifactId>
			<version>2.3.23</version>
		</dependency>
	</dependencies>
  
</project>

在src/main/resources目录下新建一个文件夹template，用于存放我们写的模板

这里只写了一个简单实体类的测试模板，test.java，模板名称可以任意取。

package com.test.entity;

public class testEntity {

	private ${type} ${name};
	
	public ${type} get${name?cap_first}() {
		return ${name};
	}
	
	public void set${name?cap_first}(${type} ${name}) {
		this.${name} = ${name};
	}
}

接下来就是依据模板生成目标文件，主要代码如下：

package com.test.main;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.Writer;
import java.util.HashMap;
import java.util.Map;

import freemarker.template.Configuration;
import freemarker.template.Template;
import freemarker.template.TemplateException;

public class GenMain {

	public static void main(String[] args) throws IOException {
		// 参数值
		Map<String, Object> map = new HashMap<String, Object>();
		map.put("type", "String");
		map.put("name", "testName");
		
		// 模板目录
		String templateDirectory = "src/main/resources/template";
		// 模板名称
		String templateFile = "test.java";
		// 模板生成后存放目录
		String targetPath = "C:/Users/Administrator/Desktop/test";
		// 模板生成后新文件名
		String fileName = "ntest.java";
		// 创建文件夹
		new File(targetPath).mkdirs();
		File nFile = new File(targetPath +"/"+ fileName);
		if (nFile.exists()) {
			throw new RuntimeException("File \'"+fileName+"\' already exists");
		}
		
		// 生成目标文件
		Writer writer = null;
		try {
			writer = new FileWriter(nFile);
			
			Template template = getConfiguration(templateDirectory).getTemplate(templateFile, "UTF-8");
			template.process(map, writer);
			
		} catch (TemplateException e) {
			throw new RuntimeException(e);
		} catch (IOException e) {
			throw new RuntimeException(e);
		} finally {
			writer.close();
		}
	}
	
	private static Configuration getConfiguration(String templateDirectory) {
		
		Configuration configuration = new Configuration(Configuration.VERSION_2_3_22);
		try {
			configuration.setTagSyntax(Configuration.AUTO_DETECT_TAG_SYNTAX);
			configuration.setDirectoryForTemplateLoading(new File(templateDirectory));
		} catch (IOException e) {
			throw new RuntimeException(e);
		}
		
		return configuration;
	}
}

运行后在指定的目录下面就会生成我们指定的ntest.java文件了，

查看生成后的代码：

package com.test.entity;

public class testEntity {

	private String testName;
	
	public String getTestName() {
		return testName;
	}
	
	public void setTestName(String testName) {
		this.testName = testName;
	}
}

最后将文件导入工程中就可以使用了，是不是很简单呢。

用相同的方式我们就可以生成需要的代码了，通过指定数据库，生成对应的从controller到mapper.xml的整套代码。

具体逻辑如下：

1，我们需要有个完整的项目，已经写好了一个demo，用于对数据库的增删改查的操作。

2，依据demo，写需要的模板，只是将实际数据换成占位符${}

3，读取数据库表的元数据，将所有需要替换占位符的参数都放到一个map中，然后用上述方式生成代码。

由于项目不同，模板也不同，这里就不一一演示了。

需要注意的是，如果使用mybaits，需要生成xml文件，由于$,#是freemarker的标签，会与xml中的冲突，

因此需要将页面中需要用的$,#也当做参数放到map中，使用的时候用${‘$’} , ${'#'}替换的方式获取。