通过实体中@TableName(“test“)找到项目中所用的表

最新推荐文章于 2025-07-29 13:21:07 发布

原创最新推荐文章于 2025-07-29 13:21:07 发布 · 542 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java

该代码示例展示了如何使用Hutool库在Java中遍历指定目录下的文件，查找以DO.java结尾的文件，提取含有@TableName注解的行，并生成一个SQL查询语句，用于查询指定数据库中的表信息。

import cn.hutool.core.io.FileUtil;
import cn.hutool.core.text.StrFormatter;
import cn.hutool.core.text.StrJoiner;

import java.io.File;
import java.io.FileFilter;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.List;

public class Test {

	public static void main(String[] args) {
		String rootPath = "E:/work/test/src/main/java/com/bjs/ceshi/entity";  //实体类路径
		
		List<File> files = FileUtil.loopFiles(FileUtil.file(rootPath), -1, new FileFilter() {
			@Override
			public boolean accept(File pathname) {
				return pathname.getName().endsWith("DO.java");   //实体类结尾名
			}
		});
		
		List<String> lines = new ArrayList<>();
		files.forEach( file -> {
			List<String> oriLines = FileUtil.readLines(file, Charset.forName("utf-8"));
			for (int i = 0; i < oriLines.size(); i++) {
				String line = oriLines.get(i);
				if(line.contains("@TableName(\"") && line.contains("\")")) {
					String tmp = line.replace("@TableName(\"", "").replace("\")", "");
					lines.add(tmp);
					System.out.println(tmp);
				}
			}
		});
		
		System.out.println();
		String sqlStr = "SELECT TABLE_NAME AS 表名,TABLE_COMMENT AS 表注释 FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = 'mall_test' and TABLE_NAME in ({});";
		if (lines.size() > 0) {
			System.out.println(StrFormatter.format(sqlStr, StrJoiner.of("','", "'", "'").append(lines)));
		}
	}
}