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

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

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

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

详细信息：Hibernate 3.2

解决方法

无论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");
}