5

我上一个问题的后续问题：Generate an SQL DB creation script with Hibernate 4

目标是让命令行工具能够生成具有给定持久性单元的 SQL 模式的文件（类似于 Hibernate 工具中的 hibernatetool-hbm2ddl Ant 任务）。

根据我之前问题的答案，这可以通过org.hibernate.tool.hbm2ddl.SchemaExport.

我不想将所有实体添加到Configuration（如上一个答案中所建议的）我想指定一个PersistenceUnit.

是否可以将持久性单元添加到 Hibernate Configuration？

就像是

Properties properties = new Properties();
properties.put( "hibernate.dialect", "org.hibernate.dialect.PostgreSQLDialect" );
...
EntityManagerFactory entityManagerFactory =
    Persistence.createEntityManagerFactory( "persistentUnitName", properties );
Configuration configuration = new Configuration();

... missing part ...

SchemaExport schemaExport = new SchemaExport( configuration );
schemaExport.setOutputFile( "schema.sql" );
...

根据评论中的要求编辑示例persistence.xml。每个类都有注释@Entity

<persistence
    xmlns="http://java.sun.com/xml/ns/persistence"
    xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
    xsi:schemaLocation="http://java.sun.com/xml/ns/persistence http://java.sun.com/xml/ns/persistence/persistence_1_0.xsd"
    version="1.0"
>

    <persistence-unit
        name="doiPersistenceUnit"
        transaction-type="JTA"
    >

        <provider>org.hibernate.ejb.HibernatePersistence</provider>
        <jta-data-source>jdbc/doi</jta-data-source>


        <class>ch.ethz.id.wai.doi.bo.Doi</class>
        [...]
        <class>ch.ethz.id.wai.doi.bo.DoiPool</class>

        <exclude-unlisted-classes>true</exclude-unlisted-classes>

        <properties>
            <property name="hibernate.show_sql"                     value="false" />
            <property name="hibernate.format_sql"                   value="false" />
            <property name="hibernate.connection.characterEncoding" value="utf8" />
            <property name="hibernate.connection.charSet"           value="utf8" />
        </properties>

    </persistence-unit>

</persistence>
4

1 回答 1

6

好吧，如果您的类是通过 xml 映射 ( s) 映射的 - 您可以使用和hbm将包含 xmls 的文档或 jar 文件直接添加到Configuration实例中。config.addJar(myJarFile)config.add(myXmlFile)

但是，如果您希望扫描带注释的类 - 我知道通过 Hibernate 没有这样直接的选项（addPackage添加元数据而不是类）。

您可以实现自己的扫描逻辑并添加所有带注释的类config.addAnnotatedClass(myAnnotatedClass)（或者可能根据您知道的包含 ORM 类的特定包执行此操作，从而可能节省一些时间）。

更新 2

哦，更好的是，您可以ManagedType通过以下方式迭代持久性单元的 s getManagedTypes()：

EntityManagerFactory entityManagerFactory =
    Persistence.createEntityManagerFactory( unitName, config.getProperties() );
final Set<ManagedType<?>> managedTypes =
    entityManagerFactory.getMetamodel().getManagedTypes();
    for ( ManagedType<?> managedType : managedTypes ) {
    final Class<?> javaType = managedType.getJavaType();
    config.addAnnotatedClass( javaType );
}

更新

您可以通过检查相关来确定PersistenceUnit每个Entity-无需解析 xmlEntityManagerFactory -

Class aClass = ... // get the class from your scanning
EntityManagerFactory entityManagerFactory =
    Persistence.createEntityManagerFactory( unitName, config.getProperties() );
ManagedType<?> managedType = null;
try {
    managedType = entityManagerFactory.getMetamodel().managedType( aClass );
} catch ( IllegalArgumentException e ) {
    // happens when aClass isn't a type managed by the persistence unit
}
if ( managedType != null ) {
    config.addAnnotatedClass( aClass );
}

确保Configuration为每个持久性单元使用不同的实例。否则，带注释的类只会累积，DDL 也会累积。

我试过了，效果很好——为两个不同的持久性单元打印了两个不同的 DDL。

于 2013-01-07T09:12:22.380 回答