如何使用Hibernate拦截器修改表创建时的DDL？

1. 自定义Dialect扩展（最常用方案） #

Hibernate的Dialect负责生成对应数据库的DDL语法，你可以继承目标数据库的Dialect类，重写生成DDL的核心方法，直接修改输出的语句片段：

• 继承对应数据库的Dialect（比如MySQL8Dialect、PostgreSQLDialect）
• 重写getCreateTableString()、getColumnType()、getAddColumnString()等方法，调整生成的DDL内容
• 在Hibernate配置中指定自定义Dialect

示例代码：

public class CustomMySQLDialect extends MySQL8Dialect {
    // 重写CREATE TABLE语句，默认添加引擎和字符集
    @Override
    public String getCreateTableString() {
        return super.getCreateTableString() + " ENGINE=InnoDB DEFAULT CHARSET=utf8mb4";
    }

    // 自定义VARCHAR类型的列定义
    @Override
    public String getColumnType(int sqlTypeCode) {
        if (sqlTypeCode == Types.VARCHAR) {
            return "VARCHAR(255) COLLATE utf8mb4_unicode_ci";
        }
        return super.getColumnType(sqlTypeCode);
    }
}

配置文件中指定自定义Dialect：

hibernate.dialect=com.yourpackage.CustomMySQLDialect

2. 手动处理SchemaExport生成的DDL #

通过Hibernate的SchemaExport工具生成DDL语句后，手动修改再执行，适合需要精细调整特定语句的场景：

• 构建Metadata实例获取实体映射信息
• 用SchemaExport生成原始DDL语句集合
• 遍历集合修改目标语句
• 执行修改后的DDL

示例代码：

Metadata metadata = new MetadataSources(serviceRegistry)
    .addAnnotatedClass(User.class)
    .buildMetadata();

SchemaExport export = new SchemaExport();
List<String> ddlStatements = new ArrayList<>();
// 捕获生成的DDL语句
export.setOutput(ddlStatements::add);
export.create(EnumSet.of(TargetType.STDOUT), metadata);

// 修改指定DDL语句
for (int i = 0; i < ddlStatements.size(); i++) {
    String original = ddlStatements.get(i);
    if (original.contains("CREATE TABLE `user`")) {
        // 给user表添加自定义注释
        ddlStatements.set(i, original.replace(";", " COMMENT '用户信息表';"));
    }
}

// 执行修改后的DDL
Session session = sessionFactory.openSession();
Transaction tx = session.beginTransaction();
for (String ddl : ddlStatements) {
    session.createNativeQuery(ddl).executeUpdate();
}
tx.commit();
session.close();

3. 通过Integrator拦截Schema生成流程 #

实现Hibernate的Integrator接口，介入Schema管理的生命周期，替换默认的Schema生成工具，全局修改DDL：

• 实现Integrator接口，在integrate()方法中替换自定义的SchemaManagementTool
• 重写Schema工具类的SQL执行方法，修改语句后再执行

示例代码：

public class CustomSchemaIntegrator implements Integrator {
    @Override
    public void integrate(Metadata metadata, SessionFactoryImplementor sessionFactory, SessionFactoryServiceRegistry serviceRegistry) {
        SchemaManagementTool tool = serviceRegistry.getService(SchemaManagementTool.class);
        if (tool instanceof SchemaManagementToolImpl) {
            ((SchemaManagementToolImpl) tool).setDatabaseSchemaCreator(new CustomSchemaCreator());
        }
    }

    // 实现其他接口方法（空实现即可）
    @Override
    public void disintegrate(SessionFactoryImplementor sessionFactory, SessionFactoryServiceRegistry serviceRegistry) {}
}

class CustomSchemaCreator extends StandardDatabaseSchemaCreator {
    @Override
    protected void executeSqlString(SqlStringGenerationContext context, String sql) {
        // 全局修改DDL，比如给所有表添加前缀
        String modifiedSql = sql.replace("CREATE TABLE ", "CREATE TABLE `t_`");
        super.executeSqlString(context, modifiedSql);
    }
}

配置文件中注册Integrator：

hibernate.integrator_provider=com.yourpackage.CustomSchemaIntegrator

内容的提问来源于stack exchange，提问作者MiguelAngel_LV