如何使用@ControllerAdvice 和 ResponseEntity

问题描述

我正在开发一个 Java 应用程序，我正在开发的功能是在用户单击导出按钮时从数据库填充其内容后下载文件，文件生成正确，我遇到的问题是如果发生异常，我会尝试处理异常，这是我的代码：

控制器方法：

  @GetMapping(value = "/exportods/{id}")
  @ResponseBody
  public ResponseEntity<ByteArrayResource> exportODSFile(@PathVariable  String id )
      throws ObjectNotFoundException,IOException{


    File odsFfile = null;
    ByteArrayResource resource = null;
    HttpHeaders headers = null;

    try {          
      //generating ods file
      odsFfile = dao.exportInvalidationRequest(id);

      Path pathObj = Paths.get(odsFfile.getAbsolutePath());

      resource = new ByteArrayResource(Files.readAllBytes(pathObj));

      headers = new HttpHeaders();
      headers.add(HttpHeaders.CONTENT_disPOSITION,"attachment; filename=" + odsFfile.getName());
      headers.add(HttpHeaders.CONTENT_TYPE,MediaType.APPLICATION_OCTET_STREAM_VALUE);
      headers.add(HttpHeaders.CONTENT_LENGTH,String.valueOf(Files.readAllBytes(pathObj).length));
      headers.add(HttpHeaders.TRANSFER_ENCODING,"binary");

    } catch (ObjectNotFoundException e) {
      throw new ODSExportResourceException(HttpStatus.BAD_REQUEST,"message");
    }  catch (IOException e) {
      throw new ODSExportResourceException(HttpStatus.BAD_REQUEST,"message");
    }
    return new ResponseEntity<>(resource,headers,HttpStatus.OK);
  }

我的控制器建议：

@ControllerAdvice
public class ODSExportExceptionHandlerAdvice {
  @ResponseBody
  @ExceptionHandler(ODSExportResourceException.class)
  public ResponseEntity handleException(ODSExportResourceException e) {
    // log exception
    return ResponseEntity.status(e.getHttpStatus()).body(e.getMessage());
  }
}

ODSExportResourceException 类

public class ODSExportResourceException extends  RuntimeException{

  private HttpStatus httpStatus = HttpStatus.INTERNAL_SERVER_ERROR;

  public HttpStatus getHttpStatus() {
    return httpStatus;
  }

  public ODSExportResourceException(HttpStatus httpStatus,String message) {
    super(message);
    this.httpStatus = httpStatus;
  }
}

前端 Angular Service 方法：

downloadODSFile(id: string) {
    return this.httpClient.get(`${this.basePath}/exportods/${id}`,{
        responseType: 'arraybuffer'
    });
}

角度组件方法：

downloadODSFile(id) {
         this.requestService.downloadODSFile(id).
            subscribe(data => {
                saveAs(new Blob([data],{ type: '\'application/vnd.oasis.opendocument.spreadsheet\'' }),id);
                 },error => {
             this.errorService.displayError(error);
             console.log('here is the error ' + error.toString())
            });
}

为了实现这段代码，我参考了这里提到的 anwser ：[lastAnswer]

当我运行应用程序并触发导出时，在理想情况下，会生成并下载文件，但如果在后面触发异常，例如 id dosn' t 引用数据库中的任何行，会触发 ObjectNotFoundException，但我没有收到 ControllerAdvice 生成的 ResponseEntity。 触发异常时如何将 ResponseEntity 作为 JSON 处理？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）