Hibernate排序查询适配YDB：别名配置方案咨询

通过JpaRepository或CriteriaQuery创建带排序的查询时，Hibernate生成的SQL如下：

select
    table0_.sys_crt_dttm as sys_crt_4_44_
from
    `table` table0_ 
where
    table0_.sys_state=? 
order by
    table0_.sys_crt_dttm asc

该SQL能在PostgreSQL正常执行，但无法在YDB中运行——YDB要求排序子句必须使用列别名。需要生成以下两种SQL之一：

方案一（使用别名排序）：

select
    table0_.sys_crt_dttm as sys_crt_4_44_
from
    `table` table0_ 
where
    table0_.sys_state=? 
order by
    sys_crt_4_44_ asc

方案二（禁用列别名）：

select
    table0_.sys_crt_dttm
from
    `table` table0_ 
where
    table0_.sys_state=? 
order by
    table0_.sys_crt_dttm asc

尝试过用SqlInterceptor通过正则sql.replace("(as\\s)([\\w_\\d]+)".toRegex(), "")移除别名，但会破坏部分查询，推测Hibernate内部依赖这些别名完成结果映射。请问是否可以通过配置Hibernate实现上述两种需求？

1. 让Hibernate在ORDER BY中使用列别名 #

要实现这一点，需要自定义Hibernate方言，重写排序子句的生成逻辑，使其优先使用列别名而非原始列名：

1. 创建自定义方言类（若YDB无官方方言，可继承PostgreSQLDialect，因为YDB兼容部分PostgreSQL语法）：

import org.hibernate.dialect.PostgreSQLDialect;
import org.hibernate.sql.ast.spi.SqlAstTranslator;
import org.hibernate.sql.ast.spi.StandardSqlAstTranslatorFactory;
import org.hibernate.sql.ast.tree.select.SelectStatement;

public class YdbDialect extends PostgreSQLDialect {
    @Override
    public StandardSqlAstTranslatorFactory getSqlAstTranslatorFactory() {
        return new StandardSqlAstTranslatorFactory() {
            @Override
            protected <T extends SelectStatement> SqlAstTranslator<T> buildTranslator(
                    SessionFactoryImplementor sessionFactory,
                    Statement statement) {
                return new YdbSqlAstTranslator<>(sessionFactory, (SelectStatement) statement);
            }
        };
    }
}

2. 自定义SqlAstTranslator，修改ORDER BY的生成逻辑：

import org.hibernate.sql.ast.spi.SqlAstTranslator;
import org.hibernate.sql.ast.tree.expression.ColumnReference;
import org.hibernate.sql.ast.tree.expression.Expression;
import org.hibernate.sql.ast.tree.select.SelectClause;
import org.hibernate.sql.ast.tree.select.SelectItem;

import java.util.Optional;

public class YdbSqlAstTranslator<T extends SelectStatement> extends SqlAstTranslator<T> {
    public YdbSqlAstTranslator(SessionFactoryImplementor sessionFactory, T statement) {
        super(sessionFactory, statement);
    }

    @Override
    protected void visitSortSpecification(SortSpecification sortSpec) {
        final Expression expression = sortSpec.getExpression();
        // 检查当前排序表达式是否是查询选择项中的列，若是则使用其别名
        if (expression instanceof ColumnReference) {
            SelectClause selectClause = getStatement().getSelectClause();
            Optional<SelectItem> matchingItem = selectClause.getSelectItems().stream()
                    .filter(item -> item.getExpression() instanceof ColumnReference)
                    .filter(item -> ((ColumnReference) item.getExpression()).getColumnReference().equals(
                            ((ColumnReference) expression).getColumnReference()))
                    .findFirst();
            if (matchingItem.isPresent()) {
                matchingItem.get().getAlias().ifPresent(this::appendSql);
                appendSortOrder(sortSpec.getSortOrder());
                return;
            }
        }
        // 不满足条件时使用默认逻辑
        super.visitSortSpecification(sortSpec);
    }
}

3. 在配置文件中指定使用自定义方言：

# application.properties
hibernate.dialect=com.yourpackage.YdbDialect

2. 禁用Hibernate生成列别名 #

可以通过Hibernate的全局配置直接关闭自动列别名生成：

在application.properties中添加：

hibernate.use_column_aliases=false

或者在hibernate.cfg.xml中配置：

<property name="hibernate.use_column_aliases">false</property>

该配置会让Hibernate在SELECT子句中直接输出原始列名，不生成类似sys_crt_4_44_的别名，此时排序子句使用原始列名即可在YDB中正常执行。

注意：该配置全局生效，需确保实体类的字段映射不依赖自动生成的别名。

关于SqlInterceptor的问题 #

直接用正则移除别名会破坏Hibernate的结果映射逻辑——Hibernate在将查询结果转换为实体对象时，依赖这些自动生成的别名匹配实体字段，强行移除会导致字段映射失败，出现Unknown column或结果为空的问题，因此不推荐这种方式。

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