如何在JPA自定义查询中实现基于SELECT结果的UPDATE操作

原代码存在的问题 #

• JPQL的UPDATE语法不支持FROM子句做联表更新，你写的查询语法不符合JPQL规范，无法执行。
• 没有用到传入的id参数做过滤，会导致全表的WorkWithdraw记录被更新，产生严重数据问题。
• Repository层误用了Controller层的@RequestParam注解，该注解在持久层无效，同时参数类型应为WorkWithdraw的主键类型Long，而非实体类本身。
• 加了@Modifying注解的更新方法返回值应为int类型，代表被更新的记录行数，你定义的Long类型返回值不符合规范。
• 关联条件Ww.id = d.id需要确认和你的实际表外键逻辑匹配，通常Deposit表会有专门的外键字段关联WorkWithdraw的主键，而非主键之间直接相等。

修正后的实现代码 #

推荐使用原生SQL实现联表更新，适配大多数关系型数据库的语法，代码如下：

package com.nilmani.workload.repository

import com.nilmani.workload.entity.Deposit
import org.springframework.data.jpa.repository.JpaRepository
import org.springframework.data.jpa.repository.Modifying
import org.springframework.data.jpa.repository.Query
import org.springframework.transaction.annotation.Transactional

interface DepositRepository:JpaRepository<Deposit,Long> {

    @Modifying
    @Transactional
    @Query(
        value = "UPDATE work_withdraw Ww INNER JOIN deposit d ON Ww.id = d.work_withdraw_id SET Ww.quantity = d.pending WHERE Ww.id = :id",
        nativeQuery = true
    )
    fun updateWithdrawQuantityById(id: Long): Int

}

注意事项 #

• 上述SQL的关联条件Ww.id = d.work_withdraw_id请根据你实际的表外键关联关系调整，如果你确实是两个表主键直接关联，修改为Ww.id = d.id即可。
• 方法返回值int代表本次操作更新的记录数，你可以在业务逻辑层判断返回值是否为1，确认对应id的记录是否更新成功。
• 如果你坚持要用JPQL实现，需要先查询对应Deposit的pending值，再单独更新WorkWithdraw表，分两步执行即可。

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