浏览代码

去掉多余导入

青苗 9 年之前
父节点
当前提交
37fb04ee1a
共有 1 个文件被更改,包括 4 次插入2 次删除
  1. 4 2
      mybatis-plus/src/main/java/com/baomidou/mybatisplus/generator/AutoGenerator.java

+ 4 - 2
mybatis-plus/src/main/java/com/baomidou/mybatisplus/generator/AutoGenerator.java

@@ -523,8 +523,10 @@ public class AutoGenerator {
 		bw.newLine();
 		bw.newLine();
 		bw.write("import com.baomidou.mybatisplus.annotations.TableId;");
 		bw.write("import com.baomidou.mybatisplus.annotations.TableId;");
 		bw.newLine();
 		bw.newLine();
-		bw.write("import com.baomidou.mybatisplus.annotations.TableName;");
-		bw.newLine();
+		if (table.contains("_")) {
+			bw.write("import com.baomidou.mybatisplus.annotations.TableName;");
+			bw.newLine();
+		}
 		bw = buildClassComment(bw, tableComment);
 		bw = buildClassComment(bw, tableComment);
 		bw.newLine();
 		bw.newLine();
 		/* 包含下划线注解 */
 		/* 包含下划线注解 */