【Mybatis】Mybatis generator如何修改Mapper

.contextGenerateAdditionalXmlFiles());

}

然后进入introspectedTable.getGeneratedJavaFiles()方法

---

@Override

public List getGeneratedJavaFiles() {

List answer = new ArrayList();

//javaModelGenerators 存的是 JavaModel 和 JavaModelExample 类

for (AbstractJavaGenerator javaGenerator : javaModelGenerators) {

//这一行才是重点，因为所有的准备数据都是在这个方法里面

List compilationUnits = javaGenerator

.getCompilationUnits();

for (CompilationUnit compilationUnit : compilationUnits) {

GeneratedJavaFile gjf = new GeneratedJavaFile(compilationUnit,

context.getJavaModelGeneratorConfiguration()

.getTargetProject(),

context.getProperty(PropertyRegistry.CONTEXT_JAVA_FILE_ENCODING),

context.getJavaFormatter());

answer.add(gjf);

}

}

// clientGenerators 然后javaModelGenerators 存的是 JavaMapper.java文件

for (AbstractJavaGenerator javaGenerator : clientGenerators) {

//这一行才是重点，因为所有的准备数据都是在这个方法里面

List compilationUnits = javaGenerator

.getCompilationUnits();

for (CompilationUnit compilationUnit : compilationUnits) {

GeneratedJavaFile gjf = new GeneratedJavaFile(compilationUnit,

context.getJavaClientGeneratorConfiguration()

.getTargetProject(),

context.getProperty(PropertyRegistry.CONTEXT_JAVA_FILE_ENCODING),

context.getJavaFormatter());

answer.add(gjf);

}

}

return answer;

}

重点方法：javaGenerator.getCompilationUnits();

---

这个方法是真正填充数据的地方

AbstractJavaGenerator 这个是抽象类，主要是用来生成Java文件的 下面有很多实现类；

比如生成

JavaModel 文件的BaseRecordGenerator

JavaModelExample文件的ExampleGenerator

Mapper.java文件的JavaMapperGenerator

这个实现类都实现了getCompilationUnits方法;这些方法都在为即将生成的文件组装数据

我们看一下JavaMapperGenerator 中的实现

@Override

public List getCompilationUnits() {

progressCallback.startTask(getString(“Progress.17”, //$NON-NLS-1$

introspectedTable.getFullyQualifiedTable().toString()));

CommentGenerator commentGenerator = context.getCommentGenerator();

FullyQualifiedJavaType type = new FullyQualifiedJavaType(

introspectedTable.getMyBatis3JavaMapperType());

Interface interfaze = new Interface(type);

interfaze.setVisibility(JavaVisibility.PUBLIC);

//看到这里喜出望外，这里就是扩展点了；因为它把inerfaze给传进去了，那我们可以在这里做一些我们想做的事情

commentGenerator.addJavaFileComment(interfaze);

//省略无关…

修改Mapper.java文件

---

在前几篇文章中我们已经创建了CommentGenerator对象了，那我们可以在这里面来做扩展

@Override

public void addJavaFileComment(CompilationUnit compilationUnit) {

//生成的是 JavaModel 和 JavaModelExample 文件

if(compilationUnit instanceof TopLevelClass){

//这里可以修改 JavaModel 和 JavaModelExample 文件

/*TopLevelClass topLevelClass = (TopLevelClass)compilationUnit;

String shortName = compilationUnit.getType().getShortName();

topLevelClass.addAnnotation(“@Resource”);

topLevelClass.addImportedType(“javax.annotation.Resource”);*/

}

//生成的是Mapper.java 文件

if(compilationUnit instanceof Interface){

Interface anInterface = (Interface)compilationUnit;

//下面的可以给JavaFile 添加注释

//topLevelClass.addFileCommentLine(“/generator by Shirc generator common…/”);

String shortName = compilationUnit.getType().getShortName();

if(shortName!=null||shortName.endsWith(“Mapper”))return;

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Java工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年Java开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，真正体系化！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！

如果你觉得这些内容对你有帮助，可以扫码获取！！（备注Java获取）

最后

面试是跳槽涨薪最直接有效的方式，马上金九银十来了，各位做好面试造飞机，工作拧螺丝的准备了吗？

掌握了这些知识点，面试时在候选人中又可以夺目不少，暴击9999点。机会都是留给有准备的人，只有充足的准备，才可能让自己可以在候选人中脱颖而出。

《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！
的准备，才可能让自己可以在候选人中脱颖而出。

[外链图片转存中…(img-DtlcVYmr-1713532485075)]

[外链图片转存中…(img-FukMnwal-1713532485078)]

《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取！