Browse Source

fixed gitee issues/IME37

hubin 7 years ago
parent
commit
822d3d9a52

+ 1 - 0
mybatis-plus-generator/src/main/java/com/baomidou/mybatisplus/generator/config/builder/ConfigBuilder.java

@@ -398,6 +398,7 @@ public class ConfigBuilder {
         if (null != globalConfig.getIdType()) {
         if (null != globalConfig.getIdType()) {
             // 指定需要 IdType 场景
             // 指定需要 IdType 场景
             tableInfo.getImportPackages().add(com.baomidou.mybatisplus.annotation.IdType.class.getCanonicalName());
             tableInfo.getImportPackages().add(com.baomidou.mybatisplus.annotation.IdType.class.getCanonicalName());
+            tableInfo.getImportPackages().add(com.baomidou.mybatisplus.annotation.TableId.class.getCanonicalName());
         }
         }
         if (StringUtils.isNotEmpty(strategyConfig.getVersionFieldName())) {
         if (StringUtils.isNotEmpty(strategyConfig.getVersionFieldName())) {
             tableInfo.getFields().forEach(f -> {
             tableInfo.getFields().forEach(f -> {

+ 3 - 2
mybatis-plus-generator/src/main/java/com/baomidou/mybatisplus/generator/config/po/TableInfo.java

@@ -45,9 +45,10 @@ public class TableInfo {
     private String serviceName;
     private String serviceName;
     private String serviceImplName;
     private String serviceImplName;
     private String controllerName;
     private String controllerName;
-
     private List<TableField> fields;
     private List<TableField> fields;
-    // 公共字段
+    /**
+     * 公共字段
+     */
     private List<TableField> commonFields;
     private List<TableField> commonFields;
     private Set<String> importPackages = new HashSet<>();
     private Set<String> importPackages = new HashSet<>();
     private String fieldNames;
     private String fieldNames;