Spring Boot自定义查询如何返回不同HTTP响应状态码？

1. 改造Repository方法，获取操作影响行数 #

你当前的自定义删除方法返回void，无法知道实际删除了多少条记录。修改方法让它返回int类型，这个值代表SQL语句执行后受影响的行数：

public interface CommentRepository extends JpaRepository<User, Long> {
                
     @Modifying
     @Transactional
     @Query(value = "delete from users where first_name=:name", nativeQuery = true)
     int deleteByName(@Param("name") String name); 
}

说明：重命名方法为deleteByName是为了避免和JpaRepository自带的delete方法冲突，@Param用于明确绑定参数，避免解析异常。

2. 在Controller中根据结果返回对应状态码 #

将Controller的返回类型从void改为ResponseEntity<Void>，这样可以灵活控制HTTP响应状态：

@RestController
@RequestMapping(path = "/api/v1/users")
public class CommentController {
    
    private final CommentRepository repository;

    // 用构造注入替代@Autowired，更符合Spring最佳实践
    public CommentController(CommentRepository repository) {
        this.repository = repository;
    }
    
    @DeleteMapping(path = "/delete")
    public ResponseEntity<Void> delete(@RequestParam String name) {
        int affectedRows = repository.deleteByName(name);
        
        if (affectedRows > 0) {
            // 删除成功，返回200 OK
            return ResponseEntity.ok().build();
        } else {
            // 未找到匹配的用户，返回404 Not Found（RESTful风格的标准状态码）
            return ResponseEntity.notFound().build();
        }
    }
}

3. 处理异常场景 #

如果数据库操作出现异常（比如约束冲突、连接失败等），可以通过两种方式处理：

方式一：方法内捕获异常 #

@DeleteMapping(path = "/delete")
public ResponseEntity<Void> delete(@RequestParam String name) {
    try {
        int affectedRows = repository.deleteByName(name);
        
        if (affectedRows > 0) {
            return ResponseEntity.ok().build();
        } else {
            return ResponseEntity.notFound().build();
        }
    } catch (DataAccessException e) {
        // 数据库操作异常，返回500 Internal Server Error
        return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).build();
    }
}

方式二：全局异常处理 #

创建全局异常处理器，统一处理所有数据库相关异常：

@RestControllerAdvice
public class GlobalExceptionHandler {
    
    @ExceptionHandler(DataAccessException.class)
    public ResponseEntity<Void> handleDatabaseException(DataAccessException e) {
        return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).build();
    }
}

额外建议 #

• 状态码选型：如果是参数非法（比如name为空），可以返回400 Bad Request；未找到资源返回404是RESTful的标准做法。
• 分层优化：建议将@Transactional迁移到Service层，Repository只负责数据访问，Service处理业务逻辑和事务，Controller专注于请求响应处理，这样架构更清晰。

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