com.github.mengxianun.result.pdf.PDFResultConverter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of air-data-result-pdf Show documentation
Json drive data
There is a newer version: 1.0.3
package com.github.mengxianun.result.pdf;

import com.github.mengxianun.core.AbstractResultConverter;
import com.google.gson.JsonElement;
import com.itextpdf.text.*;
import com.itextpdf.text.pdf.BaseFont;
import com.itextpdf.text.pdf.PdfPCell;
import com.itextpdf.text.pdf.PdfPTable;
import com.itextpdf.text.pdf.PdfWriter;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Map;

public class PDFResultConverter extends AbstractResultConverter {

	private PDFResultConverter() {
	}
	private PDFResultConverter(Map properties, Map header, JsonElement data) {
		super.dataInit(properties, header, data);
		this.initStyle();
	}
	public static PDFResultConverter getInstance(Map properties, Map header, JsonElement data) {
		return new PDFResultConverter(properties, header, data);
	}

	// 式样
	BaseFont bfChinese = null;
	Font titleFont = null;
	Font dataFont = null;

	@Override
	protected InputStream export() throws Exception {

		Document doc = new Document();

		ByteArrayOutputStream baos = null;

		try {

			Rectangle rectPageSize = new Rectangle(PageSize.A4);
			doc = new Document(rectPageSize);// 可配其余4个参数，如（rectPageSize，60,60,60,60）页面边距

			baos = new ByteArrayOutputStream();//构建字节输出流
			PdfWriter.getInstance(doc,baos);//将PDF文档对象写入到流

			doc.open();

			//信息数据表

			//创建PdfTable对象
			PdfPTable table=new PdfPTable(headers.size());

			//表头添加表格内容
			if (!data.isEmpty()) {
				Map map = data.get(0);
				for (String key : map.keySet()) {
					table.addCell(getPDFCell(getDisplayName(key), titleFont));
				}
			}
			//遍历数据集合进行添加
			for (int i = 0; i < data.size(); i++) {
				for (Map.Entry entry : data.get(i).entrySet()) {
					table.addCell(getPDFCell(entry.getValue().toString(), dataFont));
				}
			}

			table.setWidthPercentage(110);//表宽度设置可不受页面边距影响
			doc.add(table);

			if(doc != null){
				doc.close();
			}
			return new ByteArrayInputStream(baos.toByteArray());
		} catch(Exception e) {
			throw e;
		} finally {
			if (baos != null) {
				try {
					baos.close();
				} catch (IOException e) {
				}
			}
		}
	}
	private static PdfPCell getPDFCell(String string, Font font) {
		//创建单元格对象，将内容与字体放入段落中作为单元格内容
		PdfPCell cell=new PdfPCell(new Paragraph(string,font));

		cell.setHorizontalAlignment(Element.ALIGN_CENTER);
		cell.setVerticalAlignment(Element.ALIGN_MIDDLE);

		//设置最小单元格高度
		cell.setMinimumHeight(30);
		return cell;
	}
	private void initStyle() {

		// 解决中文问题
		try {
			bfChinese = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.NOT_EMBEDDED);
		} catch (DocumentException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}

		titleFont = new Font(bfChinese,18,Font.BOLD);//表头
		dataFont = new Font(bfChinese,12,Font.NORMAL);
	}
}