Spring Boot JPA中如何复用长相似原生查询的公共部分？

方案一：使用编译时常量复用查询逻辑 #

因为@Query注解的value要求是编译时常量表达式，所以可以通过定义静态常量字符串拆分公共部分与可变部分，再拼接成完整查询语句：

public interface FooRepository extends JpaRepository<...> {
    // 定义公共查询片段
    static final String BASE_SELECT = "%s, sum(table2.b), sum(table2.c)";
    static final String BASE_FROM_JOIN = " from table2 join table1 on table2.table1_id = table1.id group by %s";

    // 按a1分组的查询
    @Query(value = BASE_SELECT.replace("%s", "table1.a1") + BASE_FROM_JOIN.replace("%s", "table1.a1"), nativeQuery = true)
    List<FooDTO> getAggregateSumWithGroupByA1();

    // 按a2分组的查询
    @Query(value = BASE_SELECT.replace("%s", "table1.a2") + BASE_FROM_JOIN.replace("%s", "table1.a2"), nativeQuery = true)
    List<FooDTO> getAggregateSumWithGroupByA2();
}

也可以直接预定义每个查询的完整常量，更直观：

public interface FooRepository extends JpaRepository<...> {
    static final String GROUP_BY_A1 = "table1.a1";
    static final String QUERY_A1 = "select " + GROUP_BY_A1 + ", sum(table2.b), sum(table2.c) from table2 join table1 on table2.table1_id = table1.id group by " + GROUP_BY_A1;

    static final String GROUP_BY_A2 = "table1.a2";
    static final String QUERY_A2 = "select " + GROUP_BY_A2 + ", sum(table2.b), sum(table2.c) from table2 join table1 on table2.table1_id = table1.id group by " + GROUP_BY_A2;

    @Query(value = QUERY_A1, nativeQuery = true)
    List<FooDTO> getAggregateSumWithGroupByA1();

    @Query(value = QUERY_A2, nativeQuery = true)
    List<FooDTO> getAggregateSumWithGroupByA2();
}

这种方式完全符合注解对常量表达式的要求，同时实现了代码复用。

方案二：自定义Repository实现（动态构建SQL） #

如果需要支持更多分组列，可通过自定义Repository实现类动态生成SQL：

1. 定义自定义接口

public interface FooCustomRepository {
    List<FooDTO> getAggregateSumWithGroupBy(String groupByColumn);
}

2. 实现自定义接口

@Repository
public class FooCustomRepositoryImpl implements FooCustomRepository {
    private final EntityManager entityManager;

    public FooCustomRepositoryImpl(EntityManager entityManager) {
        this.entityManager = entityManager;
    }

    @Override
    public List<FooDTO> getAggregateSumWithGroupBy(String groupByColumn) {
        // 安全校验：仅允许合法列名，防止SQL注入
        if (!List.of("a1", "a2").contains(groupByColumn)) {
            throw new IllegalArgumentException("不支持的分组列");
        }
        // 动态构建SQL
        String sql = String.format(
            "select table1.%s, sum(table2.b), sum(table2.c) from table2 join table1 on table2.table1_id = table1.id group by table1.%s",
            groupByColumn, groupByColumn
        );
        // 执行查询并映射到DTO
        Query query = entityManager.createNativeQuery(sql, FooDTO.class);
        return query.getResultList();
    }
}

3. 让主Repository继承自定义接口

public interface FooRepository extends JpaRepository<FooEntity, Long>, FooCustomRepository {
    // 原有方法
}

使用时直接调用fooRepository.getAggregateSumWithGroupBy("a1")或"a2"即可，扩展性更强。

注意事项 #

• 动态拼接列名时必须做安全校验，只允许传入预定义的合法列名，避免SQL注入风险。
• 如果使用DTO映射，需确保FooDTO字段与查询结果的列顺序/名称匹配，或通过@SqlResultSetMapping定义映射规则。

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