Browse Source

修复自动填充代码生成错误

Caratacus 8 years ago
parent
commit
ae184df155

+ 5 - 6
src/main/java/com/baomidou/mybatisplus/generator/config/builder/ConfigBuilder.java

@@ -27,7 +27,6 @@ import java.util.List;
 import java.util.Map;
 import java.util.Set;
 
-import com.baomidou.mybatisplus.enums.FieldFill;
 import com.baomidou.mybatisplus.generator.config.ConstVal;
 import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
 import com.baomidou.mybatisplus.generator.config.GlobalConfig;
@@ -506,11 +505,11 @@ public class ConfigBuilder {
                     continue;
                 }
                 // 填充逻辑判断
-                List<TableFill> tfs = this.getStrategyConfig().getTableFillList();
-                if (null != tfs) {
-                    for (TableFill tf : tfs) {
-                        if (tf.getFieldName().equals(field.getName())) {
-                            field.setFill(FieldFill.INSERT.name());
+                List<TableFill> tableFillList = this.getStrategyConfig().getTableFillList();
+                if (null != tableFillList) {
+                    for (TableFill tableFill : tableFillList) {
+                        if (tableFill.getFieldName().equals(field.getName())) {
+                            field.setFill(tableFill.getFieldFill().name());
                             break;
                         }
                     }

+ 6 - 6
src/main/java/com/baomidou/mybatisplus/generator/config/po/TableFill.java

@@ -30,7 +30,7 @@ public class TableFill {
     /* 字段名称 */
     private String fieldName;
     /* 忽略类型 */
-    private FieldFill ignore;
+    private FieldFill fieldFill;
 
     private TableFill() {
         // to do nothing
@@ -38,7 +38,7 @@ public class TableFill {
 
     public TableFill(String fieldName, FieldFill ignore) {
         this.fieldName = fieldName;
-        this.ignore = ignore;
+        this.fieldFill = ignore;
     }
 
     public String getFieldName() {
@@ -49,11 +49,11 @@ public class TableFill {
         this.fieldName = fieldName;
     }
 
-    public FieldFill getIgnore() {
-        return ignore;
+    public FieldFill getFieldFill() {
+        return fieldFill;
     }
 
-    public void setIgnore(FieldFill ignore) {
-        this.ignore = ignore;
+    public void setFieldFill(FieldFill fieldFill) {
+        this.fieldFill = fieldFill;
     }
 }