在 Spring+Hibernate 测试用例中重新加载数据库连接

Question

我有一套使用 Spring 的 JUnit 运行程序的系统测试，数据库配置如下所示：

<bean id="dataSource" class="org.apache.commons.dbcp.BasicDataSource">
    <property name="driverClassName" value="org.postgresql.Driver" />
    <property name="url" value="${clustercatalog.jdbc.url}" />
    <property name="username" value="${clustercatalog.jdbc.username}" />
    <property name="password" value="${clustercatalog.jdbc.password}" />
    <property name="initialSize" value="5" />
    <property name="maxActive" value="50" />
    <property name="poolPreparedStatements" value="true" />
    <property name="maxOpenPreparedStatements" value="100" />
</bean>

<bean id="sessionFactory" class="org.springframework.orm.hibernate3.annotation.AnnotationSessionFactoryBean">
    <property name="dataSource" ref="dataSource" />
    <property name="packagesToScan" value="com.xxx" />
    <property name="hibernateProperties">
        <props>
            <prop key="hibernate.dialect">org.hibernate.dialect.PostgreSQLDialect</prop>
            <prop key="hibernate.show_sql">${jdbc.show.sql}</prop>
            <prop key="hibernate.id.new_generator_mappings">true</prop>
        </props>
    </property>
    <property name="namingStrategy" ref="namingStrategy" />
</bean>

<bean id="transactionManager" class="org.springframework.orm.hibernate3.HibernateTransactionManager">
    <property name="sessionFactory" ref="sessionFactory" />
</bean>
<tx:annotation-driven />

测试用例有一个设置，其中我运行一些 bash 脚本，这些脚本从以前完成的备份中在底层 PostgreSQL 数据库上运行 pg_restore。这是因为我需要在每个测试方法之前数据库的相同状态都相同。此恢复是在用 @BeforeTransaction 注释的方法中完成的。

测试类注释为

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration({ "/systemTests-applicationContext.xml", "/applicationContext.xml" })
@TransactionConfiguration(defaultRollback = false)
@Transactional()

在测试中，当我执行使用当前休眠会话的代码时，它看不到已恢复的表。当我重新启动整个测试时，就会看到它们，但显然这就是我想要的，但它证明数据库很好，但当我执行 pg_restore 时，Spring/Hibernate 丢失了。我收到 SQLGrammarException 该表不存在。

我正在寻找一种手动重新启动与数据库的连接的方法。我怎样才能做到这一点？我应该在 sessionFactory 或某些 Spring 组件上以某种方式执行此操作吗？

Answer 1

好的，我解决了这个问题。看来 Spring 和 Hibernate 都不知道数据库中发生的更改，并且如果 DBCP 池中有任何连接，它们会被重用，然后就会发生错误，因为连接下的架构已更改。所以我想，当我重置数据库时，在测试期间戳的地方是提供给 Hibernate SessionFactory 的 javax.sql.DataSource 对象。 DBCP BasicDataSource 不具备此功能，因此我：

1. 为 dataSource bean 创建了自己的包装器。
2. 将 bean 标记为原型范围，这样我就不必自己构造它，每次我需要一个新的 bean 时，
3. 的新 bean 实例调用 applicationContext （自动装配到包装类）
4. 都会向包装类添加一个刷新方法，该方法为注入包装器 bean 到 Hibernate 的 sessionFactory
5. 自动将包装器连接到测试类，当我需要在测试中调用刷新方法

代码如下：

@Component
public class RefreshableDataSource implements DataSource {

  @Autowired
  DataSource dataSource;
  @Autowired
  ApplicationContext applicationContext;

  public void refresh() {
    dataSource = (DataSource) applicationContext.getBean("dataSource");
  }

  @Override
  public Connection getConnection() throws SQLException {
    return dataSource.getConnection();
  }

  ...other DataSource methods ...
}

applicationContext.xml：

<bean id="dataSource" scope="prototype" class="org.apache.commons.dbcp.BasicDataSource">
    <property name="driverClassName" value="org.postgresql.Driver" />
            ......
</bean>

<bean id="sessionFactory" class="org.springframework.orm.hibernate3.annotation.AnnotationSessionFactoryBean">
    <property name="dataSource" ref="refreshableDataSource" />
    <property name="packagesToScan" value="com.xxx" />
    <property name="hibernateProperties">
        <props>
            <prop key="hibernate.dialect">org.hibernate.dialect.PostgreSQLDialect</prop>
            <prop key="hibernate.show_sql">${jdbc.show.sql}</prop>
            <prop key="hibernate.id.new_generator_mappings">true</prop>
        </props>
    </property>
    <property name="namingStrategy" ref="namingStrategy" />
</bean>