使用休眠截断所有表的最佳方法？

Question

我想截断一个集成测试与另一个集成测试之间的所有数据库表。使用休眠来执行此操作的最佳方法是什么？

目前我正在这样做：

public void cleanDatabase() {
        doWithSession(new Action1<Session>() {
            @Override
            public void doSomething(Session session) {
                SQLQuery query = session.createSQLQuery("truncate table stuff");
                // todo - generify this to all tables
                query.executeUpdate();
            }
        });

（doWithSession 是一个创建和关闭会话的小包装器）。我可以使用反射迭代所有映射的对象...我想知道是否有人已经解决了这个问题。

Answer 1

我猜你可能没有使用Spring。如果你这样做了， Spring 的事务测试支持将是理想的。

简而言之：Spring 在每个测试用例之前自动启动一个事务，并在测试用例之后自动回滚它，为您留下一个空（或至少未更改）的数据库。

也许您可以模仿该机制：

在 @Before 方法中打开事务，在 @After 方法中回滚它。

Answer 2

我正是为此目的编写了一个积分器。基本上，我们连接到会话工厂创建流程，迭代 Hibernate 找到的表映射，并对每个表执行 TRUNCATE TABLE xxx。由于我们无法截断具有外键约束的表，因此在截断操作之前禁用外键检查，然后重新启用。

static final class TruncatorIntegrator implements org.hibernate.integrator.spi.Integrator {

    @Override
    public void integrate(Metadata metadata,
                          SessionFactoryImplementor sessionFactory,
                          SessionFactoryServiceRegistry serviceRegistry) {
        try (Session session = sessionFactory.openSession()) {
            session.doWork(connection -> {
                try (PreparedStatement preparedStatement = connection.prepareStatement("SET FOREIGN_KEY_CHECKS = 0;")) {
                    preparedStatement.executeUpdate();
                    System.out.printf("Disabled foreign key checks%n");
                } catch (SQLException e) {
                    System.err.printf("Cannot disable foreign key checks: %s: %s%n", e, e.getCause());
                }

                metadata.collectTableMappings().forEach(table -> {
                    String tableName = table.getQuotedName();
                    try (PreparedStatement preparedStatement = connection.prepareStatement("TRUNCATE TABLE " + tableName)) {
                        preparedStatement.executeUpdate();
                        System.out.printf("Truncated table: %s%n", tableName);
                    } catch (SQLException e) {
                        System.err.printf("Couldn't truncate table %s: %s: %s%n", tableName, e, e.getCause());
                    }
                });

                try (PreparedStatement preparedStatement = connection.prepareStatement("SET FOREIGN_KEY_CHECKS = 1;")) {
                    preparedStatement.executeUpdate();
                    System.out.printf("Enabled foreign key checks%n");
                } catch (SQLException e) {
                    System.err.printf("Cannot enable foreign key checks: %s: %s%n", e, e.getCause());
                }
            });
        }
    }

    @Override
    public void disintegrate(SessionFactoryImplementor sessionFactory,
                             SessionFactoryServiceRegistry serviceRegistry) {
    }
}

用法：我们必须在会话工厂创建流程中使用这个Integrator，并且还需要为每个测试创建一个新的会话工厂。

BootstrapServiceRegistry bootstrapServiceRegistry = new BootstrapServiceRegistryBuilder().applyIntegrator(new TruncatorIntegrator()).build();
StandardServiceRegistry registry = new StandardServiceRegistryBuilder(bootstrapServiceRegistry).build();
SessionFactory sessionFactory = new Configuration().buildSessionFactory(registry);

Answer 3

你看过内存中的mysql http://dev.mysql.com/downloads/connector/mxj /？每次测试后是否也无法回滚？我相信你可以像这样配置它。

Answer 4

您可以使用 SchemaExport，尽管这看起来相当严厉。回滚事务听起来是一个更好的主意。

Answer 5

您可以使用内存数据库并在测试之间删除完整的数据库。

如果您的测试不多但时间较长，则可以使用这种方式。

但请注意，每个数据库的行为都与其他数据库有所不同。因此，在某些情况下，使用内存数据库（例如 HyperSQL）的行为与普通数据库的行为并不完全相同 - 因此这不是正确的集成测试。