如何将OpenPdf生成的OutputStream转为http4s可响应的可读可下载PDF

核心问题 #

1. 一个Document绑定多个PdfWriter：你同时给同一个Document实例绑定了FileOutputStream和ByteArrayOutputStream的PdfWriter，这会导致PDF内容写入混乱，ByteArrayOutputStream无法正确获取完整的PDF数据。
2. 返回Base64字符串而非二进制数据：http4s的Ok接收字符串时会按文本格式响应，但PDF是二进制文件，直接返回Base64编码的字符串会让浏览器无法正确解析。
3. Content-Length计算错误：你用Base64字符串的长度作为Content-Length，实际应该用原始PDF字节数组的长度。
4. 缺少Content-Disposition头：没有指定该头的话，浏览器可能无法正确触发在线查看或下载行为。

修复后的代码 #

1. 修正PDF生成方法

只保留ByteArrayOutputStream用于生成响应的PDF数据，若需本地保存可单独处理，不要和响应流混用：

import java.io.ByteArrayOutputStream
import com.lowagie.text.Document
import com.lowagie.text.pdf.PdfWriter

def createPdfSingleBook(bookF: BookF): Array[Byte] = {
  val monolingualBook = new Document()
  val baos = new ByteArrayOutputStream()
  
  // 仅绑定一个PdfWriter到ByteArrayOutputStream
  val pdfWriter = PdfWriter.getInstance(monolingualBook, baos)
  
  monolingualBook.open()
  
  // 在这里添加PDF内容，示例：
  // monolingualBook.add(new Paragraph(s"书名：${bookF.title}"))
  // monolingualBook.add(new Paragraph(s"作者：${bookF.author}"))
  
  monolingualBook.close()
  pdfWriter.close()
  
  // 返回原始字节数组，无需Base64编码
  baos.toByteArray
}

2. 修正路由处理逻辑

直接返回字节数组，调整响应头，添加Content-Disposition控制浏览器行为：

import org.http4s._
import org.http4s.dsl.io._
import cats.effect.IO

case GET -> Root / "pdf" / IntVar(bookId) => 
  getBookF(bookId) match {
    case Some(bookF: BookF) =>
      val pdfBytes = createPdfSingleBook(bookF)
      Ok(pdfBytes)
        .withHeaders(
          Header.Raw(CIString("Content-Type"), "application/pdf"),
          Header.Raw(CIString("Content-Length"), s"${pdfBytes.length}"),
          // 可选：inline让浏览器在线查看，attachment触发下载
          Header.Raw(CIString("Content-Disposition"), s"inline; filename=${bookF.id}.pdf"),
          Header.Raw(CIString("Cache-Control"), "must-revalidate, no-cache, no-store"),
          Header.Raw(CIString("Pragma"), "no-cache"),
          Header.Raw(CIString("Expires"), "0")
        )
    case None => NotFound()
  }

关键说明 #

• 单一PdfWriter绑定：确保一个Document仅对应一个PdfWriter，避免写入冲突，保证ByteArrayOutputStream获取完整的PDF二进制数据。
• 返回二进制字节数组：http4s的Ok支持直接接收Array[Byte]，自动按二进制格式响应，浏览器可正确识别PDF类型。
• Content-Disposition头：inline参数让浏览器在线打开PDF，attachment则触发下载对话框，可按需调整。
• 缓存控制头：采用严格的缓存策略，避免浏览器缓存旧的PDF内容。

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