소스 검색

去掉多余导入

青苗 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.write("import com.baomidou.mybatisplus.annotations.TableId;");
 		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.newLine();
 		/* 包含下划线注解 */