能否通过Spring Data Repository按IN参数中的ID顺序排序实体？

当然可以实现这个需求！默认情况下，数据库执行IN查询返回的结果顺序是不确定的（取决于索引、存储结构等），所以我们需要手动指定排序逻辑，这里有几种实用的方案：

不同数据库有专门的函数可以帮我们按传入的ID列表顺序排序，比如：

MySQL/MariaDB 使用FIELD()函数 #

直接在JPQL查询里调用数据库的FIELD函数，指定排序规则：

@Repository
public interface EmployeeRepository extends JpaRepository<Employee, Long> {
    @Query("SELECT e FROM Employee e WHERE e.id IN (:employeeIds) ORDER BY FIELD(e.id, :employeeIds)")
    List<Employee> findByIdInOrdered(@Param("employeeIds") List<Long> employeeIds);
}

PostgreSQL 使用array_position()函数 #

PostgreSQL可以借助数组函数来实现：

@Repository
public interface EmployeeRepository extends JpaRepository<Employee, Long> {
    @Query("SELECT e FROM Employee e WHERE e.id IN (:employeeIds) ORDER BY array_position(array[:employeeIds], e.id)")
    List<Employee> findByIdInOrdered(@Param("employeeIds") List<Long> employeeIds);
}

这种方式的优点是排序在数据库层面完成，性能更好，适合数据量较大的场景；缺点是和数据库强耦合，如果后续切换数据库需要修改查询语句。

如果不想依赖数据库特定语法，或者数据量不大，可以先查询出所有符合条件的实体，再在内存中按照传入的ID列表顺序重新排序：

首先定义基础的查询方法：

@Repository
public interface EmployeeRepository extends JpaRepository<Employee, Long> {
    List<Employee> findByIdIn(List<Long> employeeIds);
}

然后在业务代码中进行排序：

// 先查询数据
List<Employee> employees = employeeRepository.findByIdIn(employeeIds);

// 为了提升排序效率，先把ID列表转成ID->索引的Map（避免多次调用indexOf的O(n)开销）
Map<Long, Integer> idOrderMap = IntStream.range(0, employeeIds.size())
    .boxed()
    .collect(Collectors.toMap(employeeIds::get, Function.identity()));

// 根据Map中的索引排序
employees.sort(Comparator.comparingInt(emp -> idOrderMap.get(emp.getId())));

这种方式的优点是完全通用，不依赖任何数据库特性；缺点是如果返回的实体数量很多，内存排序会有性能损耗。

• 如果使用方案1，要确保你的数据库支持对应的函数，并且Spring Data JPA允许在JPQL中调用这些原生函数（大部分情况是支持的）。
• 方案2中，若employeeIds里存在查询结果中没有的ID，idOrderMap.get(emp.getId())不会返回null，因为查询结果的ID肯定都在employeeIds里（因为用了IN条件），所以不用担心空指针问题。

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