MyException - 我的异常网
当前位置:我的异常网» 软件架构设计 » 大批量数据导出Excel发生内存溢出解决方案

大批量数据导出Excel发生内存溢出解决方案

www.MyException.Cn  网友分享于:2013-10-07  浏览:63次
大批量数据导出Excel产生内存溢出解决方案
@SuppressWarnings("unchecked")
public class XlsMergeUtil {
	private static Logger logger = LoggerFactory.getLogger(XlsMergeUtil.class);

	/**
	 * 将多个Xls文件合并为一个,适用于只有一个sheet,并且格式相同的文档
	 * 
	 * @param inputs
	 *            输入的Xls文件,第一个XLS文件必须给出足够sheet空间 例如,总共200000行数据,第一个文件至少3个空白sheet
	 * @param out
	 *            输出文件
	 */
	public static void merge(InputStream[] inputs, OutputStream out) {
		if (inputs == null || inputs.length <= 1) {
			throw new IllegalArgumentException("没有传入输入流数组,或只有一个输入流.");
		}

		List<Record> rootRecords = getRecords(inputs[0]);
		Workbook workbook = Workbook.createWorkbook(rootRecords);
		List<Sheet> sheets = getSheets(workbook, rootRecords);
		if (sheets == null || sheets.size() == 0) {
			throw new IllegalArgumentException("第一篇文档的格式错误,必须有至少一个sheet");
		}
		// 以第一篇文档的第一个sheet为根,以后的数据都追加在这个sheet后面
		Sheet rootSheet = sheets.get(0);
		int rootRows = getRowsOfSheet(rootSheet); // 记录第一篇文档的行数,以后的行数在此基础上增加
		rootSheet.setLoc(rootSheet.getDimsLoc());
		Map<Integer, Integer> map = new HashMap(10000);
		int sheetIndex = 0;

		for (int i = 1; i < inputs.length; i++) { // 从第二篇开始遍历
			List<Record> records = getRecords(inputs[i]);
			// 达到最大行数限制,换一个sheet
			if (getRows(records) + rootRows >= RowRecord.MAX_ROW_NUMBER) {
				if ((++sheetIndex) > (sheets.size() - 1)) {
					logger.warn("第一个文档给出的sheets小于需要的数量,部分数据未能合并.");
					break;
				}
				rootSheet = sheets.get(sheetIndex);
				rootRows = getRowsOfSheet(rootSheet);
				rootSheet.setLoc(rootSheet.getDimsLoc());
				logger.debug("切换Sheet{}", sheetIndex);
			}
			int rowsOfCurXls = 0;
			// 遍历当前文档的每一个record
			for (Iterator itr = records.iterator(); itr.hasNext();) {
				Record record = (Record) itr.next();
				if (record.getSid() == RowRecord.sid) { // 如果是RowRecord
					RowRecord rowRecord = (RowRecord) record;
					// 调整行号
					rowRecord.setRowNumber(rootRows + rowRecord.getRowNumber());
					rootSheet.addRow(rowRecord); // 追加Row
					rowsOfCurXls++; // 记录当前文档的行数
				}
				// SST记录,SST保存xls文件中唯一的String,各个String都是对应着SST记录的索引
				else if (record.getSid() == SSTRecord.sid) {
					SSTRecord sstRecord = (SSTRecord) record;
					for (int j = 0; j < sstRecord.getNumUniqueStrings(); j++) {
						int index = workbook.addSSTString(sstRecord
								.getString(j));
						// 记录原来的索引和现在的索引的对应关系
						map.put(Integer.valueOf(j), Integer.valueOf(index));
					}
				} else if (record.getSid() == LabelSSTRecord.sid) {
					LabelSSTRecord label = (LabelSSTRecord) record;
					// 调整SST索引的对应关系
					label.setSSTIndex(map.get(Integer.valueOf(label
							.getSSTIndex())));
				}
				// 追加ValueCell
				if (record instanceof CellValueRecordInterface) {
					CellValueRecordInterface cell = (CellValueRecordInterface) record;
					int cellRow = cell.getRow() + rootRows;
					cell.setRow(cellRow);
					rootSheet.addValueRecord(cellRow, cell);
				}
			}
			rootRows += rowsOfCurXls;
		}

		byte[] data = getBytes(workbook, sheets.toArray(new Sheet[0]));
		write(out, data);
	}

	static void write(OutputStream out, byte[] data) {
		POIFSFileSystem fs = new POIFSFileSystem();
		// Write out the Workbook stream
		try {
			fs.createDocument(new ByteArrayInputStream(data), "Workbook");
			fs.writeFilesystem(out);
			out.flush();
		} catch (IOException e) {
			e.printStackTrace();
		} finally {
			try {
				out.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}

	static List<Sheet> getSheets(Workbook workbook, List records) {
		int recOffset = workbook.getNumRecords();
		int sheetNum = 0;

		// convert all LabelRecord records to LabelSSTRecord
		convertLabelRecords(records, recOffset, workbook);
		List<Sheet> sheets = new ArrayList();
		while (recOffset < records.size()) {
			Sheet sh = Sheet.createSheet(records, sheetNum++, recOffset);

			recOffset = sh.getEofLoc() + 1;
			if (recOffset == 1) {
				break;
			}
			sheets.add(sh);
		}
		return sheets;
	}

	static int getRows(List<Record> records) {
		int row = 0;
		for (Iterator itr = records.iterator(); itr.hasNext();) {
			Record record = (Record) itr.next();
			if (record.getSid() == RowRecord.sid) {
				row++;
			}
		}
		return row;
	}

	static int getRowsOfSheet(Sheet sheet) {
		int rows = 0;
		sheet.setLoc(0);
		while (sheet.getNextRow() != null) {
			rows++;
		}
		return rows;
	}

	@SuppressWarnings("deprecation")
	static List<Record> getRecords(InputStream input) {
		try {
			POIFSFileSystem poifs = new POIFSFileSystem(input);
			InputStream stream = poifs.getRoot().createDocumentInputStream(
					"Workbook");
			return org.apache.poi.hssf.record.RecordFactory
					.createRecords(stream);
		} catch (IOException e) {
			logger.error("IO异常:{}", e.getMessage());
			e.printStackTrace();
		}
		return Collections.EMPTY_LIST;
	}

	static void convertLabelRecords(List records, int offset, Workbook workbook) {

		for (int k = offset; k < records.size(); k++) {
			Record rec = (Record) records.get(k);

			if (rec.getSid() == LabelRecord.sid) {
				LabelRecord oldrec = (LabelRecord) rec;

				records.remove(k);
				LabelSSTRecord newrec = new LabelSSTRecord();
				int stringid = workbook.addSSTString(new UnicodeString(oldrec
						.getValue()));

				newrec.setRow(oldrec.getRow());
				newrec.setColumn(oldrec.getColumn());
				newrec.setXFIndex(oldrec.getXFIndex());
				newrec.setSSTIndex(stringid);
				records.add(k, newrec);
			}
		}
	}

	public static byte[] getBytes(Workbook workbook, Sheet[] sheets) {
		// HSSFSheet[] sheets = getSheets();
		int nSheets = sheets.length;

		// before getting the workbook size we must tell the sheets that
		// serialization is about to occur.
		for (int i = 0; i < nSheets; i++) {
			sheets[i].preSerialize();
		}

		int totalsize = workbook.getSize();
		// pre-calculate all the sheet sizes and set BOF indexes
		int[] estimatedSheetSizes = new int[nSheets];
		for (int k = 0; k < nSheets; k++) {
			workbook.setSheetBof(k, totalsize);
			int sheetSize = sheets[k].getSize();
			estimatedSheetSizes[k] = sheetSize;
			totalsize += sheetSize;
		}
		logger.debug("分配内存{}bytes", totalsize);
		byte[] retval = new byte[totalsize];
		int pos = workbook.serialize(0, retval);

		for (int k = 0; k < nSheets; k++) {
			int serializedSize = sheets[k].serialize(pos, retval);
			if (serializedSize != estimatedSheetSizes[k]) {
				// Wrong offset values have been passed in the call to
				// setSheetBof()
				// above.
				// For books with more than one sheet, this discrepancy would
				// cause
				// excel
				// to report errors and loose data while reading the workbook
				throw new IllegalStateException(
						"Actual serialized sheet size (" + serializedSize
								+ ") differs from pre-calculated size ("
								+ estimatedSheetSizes[k] + ") for sheet (" + k
								+ ")");
				// TODO - add similar sanity check to ensure that
				// Sheet.serializeIndexRecord() does not
				// write mis-aligned offsets either
			}
			pos += serializedSize;
		}
		return retval;
	}

	public static void main(String[] args) throws Exception {
		final String PATH = "E:\\projects\\java\\ws_0\\export\\data\\";
		InputStream[] inputs = new InputStream[25];
		inputs[0] = new java.io.FileInputStream(PATH + "07_10.xls");
		for (int i = 1; i < 25; i++) {
			inputs[i] = new java.io.FileInputStream(PATH + "07_01.xls");
		}
		OutputStream out = new FileOutputStream(PATH + "xx.xls");
		long t1 = System.currentTimeMillis();
		merge(inputs, out);
		System.out.println(System.currentTimeMillis() - t1);
	}

}

文章评论

十大编程算法助程序员走上高手之路
十大编程算法助程序员走上高手之路
一个程序员的时间管理
一个程序员的时间管理
程序员和编码员之间的区别
程序员和编码员之间的区别
做程序猿的老婆应该注意的一些事情
做程序猿的老婆应该注意的一些事情
漫画:程序员的工作
漫画:程序员的工作
程序员的鄙视链
程序员的鄙视链
程序员眼里IE浏览器是什么样的
程序员眼里IE浏览器是什么样的
10个帮程序员减压放松的网站
10个帮程序员减压放松的网站
程序员应该关注的一些事儿
程序员应该关注的一些事儿
程序员最害怕的5件事 你中招了吗?
程序员最害怕的5件事 你中招了吗?
60个开发者不容错过的免费资源库
60个开发者不容错过的免费资源库
每天工作4小时的程序员
每天工作4小时的程序员
中美印日四国程序员比较
中美印日四国程序员比较
我跳槽是因为他们的显示器更大
我跳槽是因为他们的显示器更大
鲜为人知的编程真相
鲜为人知的编程真相
Web开发人员为什么越来越懒了?
Web开发人员为什么越来越懒了?
为什么程序员都是夜猫子
为什么程序员都是夜猫子
什么才是优秀的用户界面设计
什么才是优秀的用户界面设计
科技史上最臭名昭著的13大罪犯
科技史上最臭名昭著的13大罪犯
10个调试和排错的小建议
10个调试和排错的小建议
旅行,写作,编程
旅行,写作,编程
如何区分一个程序员是“老手“还是“新手“?
如何区分一个程序员是“老手“还是“新手“?
团队中“技术大拿”并非越多越好
团队中“技术大拿”并非越多越好
初级 vs 高级开发者 哪个性价比更高?
初级 vs 高级开发者 哪个性价比更高?
写给自己也写给你 自己到底该何去何从
写给自己也写给你 自己到底该何去何从
程序员的一天:一寸光阴一寸金
程序员的一天:一寸光阴一寸金
看13位CEO、创始人和高管如何提高工作效率
看13位CEO、创始人和高管如何提高工作效率
那些争议最大的编程观点
那些争议最大的编程观点
Web开发者需具备的8个好习惯
Web开发者需具备的8个好习惯
程序员必看的十大电影
程序员必看的十大电影
当下全球最炙手可热的八位少年创业者
当下全球最炙手可热的八位少年创业者
代码女神横空出世
代码女神横空出世
Java程序员必看电影
Java程序员必看电影
不懂技术不要对懂技术的人说这很容易实现
不懂技术不要对懂技术的人说这很容易实现
聊聊HTTPS和SSL/TLS协议
聊聊HTTPS和SSL/TLS协议
老美怎么看待阿里赴美上市
老美怎么看待阿里赴美上市
如何成为一名黑客
如何成为一名黑客
“懒”出效率是程序员的美德
“懒”出效率是程序员的美德
Java 与 .NET 的平台发展之争
Java 与 .NET 的平台发展之争
为啥Android手机总会越用越慢?
为啥Android手机总会越用越慢?
程序员都该阅读的书
程序员都该阅读的书
程序员周末都喜欢做什么?
程序员周末都喜欢做什么?
我是如何打败拖延症的
我是如何打败拖延症的
编程语言是女人
编程语言是女人
 程序员的样子
程序员的样子
软件开发程序错误异常ExceptionCopyright © 2009-2015 MyException 版权所有