java - 如何使用 hibernate 配置截断所有表数据？

Question

如何使用 hibernate 配置截断所有表数据？

例如 hibernate.hbm2ddl.auto=validate 将检查数据库中实体模型的有效性。现在我需要一个选项来在应用程序启动期间截断所有表。

Answer 1

Hibernate 只能使用 hibernate.hbm2ddl.auto=create 或 hibernate.hbm2ddl.auto=create-drop

重新创建模式

另一种选择是在包含所需TRUNCATE 语句的类路径的根目录中创建一个import.sql。该文件在启动时自动执行(验证后)。在这种情况下，您必须使表名与数据库保持同步。

最后一个可能的解决方案是在启动时拥有自己的“截断实现”，例如 Integrator:

public class Truncator implements Integrator {
    public void integrate(final Configuration config, 
            final SessionFactoryImplementor sessionFactory,
            final SessionFactoryServiceRegistry sessionFactoryServiceRegistry) {

        // Create session
        Session session = sessionFactory.openSession();

        try {

            // Find all involved tables
            for (final ClassMetadata metadata : sessionFactory.getAllClassMetadata().values()) {
                final String tableName = ((AbstractEntityPersister) metadata).getTableName();
                if (tableName != null) {
                    session.createSQLQuery("TRUNCATE TABLE " + tableName).executeUpdate();
                }
            }

            final Map<String, CollectionMetadata> allCollectionMetadata = sessionFactory.getAllCollectionMetadata();
            for (final CollectionMetadata metadata : allCollectionMetadata.values()) {
                final String tableName = ((AbstractCollectionPersister) metadata).getTableName();
                if (tableName != null) {
                    session.createSQLQuery("TRUNCATE TABLE " + tableName).executeUpdate();
                }
            }
        } finally {
            session.close();
        }
    }
}

Integrator 必须在 META-INF/services/org.hibernate.integrator.spi.Integrator 中声明，它只包含完整的限定类名。

如果您使用增强功能(如 Envers)，您将需要找到更多表名 - 因此这更像是简单数据模型的概念证明。