spring boot hibernate 获取表名

JAVA herman 66浏览 0评论

这是一个很扯淡的需求,我现在有几十行数据,需要通过 Hibernate 保存到数据库。这些数据中包含的有各个记录对应的表信息,我们需要先把数据中的内容根据表信息转换成对应的实体对象,然后通过 Hibernate 存储到数据库中。

这个表名和数据都是会变的,大约有十几个这样的数据。每次数据行数和表可能都是不同的。

其实这个功能也没什么逻辑,重点是要根据表名关联到对应的实体类。在网上搜索了一些资料,发现 Hibernate 中的 SessionFactory 提供了两个重要的方法。

unwrap 和 getAllClassMetadata。我们正式根据这两个函数实现这个神奇的功能的。

关键代码如下:

@Bean
public Map metaMap(EntityManagerFactory factory) throws ClassNotFoundException {
	if (factory.unwrap(SessionFactory.class) == null) {
		throw new NullPointerException("factory is not a hibernate factory");
	}
	SessionFactory sessionFactory = factory.unwrap(SessionFactory.class);
	 Map<String, ClassMetadata> metaMap = sessionFactory.getAllClassMetadata();
	Map<String, Class> map = new HashMap<>();
	for (String key : metaMap.keySet()) {
		AbstractEntityPersister classMetadata = (AbstractEntityPersister) metaMap
				.get(key);
		String tableName = classMetadata.getTableName().toLowerCase();
		int index = tableName.indexOf(".");
		if (index >= 0) {
			tableName = tableName.substring(index + 1);
		}
		map.put(tableName, Class.forName(key));
	}
	return map;
}

这样在 map 中就可以根据表名获取到对应的实体类了。

参考资料

业余草公众号

最后,欢迎关注我的个人微信公众号:业余草(yyucao)!

本文原文出处:业余草: » spring boot hibernate 获取表名