修改zheng-common代码生成工具类，自动检测模板路径，代替路径硬编码

shuzheng · shuzheng · commit ed12d88568da · 2017-05-03T17:08:15.000+08:00
diff --git a/zheng-common/src/main/java/com/zheng/common/util/MybatisGeneratorUtil.java b/zheng-common/src/main/java/com/zheng/common/util/MybatisGeneratorUtil.java
@@ -20,13 +20,13 @@
 public class MybatisGeneratorUtil {
 
 	// generatorConfig模板路径
-	private static String generatorConfig_vm = "zheng-common/src/main/resources/template/generatorConfig.vm";
+	private static String generatorConfig_vm = "/template/generatorConfig.vm";
 	// Service模板路径
-	private static String service_vm = "zheng-common/src/main/resources/template/Service.vm";
+	private static String service_vm = "/template/Service.vm";
 	// ServiceMock模板路径
-	private static String serviceMock_vm = "zheng-common/src/main/resources/template/ServiceMock.vm";
+	private static String serviceMock_vm = "/template/ServiceMock.vm";
 	// ServiceImpl模板路径
-	private static String serviceImpl_vm = "zheng-common/src/main/resources/template/ServiceImpl.vm";
+	private static String serviceImpl_vm = "/template/ServiceImpl.vm";
 
 	/**
 	 * 根据模板生成generatorConfig.xml文件
@@ -50,6 +50,11 @@ public static void generator(
 			String package_name,
 			Map<String, String> last_insert_id_tables) throws Exception{
 
+		generatorConfig_vm = MybatisGeneratorUtil.class.getResource(generatorConfig_vm).getPath().replaceFirst("/", "");
+		service_vm = MybatisGeneratorUtil.class.getResource(service_vm).getPath().replaceFirst("/", "");
+		serviceMock_vm = MybatisGeneratorUtil.class.getResource(serviceMock_vm).getPath().replaceFirst("/", "");
+		serviceImpl_vm = MybatisGeneratorUtil.class.getResource(serviceImpl_vm).getPath().replaceFirst("/", "");
+
 		String targetProject = module + "/" + module + "-dao";
 		String module_path = module + "/" + module + "-dao/src/main/resources/generatorConfig.xml";
 		String sql = "SELECT table_name FROM INFORMATION_SCHEMA.TABLES WHERE table_schema = '" + database + "' AND table_name LIKE '" + table_prefix + "_%';";
diff --git a/zheng-common/src/main/java/com/zheng/common/util/VelocityUtil.java b/zheng-common/src/main/java/com/zheng/common/util/VelocityUtil.java
@@ -1,12 +1,14 @@
 package com.zheng.common.util;
 
+import org.apache.commons.lang.StringUtils;
 import org.apache.velocity.Template;
 import org.apache.velocity.VelocityContext;
 import org.apache.velocity.app.Velocity;
 import org.apache.velocity.app.VelocityEngine;
 
 import java.io.File;
 import java.io.FileWriter;
+import java.util.Properties;
 
 /**
  * Velocity工具类
@@ -23,9 +25,11 @@ public class VelocityUtil {
 	 */
 	public static void generate(String inputVmFilePath, String outputFilePath, VelocityContext context) throws Exception {
 		try {
-			Velocity.init();
-			VelocityEngine engine = new VelocityEngine();
-			Template template = engine.getTemplate(inputVmFilePath, "utf-8");
+			Properties properties = new Properties();
+			properties.setProperty(VelocityEngine.FILE_RESOURCE_LOADER_PATH, getPath(inputVmFilePath));
+			Velocity.init(properties);
+			//VelocityEngine engine = new VelocityEngine();
+			Template template = Velocity.getTemplate(getFile(inputVmFilePath), "utf-8");
 			File outputFile = new File(outputFilePath);
 			FileWriter writer = new FileWriter(outputFile);
 			template.merge(context, writer);
@@ -35,4 +39,30 @@ public static void generate(String inputVmFilePath, String outputFilePath, Veloc
 		}
 	}
 
+	/**
+	 * 根据文件绝对路径获取目录
+	 * @param filePath
+	 * @return
+	 */
+	public static String getPath(String filePath) {
+		String path = "";
+		if (StringUtils.isNotBlank(filePath)) {
+			path = filePath.substring(0, filePath.lastIndexOf("/") + 1);
+		}
+		return path;
+	}
+
+	/**
+	 * 根据文件绝对路径获取文件
+	 * @param filePath
+	 * @return
+	 */
+	public static String getFile(String filePath) {
+		String file = "";
+		if (StringUtils.isNotBlank(filePath)) {
+			file = filePath.substring(filePath.lastIndexOf("/") + 1);
+		}
+		return file;
+	}
+
 }
