Как преобразовать документ iTextPDF в байтовый массив

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

Проблема заключалась в том, что после закрытия документа (document.close()) закрывался и поток.

Обходной путь заключался в том, чтобы создать ByteArrayOutputStream, когда я создал документ, и получить для него вывод PdfWriter. Затем я мог делать все, что хотел, с байтами PDF... в моем случае я преобразовал их в StreamedContent для отправки пользователю.

Создайте переменную для хранения байтов:

  private ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();

Пусть PdfWriter выводит данные в byte[] при создании документа:

  Document document = new Document(PageSize.LETTER, 0.75F, 0.75F, 0.75F, 0.75F);
  PdfWriter.getInstance(document, byteArrayOutputStream);  // Do this BEFORE document.open()

  document.open();
  createPDF(document);  // Whatever function that you use to create your PDF
  document.close();

Как только вы закончите создание PDF-файла, просто получите байты и делайте с ними, что хотите.

  byte[] pdfBytes = byteArrayOutputStream.toByteArray();

Я не знаю, как выглядит ваш класс reportService, но это может быть хорошим местом для его размещения.

Надеюсь это поможет.

Из комментариев видно, что он не имеет ничего общего с тем, как PDF создается во время выполнения, а с тем, как он хранится в БД. Вам также необходимо предоставить этот код.

Вот что у меня есть на самом деле (мой тест):

Держатель:

import java.io.Serializable;

import javax.persistence.Column;
import javax.persistence.Entity;
import javax.persistence.GeneratedValue;
import javax.persistence.GenerationType;
import javax.persistence.Id;



@Entity
public class PDFHolder implements Serializable {

@Id
@GeneratedValue(strategy=GenerationType.AUTO)
private long id;

@Column(columnDefinition = "LONGBLOB")
private byte[] documentPDF;

public byte[] getDocumentPDF() {
    return documentPDF;
}

public void setDocumentPDF(byte[] documentPDF) {
    this.documentPDF = documentPDF;
}

public long getId() {
    return id;
}

public void setId(long id) {
    this.id = id;
}
}

Репозиторий, который делает сохранение:

@Repository
public interface PDFHolderRepository extends JpaRepository<PDFHolder, Long> {}

И собственно вставка:

private void generateDatabaseRows() {

    try{
        String urlPDF = "http://cetatenie.just.ro/LinkClick.aspx?fileticket=K13DZkoIE2o%3d&tabid=57&mid=593";
        URL url = new URL(urlPDF);

        ByteBuffer byteBufferResponse = this.getAsByteArray(url.openConnection());
        byte [] responseArray = byteBufferResponse.array();
        System.out.println("Size of the PDF : " + responseArray.length);

        // Save it to DB
        PDFHolder pdfHolder = new PDFHolder();
        pdfHolder.setDocumentPDF(responseArray);

        pdfHolderRepository.save(pdfHolder);

    } catch(Exception e){
        e.printStackTrace();
    }
}


private ByteBuffer getAsByteArray(final URLConnection urlConnection) throws IOException {
    final ByteArrayOutputStream tmpOut = new ByteArrayOutputStream();
    final InputStream inputStream = urlConnection.getInputStream();
    final byte[] buf = new byte[1024];
    int len;
    while (true) {
        len = inputStream.read(buf);
        if (len == -1) {
            break;
        }
        tmpOut.write(buf, 0, len);
    }
    tmpOut.close();
    return ByteBuffer.wrap(tmpOut.toByteArray(), 0, tmpOut.size());
}

Конечно, у меня есть репозиторий @Autowired в этом классе:

@Autowired
PDFHolderRepository pdfHolderRepository;

Просто обновляю ответ для версии iText 7.

private static byte[] createPDF() {
    ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
    PdfWriter writer = new PdfWriter(byteArrayOutputStream);
    PdfDocument pdfDocument = new PdfDocument(writer);
    Document document = new Document(pdfDocument);

    //Write the file content

    document.close();
    return byteArrayOutputStream.toByteArray();
}