修复强制生成字段注解的配置无法生效的问题

davidfantasy · Apr 17, 2021 · 8af78d9 · 8af78d9
1 parent 287e658
commit 8af78d9
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/...ain/java/com/github/davidfantasy/mybatisplus/generatorui/service/MbpGeneratorService.java b/...ain/java/com/github/davidfantasy/mybatisplus/generatorui/service/MbpGeneratorService.java
@@ -131,6 +131,13 @@ private void genCode(AutoGenerator ag, GenSetting genSetting, UserConfig userCon
                 focList.add(new FileOutConfig(outputFileInfo.getAvailableTemplatePath()) {
                     @Override
                     public String outputFile(TableInfo tableInfo) {
+                        //避免使用了NameConverter导致entityTableFieldAnnotationEnable属性失效的问题，在此强制进行设置
+                        if (userConfig.getEntityStrategy().isEntityTableFieldAnnotationEnable()) {
+                            tableInfo.getFields().forEach(field -> {
+                                field.setConvert(true);
+                            });
+                            tableInfo.getImportPackages().add("com.baomidou.mybatisplus.annotation.TableField");
+                        }
                         return filePath;
                     }
                 });