From 211840b64fa1132d76d6dff6c779e9ba2c0c450f Mon Sep 17 00:00:00 2001 From: admin <weikou2014> Date: 星期四, 07 四月 2022 19:13:54 +0800 Subject: [PATCH] 优化自动化代码 --- src/main/java/org/yeshi/utils/generater/mybatis/MyBatisMapperUtil.java | 27 +++++++++++++++------------ 1 files changed, 15 insertions(+), 12 deletions(-) diff --git a/src/main/java/org/yeshi/utils/generater/mybatis/MyBatisMapperUtil.java b/src/main/java/org/yeshi/utils/generater/mybatis/MyBatisMapperUtil.java index 8487d86..e8427d3 100644 --- a/src/main/java/org/yeshi/utils/generater/mybatis/MyBatisMapperUtil.java +++ b/src/main/java/org/yeshi/utils/generater/mybatis/MyBatisMapperUtil.java @@ -85,16 +85,16 @@ } if (type.equalsIgnoreCase("Date")) { String tempProperty = "min" + property.substring(0, 1).toUpperCase() + property.substring(1); - buffer.append(String.format("\tpublic %s %s", type, tempProperty)); - queryColumnData.add(new MybatisColumnData(columnName, tempProperty, type)); + buffer.append(String.format("\tpublic %s %s;", type, tempProperty)); + queryColumnData.add(new MybatisColumnData(columnName, tempProperty, type,ColumnParseUtil.getJDBCType(genericType))); buffer.append("\n\t"); tempProperty = "max" + property.substring(0, 1).toUpperCase() + property.substring(1); - buffer.append(String.format("\tpublic %s %s", type, tempProperty)); - queryColumnData.add(new MybatisColumnData(columnName, tempProperty, type)); + buffer.append(String.format("\tpublic %s %s;", type, tempProperty)); + queryColumnData.add(new MybatisColumnData(columnName, tempProperty, type,ColumnParseUtil.getJDBCType(genericType))); buffer.append("\n\t"); } else { - buffer.append(String.format("\tpublic %s %s", type, property)); - queryColumnData.add(new MybatisColumnData(columnName, property, type)); + buffer.append(String.format("\tpublic %s %s;", type, property)); + queryColumnData.add(new MybatisColumnData(columnName, property, type,ColumnParseUtil.getJDBCType(genericType))); buffer.append("\n\t"); } } @@ -115,6 +115,8 @@ return queryColumnData; } + + public static void createMapper(Class<?> clz) { // 鐢熸垚mapper java鏂囦欢 String pks = getDaoPackageName(clz); @@ -126,12 +128,13 @@ buffer.append(String.format("public interface %sMapper extends BaseMapper<%s> {", clz.getSimpleName(), clz.getSimpleName())); buffer.append("\n\n\t"); - buffer.append(String.format("%s selectByPrimaryKeyForUpdate(@Param(\"id\") id);",clz.getSimpleName())); + //TODO 纭畾ID绫诲瀷 + buffer.append(String.format("%s selectByPrimaryKeyForUpdate(@Param(\"id\") Long id);", clz.getSimpleName())); buffer.append("\n\n\t"); - buffer.append(String.format("List<%s> list(@Param(\"query\") query);", clz.getSimpleName())); + buffer.append(String.format("List<%s> list(@Param(\"query\") DaoQuery query);", clz.getSimpleName())); buffer.append("\n\n\t"); - buffer.append("long count(@Param(\"query\") query);"); + buffer.append("long count(@Param(\"query\") DaoQuery query);"); buffer.append("\n\n\t"); List<MybatisColumnData> queryColumnData = createMapperDaoQuery(importPosition, buffer, clz); @@ -236,7 +239,7 @@ select.addText("select"); org.dom4j.Element include = select.addElement("include"); include.addAttribute("refid", "Base_Column_List"); - select.addText(String.format("from %s where %s = #{%s,jdbcType=BIGINT}", tableName, idKeys.column, + select.addText(String.format(" from %s where %s = #{%s,jdbcType=BIGINT}", tableName, idKeys.column, idKeys.attribute)); @@ -247,7 +250,7 @@ select.addText("select"); include = select.addElement("include"); include.addAttribute("refid", "Base_Column_List"); - select.addText(String.format("from %s where %s = #{%s,jdbcType=BIGINT} for update", tableName, idKeys.column, + select.addText(String.format(" from %s where %s = #{%s,jdbcType=BIGINT} for update", tableName, idKeys.column, idKeys.attribute)); //娣诲姞sql @@ -274,7 +277,7 @@ select.addText("select"); include = select.addElement("include"); include.addAttribute("refid", "Base_Column_List"); - select.addText(String.format("from %s where 1=1", tableName)); + select.addText(String.format(" from %s where 1=1", tableName)); include = select.addElement("include"); include.addAttribute("refid", "listWhereSQL"); -- Gitblit v1.8.0