如何确定Hibernate会话正在使用的DataSource？

Question

我有几个应该使用HSQLDB的单元测试,但我知道其中一些实际上正在使用物理数据库.我想在测试中添加一个检查,以确保正在使用的DataSource用于HSQLDB,而不是实时DB.

从hibernate session对象(org.hibernate.classic.Session),我如何检查DataSource

更新: 我也可以访问会话工厂(org.hibernate.impl.SessionFactory).

详细信息: Hibernate 3.2

Answer 1

无论Hibernate/Spring等的包装器和具体实现如何,您可以检查不是DataSource,而是检查数据库类型(这可能是合适的).

想法是使用DatabaseMetaData并检查它的类型(因为Hibernate检测到方言):

private boolean isTestDb(Session session) {
    return session.doReturningWork(new ReturningWork<Boolean>() {
        @Override
        public Boolean execute(Connection connection) throws SQLException {
            DatabaseMetaData metaData = connection.getMetaData();
            return metaData.getDatabaseProductName().startsWith("HSQL");
        }
    });
}

注意,可以按照您想要的方式更改方法体(检查JDBC URL,检查驱动程序名称,检查几乎所有内容).

编辑:上面的方法适用于hibernate 3.5+.

对于Hibernate早期版本(例如3.2),它可能更容易:

private boolean isTestDb(Session session) {
    Conection connection = session.connection();//deprecated method, which was dumped in hibernate 3.5+
    DatabaseMetaData metaData = connection.getMetaData();
    return metaData.getDatabaseProductName().startsWith("HSQL");
}