浏览代码

code style

miemie 5 年之前
父节点
当前提交
ebbce007f4

+ 1 - 2
mybatis-plus-extension/src/main/java/com/baomidou/mybatisplus/extension/injector/methods/InsertBatchSomeColumn.java

@@ -21,7 +21,6 @@ import com.baomidou.mybatisplus.core.injector.AbstractMethod;
 import com.baomidou.mybatisplus.core.metadata.TableFieldInfo;
 import com.baomidou.mybatisplus.core.metadata.TableInfo;
 import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
-import com.baomidou.mybatisplus.core.toolkit.StringUtils;
 import com.baomidou.mybatisplus.core.toolkit.sql.SqlScriptUtils;
 import lombok.AllArgsConstructor;
 import lombok.NoArgsConstructor;
@@ -88,7 +87,7 @@ public class InsertBatchSomeColumn extends AbstractMethod {
         String keyProperty = null;
         String keyColumn = null;
         // 表包含主键处理逻辑,如果不包含主键当普通字段处理
-        if (StringUtils.isNotBlank(tableInfo.getKeyProperty())) {
+        if (tableInfo.havePK()) {
             if (tableInfo.getIdType() == IdType.AUTO) {
                 /* 自增主键 */
                 keyGenerator = new Jdbc3KeyGenerator();