wutianbin 7 years ago
parent
commit
9e67514aaf

+ 3 - 1
src/main/java/org/geek/szbay/assistant/code/CodeHelpper.java

@@ -39,7 +39,9 @@ public class CodeHelpper {
     
     /** 表的前导字符 */
     //private String preTableName = "*";
-    private String preTableName = "fp\\_exam";
+    private String preTableName = "oa\\_";
+    //private String preTableName = "fp\\_exam";
+    //private String preTableName = "sys\\_";
     
     /** 指定实体生成所在包的路径 */
     private String entityOutPath = "com.donghaiair.flyprepare.admin.domain";

+ 10 - 0
src/main/java/org/geek/szbay/assistant/code/generator/MapperXmlMaker.java

@@ -72,6 +72,7 @@ public class MapperXmlMaker {
 		parseAllField(sb, colNames);
 
 		parseDelete(sb, key, keyAttr, tableName, entityFullClassName, colNames, attrNames);
+		parseDeleteByKey(sb, key, keyAttr, tableName, entityFullClassName, colNames, attrNames);
 		parseInsert(sb, key, keyAttr, tableName, entityFullClassName, colNames, attrNames);
 		parseUpdate(sb, key, keyAttr, tableName, entityFullClassName, colNames, attrNames);
 		parseUpdateByExample(sb, key, keyAttr, tableName, entityFullClassName, colNames, attrNames);
@@ -147,6 +148,15 @@ public class MapperXmlMaker {
 		sb.append("\r\n");
 	}
 
+	private static final void parseDeleteByKey(StringBuffer sb, String key, String keyAttr, String tableName,
+			String entityFullClassName, List<String> colNames, List<String> attrNames) {
+		sb.append("    <delete id=\"deleteByKey\">\r\n");
+		sb.append("        delete from " + tableName + "\r\n");
+		sb.append("        where " + key + " = #{delete." + key + "}\r\n");
+		sb.append("    </delete>\r\n");
+		sb.append("\r\n");
+	}
+
 	private static final void parseInsert(StringBuffer sb, String key, String keyAttr, String tableName,
 			String entityFullClassName, List<String> colNames, List<String> attrNames) {
 		sb.append("    <insert id=\"save\" useGeneratedKeys=\"");

+ 1 - 1
src/main/java/org/geek/szbay/assistant/code/generator/ServiceImplMysqlMaker.java

@@ -49,7 +49,7 @@ public class ServiceImplMysqlMaker {
         sb.append("public class ").append(className)
         	.append(" extends BaseServiceImpl<").append(entityClassName).append(">\r\n")
         	.append("        implements ").append(serviceClassName)
-        	.append("<").append(entityClassName).append("> {\r\n");
+        	.append(" {\r\n");
         sb.append("\r\n");
         
         String daoObject = StringUtils.firstToLowerCase(daoClassName);

+ 3 - 3
src/main/java/org/geek/szbay/assistant/code/generator/ServiceMysqlMaker.java

@@ -27,8 +27,8 @@ public class ServiceMysqlMaker {
           
         sb.append("package " + outPath + ";\r\n");  
         sb.append("\r\n");  
-//        sb.append("import " + entityFullClassName + ";\r\n");
-//        sb.append("\r\n");  
+        sb.append("import " + entityFullClassName + ";\r\n");
+        sb.append("\r\n");  
         
         //注释部分  
         sb.append(MakeHelpper.makeClassComments(tableComment + "(" + tableName + ") 实体类的数据服务类", authorName));
@@ -37,7 +37,7 @@ public class ServiceMysqlMaker {
 //        sb.append("public interface " + className + " {\r\n");
         //public interface FpExamDefineService<FpExamDefineDO> extends BaseService<FpExamDefineDO> {
         sb.append("public interface ").append(className)
-        		.append("<").append(entityClassName).append("> extends BaseService<")
+        		.append(" extends BaseService<")
         		.append(entityClassName).append("> {\r\n");	
         sb.append("\r\n");