Mybatis-generator修改源代码实现自定义方法,返回List对象（三）

前两篇文章我们讲了如何获取源代码即创建工程、修改源代码为dao（mapper）层添加一个方法，那么这一篇，我们来讲如何在xml添加这个方法所需要sql

3、实现XML文件添加Dao(Mapper)层的实现

前面有讲过，下图中的两个包，分别是管理dao(Mapper)文件和xml文件的， 之前我们已经介绍了javamapper这个包了，而且在dao层添加了方法，那么我们现在来讲xmlmapper这个包，跟前面的很像，这个包下面有个XMLMapperGenerator类，里头有个getSqlMapElement方法，它也有很多的addxxx的方法， 而这些方法里头调用的，是org.mybatis.generator.codegen.mybatis3.xmlmapper.elements包下面类的对象

如果第二篇的对象你已经完成了，那么这里要做的，其实和第二篇差不多了， 我们先在elements包里，用SelectByPrimaryKeyElementGenerator类复制出来一个，类名叫SelectByObjectElementGenerator，这个类里的内容，我们等会儿再来修改， 先去XMLMapperGenerator类添加一个方法

//自定义方法 protected void addSelectByObject(XmlElement parentElement) { if (introspectedTable.getRules().generateSelectByPrimaryKey()) { AbstractXmlElementGenerator elementGenerator = new SelectByObjectElementGenerator(); initializeAndExecuteGenerator(elementGenerator, parentElement); } }

然后在该类的getSqlMapElement方法中调用这个方法

那么这样做完以后，理论上新生成的xml文件，就会多出一个了，但是里头面的sql依然是SelectByPrimaryKey方法的sql，接下去，我们就去修改SelectByObjectElementGenerator类的addElements方法，将其改为我们所需要的sql。

@Override public void addElements(XmlElement parentElement) { //先创建一个select标签 XmlElement answer = new XmlElement("select"); //$NON-NLS-1$ //设置该select标签的id，正式第二篇里在枚举中设置的值 answer.addAttribute(new Attribute("id", introspectedTable.getSelectByObject())); //添加parameterType String parameterType; if (introspectedTable.getRules().generateRecordWithBLOBsClass()) { parameterType = introspectedTable.getRecordWithBLOBsType(); } else { parameterType = introspectedTable.getBaseRecordType(); } answer.addAttribute(new Attribute("parameterType", parameterType)); //设置resultMap为BaseResultMap answer.addAttribute(new Attribute("resultMap",introspectedTable.getBaseResultMapId())); context.getCommentGenerator().addComment(answer); //接下去是拼接我们的sql StringBuilder sb = new StringBuilder(); sb.append("select "); Iterator<IntrospectedColumn> iter = introspectedTable.getAllColumns().iterator(); //循环插入字段名称 while (iter.hasNext()) { sb.append(MyBatis3FormattingUtilities.getSelectListPhrase(iter.next())); if (iter.hasNext()) { sb.append(", "); } if (sb.length() > 80) { answer.addElement(new TextElement(sb.toString())); sb.setLength(0); } } if (sb.length() > 0) { answer.addElement((new TextElement(sb.toString()))); } sb.setLength(0); sb.append("from "); sb.append(introspectedTable.getAliasedFullyQualifiedTableNameAtRuntime()); answer.addElement(new TextElement(sb.toString())); //在这里添加where条件 XmlElement selectTrimElement = new XmlElement("trim"); //设置trim标签 selectTrimElement.addAttribute(new Attribute("prefix", "where")); selectTrimElement.addAttribute(new Attribute("prefixOverrides", "and")); //添加where和and answer.addElement(selectTrimElement); //循环所有的列 for (IntrospectedColumn introspectedColumn : introspectedTable.getAllColumns()) { XmlElement selectNotNullElement = new XmlElement("if"); //$NON-NLS-1$ sb.setLength(0); sb.append(introspectedColumn.getJavaProperty()); sb.append(" != null "); selectNotNullElement.addAttribute(new Attribute("test", sb.toString())); sb.setLength(0); sb.append(" and "); //添加and sb.append(MyBatis3FormattingUtilities .getEscapedColumnName(introspectedColumn)); sb.append(" = "); //添加等号 sb.append(MyBatis3FormattingUtilities .getParameterClause(introspectedColumn)); selectNotNullElement.addElement(new TextElement(sb.toString())); selectTrimElement.addElement(selectNotNullElement); } // String orderByClause = introspectedTable.getTableConfigurationProperty(PropertyRegistry.TABLE_SELECT_ALL_ORDER_BY_CLAUSE); boolean hasOrderBy = StringUtility.stringHasValue(orderByClause); if (hasOrderBy) { sb.setLength(0); sb.append("order by "); //$NON-NLS-1$ sb.append(orderByClause); answer.addElement(new TextElement(sb.toString())); } if (context.getPlugins().sqlMapSelectAllElementGenerated( answer, introspectedTable)) { parentElement.addElement(answer); } }

修改完成以后，保存，执行MyTestRun，再查看xml文件，就会看到

这样，我们在使用dao（Mapper）层的SelectByObject方法的时候，就会调用这条sql了， 那么我们的目的就达到了，欢迎留言共同探讨