Возвращение сгенерированного pdf с использованием spring MVC

Я использую Spring MVC. Мне нужно написать службу, которая будет вводить данные из тела запроса, добавить данные в pdf и вернуть PDF файл в браузер. PDF-документ создается с помощью itextpdf. Как это сделать, используя Spring MVC. Я попытался использовать этот

@RequestMapping(value="/getpdf", method=RequestMethod.POST)
public Document getPDF(HttpServletRequest request , HttpServletResponse response, 
      @RequestBody String json) throws Exception {
    response.setContentType("application/pdf");
    response.setHeader("Content-Disposition", "attachment:filename=report.pdf");
    OutputStream out = response.getOutputStream();
    Document doc = PdfUtil.showHelp(emp);
    return doc;
}

showhelp функция, которая генерирует PDF. Я просто помещаю некоторые случайные данные в pdf на время.

public static Document showHelp(Employee emp) throws Exception {
    Document document = new Document();

    PdfWriter.getInstance(document, new FileOutputStream("C:/tmp/report.pdf"));
    document.open();
    document.add(new Paragraph("table"));
    document.add(new Paragraph(new Date().toString()));
    PdfPTable table=new PdfPTable(2);

    PdfPCell cell = new PdfPCell (new Paragraph ("table"));

    cell.setColspan (2);
    cell.setHorizontalAlignment (Element.ALIGN_CENTER);
    cell.setPadding (10.0f);
    cell.setBackgroundColor (new BaseColor (140, 221, 8));                                  

    table.addCell(cell);                                    
    ArrayList<String[]> row=new ArrayList<String[]>();
    String[] data=new String[2];
    data[0]="1";
    data[1]="2";
    String[] data1=new String[2];
    data1[0]="3";
    data1[1]="4";
    row.add(data);
    row.add(data1);

    for(int i=0;i<row.size();i++) {
      String[] cols=row.get(i);
      for(int j=0;j<cols.length;j++){
        table.addCell(cols[j]);
      }
    }

    document.add(table);
    document.close();

    return document;   
}

Я уверен, что это неправильно. Я хочу, чтобы PDF файл был создан и сохранил/открыл диалоговое окно, которое будет открыто через браузер, чтобы его можно было сохранить в файловой системе клиента. Пожалуйста, помогите мне.

Ответ 1

Вы были на правильном пути с response.getOutputStream(), но вы нигде не используете его вывод в вашем коде. По сути, вам нужно направить байты файла PDF непосредственно в выходной поток и очистить ответ. Spring вы можете сделать это так:

@RequestMapping(value="/getpdf", method=RequestMethod.POST)
public ResponseEntity<byte[]> getPDF(@RequestBody String json) {
    // convert JSON to Employee 
    Employee emp = convertSomehow(json);

    // generate the file
    PdfUtil.showHelp(emp);

    // retrieve contents of "C:/tmp/report.pdf" that were written in showHelp
    byte[] contents = (...);

    HttpHeaders headers = new HttpHeaders();
    headers.setContentType(MediaType.APPLICATION_PDF);
    // Here you have to set the actual filename of your pdf
    String filename = "output.pdf";
    headers.setContentDispositionFormData(filename, filename);
    headers.setCacheControl("must-revalidate, post-check=0, pre-check=0");
    ResponseEntity<byte[]> response = new ResponseEntity<>(contents, headers, HttpStatus.OK);
    return response;
}

Заметки:

  • используйте осмысленные имена для своих методов: называть метод, который пишет документ PDF, showHelp не очень хорошая идея
  • чтение файла в byte[]: пример здесь
  • Я бы предложил добавить случайную строку к имени временного PDF файла внутри showHelp() чтобы избежать перезаписи файла, если два пользователя отправляют запрос одновременно

Ответ 2

Приложение к Кригерс ответь; MediaType.parseMediaType добавляет "; charset = utf8" к типу mime, вызывая повреждение PDF файла. Я обнаружил в Sping 4.x, что мне пришлось установить заголовок типа контента вручную:

headers.add("Content-Type", "application/pdf");

Надеюсь, что это помогает кому-то!