소스 검색

!48 StrategyConfig增加fieldPrefix字段前缀处理
Merge pull request !48 from tianduo4/master

青苗 7 년 전
부모
커밋
2d782af8e3

+ 15 - 2
mybatis-plus-generate/src/main/java/com/baomidou/mybatisplus/generator/config/StrategyConfig.java

@@ -15,12 +15,12 @@
  */
 package com.baomidou.mybatisplus.generator.config;
 
-import java.util.List;
-
 import com.baomidou.mybatisplus.generator.config.po.TableFill;
 import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
 import com.baomidou.mybatisplus.toolkit.StringUtils;
 
+import java.util.List;
+
 /**
  * <p>
  * 策略配置项
@@ -51,6 +51,11 @@ public class StrategyConfig {
      */
     private String[] tablePrefix;
 
+    /**
+     * 字段前缀
+     */
+    private String[] fieldPrefix;
+
     /**
      * 自定义继承的Entity类全称,带包名
      */
@@ -386,4 +391,12 @@ public class StrategyConfig {
         this.tableFillList = tableFillList;
         return this;
     }
+
+    public String[] getFieldPrefix() {
+        return fieldPrefix;
+    }
+
+    public void setFieldPrefix(String[] fieldPrefix) {
+        this.fieldPrefix = fieldPrefix;
+    }
 }

+ 9 - 19
mybatis-plus-generate/src/main/java/com/baomidou/mybatisplus/generator/config/builder/ConfigBuilder.java

@@ -15,24 +15,7 @@
  */
 package com.baomidou.mybatisplus.generator.config.builder;
 
-import java.io.File;
-import java.sql.Connection;
-import java.sql.PreparedStatement;
-import java.sql.ResultSet;
-import java.sql.SQLException;
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.HashSet;
-import java.util.List;
-import java.util.Map;
-import java.util.Set;
-
-import com.baomidou.mybatisplus.generator.config.ConstVal;
-import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
-import com.baomidou.mybatisplus.generator.config.GlobalConfig;
-import com.baomidou.mybatisplus.generator.config.PackageConfig;
-import com.baomidou.mybatisplus.generator.config.StrategyConfig;
-import com.baomidou.mybatisplus.generator.config.TemplateConfig;
+import com.baomidou.mybatisplus.generator.config.*;
 import com.baomidou.mybatisplus.generator.config.po.TableField;
 import com.baomidou.mybatisplus.generator.config.po.TableFill;
 import com.baomidou.mybatisplus.generator.config.po.TableInfo;
@@ -41,6 +24,13 @@ import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
 import com.baomidou.mybatisplus.generator.config.rules.QuerySQL;
 import com.baomidou.mybatisplus.toolkit.StringUtils;
 
+import java.io.File;
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.*;
+
 /**
  * <p>
  * 配置汇总 传递给文件生成工具
@@ -598,7 +588,7 @@ public class ConfigBuilder {
      * @return 根据策略返回处理后的名称
      */
     private String processName(String name, NamingStrategy strategy) {
-        return processName(name, strategy, null);
+        return processName(name, strategy, this.strategyConfig.getFieldPrefix());
     }
 
     /**