浏览代码

修复代码生成器非windows系统下模板路径问题

shuzheng 8 年之前
父节点
当前提交
1f9068d224
共有 1 个文件被更改,包括 12 次插入7 次删除
  1. 12 7
      zheng-common/src/main/java/com/zheng/common/util/MybatisGeneratorUtil.java

+ 12 - 7
zheng-common/src/main/java/com/zheng/common/util/MybatisGeneratorUtil.java

@@ -50,10 +50,18 @@ public class MybatisGeneratorUtil {
 			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 os = System.getProperty("os.name");
+		if (os.toLowerCase().startsWith("win")) {
+			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("/", "");
+		} else {
+			generatorConfig_vm = MybatisGeneratorUtil.class.getResource(generatorConfig_vm).getPath();
+			service_vm = MybatisGeneratorUtil.class.getResource(service_vm).getPath();
+			serviceMock_vm = MybatisGeneratorUtil.class.getResource(serviceMock_vm).getPath();
+			serviceImpl_vm = MybatisGeneratorUtil.class.getResource(serviceImpl_vm).getPath();
+		}
 
 		String targetProject = module + "/" + module + "-dao";
 		String basePath = MybatisGeneratorUtil.class.getResource("/").getPath().replace("/target/classes/", "").replace(targetProject, "").replaceFirst("/", "");
@@ -153,9 +161,6 @@ public class MybatisGeneratorUtil {
 			}
 		}
 		System.out.println("========== 结束生成Service ==========");
-
-		System.out.println("========== 开始生成Controller ==========");
-		System.out.println("========== 结束生成Controller ==========");
 	}
 
 	// 递归删除非空文件夹