上一篇我们讲了如何获取Mybatis-generator的源代码和创建工程,以及通过main方法来生成XML、实体类、mapper文件,这一篇我们来讲通过修改代码来为mapper添加一个方法
2、组合源代码中的示例,实现Dao(Mapper)层添加一个方法
结合网上的教程研究后,我知道了这两个包,分别是生成dao(mapper)文件和XML文件的JAVA类,那么我们先研究javamapper这个包
这个包里的 JavaMapperGenerator.java,有一个方法 getCompilationUnits,它是控制生成的dao文件具体有哪些方法的,代码就是
但是并不是所有的方法都生成,具体逻辑我还没有去研究过
然后我查看这个包下面的叫做 elements包,它里头有很多的类,其实就是截图中addXXX方法需要调用到的类,这些方法会实例化elements包下的类,并使用它组合出方法。
打开SelectByPrimaryKeyMethodGenerator 这个类,这就是根据ID查询数据的方法,
主要介绍这个类里的addInterfaceElements方法,它将我们写方法时候的作用范围,返回类型,方法名,参数,以及import,都专门设计了对象,下面我将注释写出来