SpringBoot+Hibernate适配Databricks SparkSQL的方言及语法问题

核心问题根源 #

Hibernate使用MySQLDialect时会自动给表名、字段名添加反引号（`），但SparkSQL不支持这种语法，同时官方暂未提供SparkSQL专属方言。

方案1：自定义Hibernate方言（推荐） #

直接继承基础方言，重写标识符引号处理逻辑，同时适配SparkSQL的语法特性：

1. 创建自定义方言类：

import org.hibernate.dialect.Dialect;
import org.hibernate.dialect.function.StandardSQLFunction;
import org.hibernate.type.StandardBasicTypes;

public class SparkSQLDialect extends Dialect {
    public SparkSQLDialect() {
        super();
        // 注册SparkSQL支持的常用函数（按需添加）
        registerFunction("concat", new StandardSQLFunction("concat", StandardBasicTypes.STRING));
    }

    @Override
    public String getIdentifierQuoteString() {
        return ""; // 关键：取消引号生成，彻底解决反引号问题
    }

    @Override
    public boolean supportsIdentityColumns() {
        return false; // SparkSQL不支持IDENTITY自增主键语法，按需调整
    }
}

2. 修改JPA配置，替换方言：

jpa:
  show-sql: true
  hibernate:
    ddlAuto: none
    naming:
      implicit-strategy: org.hibernate.boot.model.naming.ImplicitNamingStrategyLegacyHbmImpl
      physical-strategy: org.springframework.boot.orm.jpa.hibernate.SpringPhysicalNamingStrategy
  properties:
    hibernate:
      dialect: com.yourpackage.SparkSQLDialect # 替换为自定义方言的实际路径
      default_schema: catalog_name.schema_name # 配置Databricks三级命名空间的catalog+schema
  database: default

方案2：SQL语句拦截器（临时应急） #

如果自定义方言成本较高，可通过Hibernate拦截器修改生成的SQL，批量移除反引号：

1. 创建拦截器类：

import org.hibernate.EmptyInterceptor;

public class SparkSQLInterceptor extends EmptyInterceptor {
    @Override
    public String onPrepareStatement(String sql) {
        // 全局移除所有反引号
        return sql.replace("`", "");
    }
}

2. 添加到JPA配置：

jpa:
  properties:
    hibernate:
      session_factory:
        interceptor: com.yourpackage.SparkSQLInterceptor

注意：此方案属于临时hack，可能会破坏特殊标识符的转义逻辑，仅建议应急使用。

额外适配要点 #

• 三级命名空间：除了配置default_schema，也可在实体类的@Table注解中显式指定完整命名空间：

@Entity
@Table(catalog = "catalog_name", schema = "schema_name", name = "table_name")
public class User { ... }

• MySQL特有语法替换：检查实体类中的MySQL专属注解（如@GeneratedValue(strategy = GenerationType.IDENTITY)），替换为SparkSQL支持的主键生成方式（比如手动生成UUID或自增序列）。

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