直接上代码吧
public class GenerateListPlugin extends PluginAdapter { private FullyQualifiedJavaType primaryKeyType; private FullyQualifiedJavaType modelType; @Override public boolean validate(List warnings) { return true; } @Override public boolean clientSelectByPrimaryKeyMethodGenerated(Method method, Interface interfaze, IntrospectedTable introspectedTable) { primaryKeyType = method.getParameters().get(0).getType(); modelType = method.getReturnType(); return super.clientSelectByPrimaryKeyMethodGenerated(method, interfaze, introspectedTable); } @Override public boolean clientGenerated(Interface interfaze, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { //batch insert { Method method = new Method(); method.setName("batchInsert"); method.setReturnType(new FullyQualifiedJavaType("int")); method.addParameter(new Parameter(new FullyQualifiedJavaType("java.util.List<" + modelType.getShortName() + ">"), "list")); interfaze.addMethod(method); interfaze.addImportedType(new FullyQualifiedJavaType("java.util.List")); } //list all { Method method = new Method(); method.setName("listAll"); method.setReturnType(new FullyQualifiedJavaType("java.util.List<" + modelType.getShortName() + ">")); interfaze.addMethod(method); interfaze.addImportedType(new FullyQualifiedJavaType("java.util.List")); } //list by primary key List primaryKeyColumns = introspectedTable.getPrimaryKeyColumns(); if(primaryKeyColumns != null && primaryKeyColumns.size() == 1) { Method listMethod = new Method(); listMethod.setName("listByPrimaryKey"); listMethod.setReturnType(new FullyQualifiedJavaType("java.util.List<"+modelType.getShortName()+">")); Parameter parameter = new Parameter(new FullyQualifiedJavaType("java.util.List<"+primaryKeyType.getShortName()+">"), "list"); listMethod.addParameter(parameter); interfaze.addMethod(listMethod); interfaze.addImportedType(new FullyQualifiedJavaType("java.util.List")); } return super.clientGenerated(interfaze, topLevelClass, introspectedTable); } @Override public boolean sqlMapDocumentGenerated(Document document, IntrospectedTable introspectedTable) { XmlElement root = document.getRootElement(); //batch insert { //batch insert XmlElement batchInsert = new XmlElement("insert"); batchInsert.addAttribute(new Attribute("id", "batchInsert")); batchInsert.addAttribute(new Attribute("parameterType", "list")); StringBuilder cols = new StringBuilder(); StringBuilder vals = new StringBuilder(); List baseColumns = introspectedTable.getBaseColumns(); boolean isFirst = true; for(IntrospectedColumn column : baseColumns) { if(!isFirst) { cols.append(','); vals.append(','); } else { isFirst = false; } cols.append(column.getActualColumnName()); //#{id,jdbcType=BIGINT} vals.append("#{item.").append(column.getJavaProperty()).append(",jdbcType=").append(column.getJdbcTypeName()).append('}'); } StringBuilder insertBuilder = new StringBuilder(); insertBuilder.append("insert into").append(introspectedTable.getFullyQualifiedTable().getIntrospectedTableName()) .append("(").append(cols).append(") values "); TextElement insert = new TextElement(insertBuilder.toString()); batchInsert.addElement(insert); XmlElement foreach = new XmlElement("foreach"); foreach.addAttribute(new Attribute("collection", "list")); foreach.addAttribute(new Attribute("item", "item")); foreach.addAttribute(new Attribute("separator", ",")); foreach.addElement(new TextElement("(" + vals.toString() + ")")); batchInsert.addElement(foreach); root.addElement(batchInsert); } //list all { XmlElement selectAll = new XmlElement("select"); selectAll.addAttribute(new Attribute("id", "listAll")); selectAll.addAttribute(new Attribute("resultMap", "BaseResultMap")); TextElement selectSql = new TextElement("select"); selectAll.addElement(selectSql); XmlElement include = new XmlElement("include"); include.addAttribute(new Attribute("refid", "Base_Column_List")); selectAll.addElement(include); selectAll.addElement(new TextElement("from " + introspectedTable.getTableConfiguration().getTableName())); root.addElement(selectAll); } //list by primary key List primaryKeyColumns = introspectedTable.getPrimaryKeyColumns(); if(primaryKeyColumns != null && primaryKeyColumns.size() == 1) { XmlElement listByPrimaryKey = new XmlElement("select"); listByPrimaryKey.addAttribute(new Attribute("id", "listByPrimaryKey")); listByPrimaryKey.addAttribute(new Attribute("resultMap", "BaseResultMap")); TextElement selectEle = new TextElement("select"); listByPrimaryKey.addElement(selectEle); XmlElement includeEle = new XmlElement("include"); includeEle.addAttribute(new Attribute("refid", "Base_Column_List")); listByPrimaryKey.addElement(includeEle); listByPrimaryKey.addElement(new TextElement("from " + introspectedTable.getTableConfiguration().getTableName())); TextElement where = new TextElement("where " + primaryKeyColumns.get(0).getActualColumnName() + " in"); listByPrimaryKey.addElement(where); XmlElement foreach = new XmlElement("foreach"); foreach.addAttribute(new Attribute("collection", "list")); foreach.addAttribute(new Attribute("item", "item")); foreach.addAttribute(new Attribute("open", "(")); foreach.addAttribute(new Attribute("separator", ",")); foreach.addAttribute(new Attribute("close", ")")); foreach.addElement(new TextElement("#{item}")); listByPrimaryKey.addElement(foreach); root.addElement(listByPrimaryKey); } return super.sqlMapDocumentGenerated(document, introspectedTable); }} 生成batchInsert、listByPrimaryKey、listAll方法
listByPrimaryKey方法不支持复合主键。