使用Java给前端返回PDF文件流

在Web开发中，有时候我们需要给前端返回一个PDF文件，这种需求在很多场景下都会出现，比如生成报表、导出数据等。而使用Java来实现给前端返回PDF文件流是一种常见的方式。

生成PDF文件

首先，我们需要使用Java来生成一个PDF文件。在Java中，我们可以使用第三方库iText来实现PDF文件的生成。下面是一个简单的示例代码，用来生成一个包含饼状图的PDF文件：

import com.itextpdf.text.Document;
import com.itextpdf.text.Element;
import com.itextpdf.text.PageSize;
import com.itextpdf.text.Paragraph;
import com.itextpdf.text.pdf.PdfWriter;
import org.jfree.chart.ChartFactory;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.data.general.DefaultPieDataset;
import org.jfree.data.general.PieDataset;

import java.io.FileOutputStream;

public class PDFGenerator {

    public static void generatePDFWithPieChart() {
        Document document = new Document(PageSize.A4);

        try {
            PdfWriter.getInstance(document, new FileOutputStream("output.pdf"));
            document.open();

            PieDataset dataset = createDataset();
            JFreeChart chart = createChart(dataset);

            // Convert chart to image
            document.add(new Paragraph("Pie Chart Example:"));
            document.add(new Paragraph(" "));
            document.add(new Paragraph(" "));
            document.add(new Paragraph(" "));
            document.add(chart.createBufferedImage(500, 500, null, null));

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

    private static PieDataset createDataset() {
        DefaultPieDataset dataset = new DefaultPieDataset();
        dataset.setValue("Category 1", 40);
        dataset.setValue("Category 2", 30);
        dataset.setValue("Category 3", 20);
        dataset.setValue("Category 4", 10);
        return dataset;
    }

    private static JFreeChart createChart(PieDataset dataset) {
        JFreeChart chart = ChartFactory.createPieChart(
                "Pie Chart",  // chart title
                dataset,      // data
                true,         // include legend
                true,
                false);

        return chart;
    }

    public static void main(String[] args) {
        generatePDFWithPieChart();
    }
}

返回PDF文件流

接下来，我们需要将生成的PDF文件流返回给前端。这可以通过Servlet来实现，Servlet允许我们直接操作HTTP响应流。下面是一个简单的Servlet示例代码，用来将生成的PDF文件返回给前端：

import java.io.IOException;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

public class PDFServlet extends HttpServlet {

    protected void doGet(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        response.setContentType("application/pdf");
        response.setHeader("Content-Disposition", "attachment; filename=output.pdf");

        PDFGenerator.generatePDFWithPieChart();

        // Read PDF file and write it to response output stream
        // Here we can use FileInputStream and OutputStream to achieve this

        response.flushBuffer();
    }
}

通过以上代码，我们可以在浏览器中访问该Servlet，即可下载到包含饼状图的PDF文件。

总之，通过使用Java生成PDF文件，并通过Servlet将PDF文件流返回给前端，我们可以实现在Web应用中动态生成并导出PDF文件，满足各种需求，使我们的应用更加灵活和实用。

希望本文对您有所帮助，谢谢阅读。