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了，
那么我们的目的就达到了，欢迎留言共同探讨