整合maven+mybatis+generator生成java自定义model实体类,dao接口和mapper映射文件

package com.xe.demo.common.generator; import java.sql.Types; import java.util.HashSet; import java.util.List; import java.util.Properties; import java.util.Set; import org.mybatis.generator.api.IntrospectedColumn; import org.mybatis.generator.api.IntrospectedTable; import org.mybatis.generator.api.PluginAdapter; import org.mybatis.generator.api.dom.java.FullyQualifiedJavaType; import org.mybatis.generator.api.dom.java.Interface; import org.mybatis.generator.api.dom.java.Method; import org.mybatis.generator.api.dom.java.TopLevelClass; import org.mybatis.generator.api.dom.xml.Attribute; import org.mybatis.generator.api.dom.xml.Document; import org.mybatis.generator.api.dom.xml.TextElement; import org.mybatis.generator.api.dom.xml.XmlElement; import org.mybatis.generator.codegen.mybatis3.MyBatis3FormattingUtilities; import org.mybatis.generator.config.CommentGeneratorConfiguration; import org.mybatis.generator.config.Context; /** * 自定义插件 * @author CZH */ public class GenPlugin2 extends PluginAdapter { private Set<String> mappers = new HashSet<String>(); // 注释生成器 private CommentGeneratorConfiguration commentCfg; @Override public void setContext(Context context) { super.setContext(context); // 设置默认的注释生成器 commentCfg = new CommentGeneratorConfiguration(); commentCfg.setConfigurationType(GenCommentGenerator.class.getCanonicalName()); context.setCommentGeneratorConfiguration(commentCfg); // 支持oracle获取注释#114 context.getJdbcConnectionConfiguration().addProperty("remarksReporting", "true"); } @Override public void setProperties(Properties properties) { super.setProperties(properties); String mappers = this.properties.getProperty("mappers"); for (String mapper : mappers.split(",")) { this.mappers.add(mapper); } } @Override public boolean validate(List<String> warnings) { return true; } /** * 生成的Mapper接口 * @param interfaze * @param topLevelClass * @param introspectedTable * @return */ @Override public boolean clientGenerated(Interface interfaze, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { // import接口 for (String mapper : mappers) { interfaze.addImportedType(new FullyQualifiedJavaType(mapper)); interfaze.addSuperInterface(new FullyQualifiedJavaType(mapper)); } return true; } @Override public boolean sqlMapDocumentGenerated(Document document, IntrospectedTable introspectedTable) { XmlElement rootElement = document.getRootElement(); // 数据库表名 String tableName = introspectedTable.getFullyQualifiedTableNameAtRuntime(); // 主键 IntrospectedColumn pkColumn = introspectedTable.getPrimaryKeyColumns().get(0); // 公共字段 XmlElement columnSql = new XmlElement("sql"); columnSql.addAttribute(new Attribute("id", "sql_columns")); StringBuilder columnStr = new StringBuilder(); // 添加公共where XmlElement whereSql = new XmlElement("sql"); whereSql.addAttribute(new Attribute("id", "sql_where")); XmlElement where = new XmlElement("where"); StringBuilder whereStr = new StringBuilder(); //拼装更新字段 XmlElement updateSql = new XmlElement("sql"); updateSql.addAttribute(new Attribute("id", "sql_update")); // 新增数据 XmlElement save = new XmlElement("insert"); save.addAttribute(new Attribute("id", "save")); save.addAttribute(new Attribute("keyProperty", pkColumn.getJavaProperty())); save.addAttribute(new Attribute("useGeneratedKeys", "true")); StringBuilder saveStr = new StringBuilder("insert into ").append(tableName).append("("); // 要插入的字段(排除自增主键) StringBuilder saveColumn = new StringBuilder(); // 要保存的值 StringBuilder saveValue = new StringBuilder(); // 批量保存 XmlElement batchSave = new XmlElement("insert"); batchSave.addAttribute(new Attribute("id", "batchSave")); StringBuilder btcSaveStr = new StringBuilder("insert into ").append(tableName).append("("); // 更新数据 XmlElement update = new XmlElement("update"); update.addAttribute(new Attribute("id", "update")); StringBuilder updateStr = new StringBuilder("update ").append(tableName).append(" set "). append(pkColumn.getActualColumnName()).append(" = #{").append(pkColumn.getJavaProperty()).append("}"); update.addElement(new TextElement(updateStr.toString())); // 批量更新 XmlElement batchUpdate = new XmlElement("update"); batchUpdate.addAttribute(new Attribute("id", "batchUpdate")); XmlElement foreachUpdate = new XmlElement("foreach"); foreachUpdate.addAttribute(new Attribute("collection", "list")); foreachUpdate.addAttribute(new Attribute("item", "item")); foreachUpdate.addAttribute(new Attribute("index", "index")); foreachUpdate.addAttribute(new Attribute("open", "")); foreachUpdate.addAttribute(new Attribute("close", "")); foreachUpdate.addAttribute(new Attribute("separator", ";")); batchUpdate.addElement(foreachUpdate); StringBuilder btcUpdateStr = new StringBuilder("update ").append(tableName).append(" set "). append(pkColumn.getActualColumnName()).append(" = #{item.").append(pkColumn.getJavaProperty()).append("}");; foreachUpdate.addElement(new TextElement(btcUpdateStr.toString())); // 数据库字段名 String columnName = null; // java字段名 String javaProperty = null; for (IntrospectedColumn introspectedColumn : introspectedTable.getAllColumns()) { columnName = MyBatis3FormattingUtilities.getEscapedColumnName(introspectedColumn); javaProperty = introspectedColumn.getJavaProperty(); // 拼装字段 columnStr.append(columnName).append(","); // 拼装公共条件 XmlElement isNotNullElement = new XmlElement("if"); whereStr.setLength(0); whereStr.append("null != item.").append(javaProperty).append(" and '' != item.").append(javaProperty); isNotNullElement.addAttribute(new Attribute("test", whereStr.toString())); whereStr.setLength(0); whereStr.append(" , ").append(columnName).append(" = #{item.").append(javaProperty).append("}"); isNotNullElement.addElement(new TextElement(whereStr.toString())); where.addElement(isNotNullElement); updateSql.addElement(isNotNullElement); //保存SQL if (!introspectedColumn.isAutoIncrement()) { saveColumn.append(",").append(columnName); if(Types.TIMESTAMP == introspectedColumn.getJdbcType()){ saveValue.append(", now()"); }else{ saveValue.append(", #{item.").append(javaProperty).append("}"); } } } String columns = columnStr.substring(0, columnStr.length() - 1); columnStr = new StringBuilder("select ").append(columns).append(" from ").append(tableName); columnSql.addElement(new TextElement(columns)); rootElement.addElement(columnSql); whereSql.addElement(new TextElement(where.getFormattedContent(0).replaceAll(",", "and"))); rootElement.addElement(whereSql); rootElement.addElement(updateSql); saveStr.append(saveColumn.substring(1)).append(") values(").append(saveValue.substring(1)).append(")"); save.addElement(new TextElement(saveStr.toString())); btcSaveStr.append(saveColumn.substring(1)).append(") values"); batchSave.addElement(new TextElement(btcSaveStr.toString())); btcSaveStr.setLength(0); XmlElement foreach = new XmlElement("foreach"); foreach.addAttribute(new Attribute("collection", "list")); foreach.addAttribute(new Attribute("item", "item")); foreach.addAttribute(new Attribute("index", "index")); foreach.addAttribute(new Attribute("separator", ",")); foreach.addElement(new TextElement("(" + saveValue.toString().substring(1) + ")")); batchSave.addElement(foreach); XmlElement include = new XmlElement("include"); include.addAttribute(new Attribute("refid", "sql_update")); columnStr = new Str