Spring Data Repository更新操作如何返回修改的实体列表？

Spring Data JPA中被@Modifying标记的更新/删除查询，确实只支持返回void或int/Integer（受影响行数），这是JPA规范的限制。要获取被修改的实体ID列表，可通过以下几种方案实现：

方案一：先查询待更新ID，再执行更新 #

适合数据量不大、需要跨数据库兼容的场景，步骤如下：

1. 在Repository中新增查询方法，提前获取符合条件的实体ID列表
2. 调用原更新方法执行修改操作
3. 业务层直接使用之前查询到的ID列表

代码示例：

// Repository接口新增查询ID的方法
@Query("SELECT e.id FROM Entity e WHERE e.companyId = :sourceId")
List<Long> findEntityIdsByCompanyId(String sourceId);

// 保留原更新方法
static final String UPDATE_ENTITY =
"""UPDATE entity SET updated_on = :updatedDateTime, updated_by = :createdBy, company_id = :targetId where company_id = :sourceId""";

@Modifying
@Transactional
@Query(value = UPDATE_ENTITY, nativeQuery = true)
int transferEntity(String sourceId, String targetId, Timestamp updatedDateTime, String createdBy);

注意：该方案存在并发风险——若查询完成后、更新执行前，有其他操作修改了符合条件的实体，会导致查询到的ID与实际被更新的ID不一致。如需避免，可在查询时加锁（如SELECT e.id FROM Entity e WHERE e.companyId = :sourceId FOR UPDATE），但会牺牲部分性能。

方案二：利用数据库的RETURNING语法（依赖数据库特性） #

部分数据库支持在UPDATE语句后通过RETURNING子句返回被修改的数据，比如PostgreSQL原生支持，MySQL 8.0.19及以上版本也支持该语法。修改SQL和Repository方法如下：

static final String UPDATE_ENTITY =
"""UPDATE entity SET updated_on = :updatedDateTime, updated_by = :createdBy, company_id = :targetId 
    WHERE company_id = :sourceId
    RETURNING id""";

// 移除@Modifying注解，此时查询会返回结果而非仅执行修改
@Transactional
@Query(value = UPDATE_ENTITY, nativeQuery = true)
List<Long> transferEntity(String sourceId, String targetId, Timestamp updatedDateTime, String createdBy);

优点：原子性强，无并发不一致问题，性能更高；缺点：依赖数据库版本和类型，跨数据库兼容性差。

方案三：通过EntityManager手动执行查询 #

直接使用JPA的EntityManager执行原生更新查询，灵活控制查询逻辑并获取返回的ID列表：

@Repository
public class EntityRepositoryImpl {

    @PersistenceContext
    private EntityManager entityManager;

    @Transactional
    public List<Long> transferEntityAndGetIds(String sourceId, String targetId, Timestamp updatedDateTime, String createdBy) {
        String sql = """
            UPDATE entity SET updated_on = ?, updated_by = ?, company_id = ? 
            WHERE company_id = ? 
            RETURNING id
            """;
        Query query = entityManager.createNativeQuery(sql);
        query.setParameter(1, updatedDateTime);
        query.setParameter(2, createdBy);
        query.setParameter(3, targetId);
        query.setParameter(4, sourceId);
        return query.getResultList();
    }
}

这种方式适合复杂场景，无需依赖Spring Data的注解限制，但需手动管理参数和结果转换。

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