首页 > 代码库 > 百万甚至千万级数据下载
百万甚至千万级数据下载
(#)直奔主题,这次要说的是针对百万千万数据的下载,使用的是poi excel的下载,这个如果大家不熟悉的话,可以去看看这个基础博客,写的不错
http://www.jerehedu.com/fenxiang/160218_for_detail.htm
(#)然而问题来了,(1)excel如何装这么多的数据呢?(2)jvm肯定是一次放不下的 针对与问题(1),其实比较好解决,excel07提供了一个新的
XSSFWorkbook,它会将数据存储在磁盘上,内存中只加载一部分够自己使用就好,那么对于问题(2)怎么解决呢?其实想想也不是很难,只要使用
lazy的加载模式就没有问题了(lazy模式只会在使用的时候才将数据加载),那么在java中lazy的模式首先想到的就是Iterable来包装
(#)这里如果大家的tomcat没有打印gc.log日志,因为这个代码运行的时候,首先要关注的就是gc的问题,如果发现一直在fgc那么肯定就要跪了,这个时候怎么看gc
日志呢?嗯,如果不清楚的话,首先要做的就是去熟悉一下jvm带的命令,我们使用jstat -gcutil -uid 来看一下,如下:
(#) 废话不多说,show the code
首先是读取数据库数据的代码,如下:
Iterable<List<AirlineRecord>> airRecords = new Iterable<List<AirlineRecord>>() { @Override public Iterator<List<AirlineRecord>> iterator() { return new AbstractIterator<List<AirlineRecord>>() { int totalCount = airlineRecordDao.selectCountByState(null); int sizeCount = totalCount / SIZE; int index = 0; @Override protected List<AirlineRecord> computeNext() { if (index > sizeCount) { return endOfData(); } List<AirlineRecord> records = airlineRecordDao.selectByState(null, new RowBounds(index * SIZE, SIZE)); index++; return records; } }; } };
我的数据库中的数据量是100多万的,RowBounds应该使用一个拦截器改成物理分,要不还是会有问题的,然后剩下的就很简单了,利用Iterable本身的lazy属性,现
在不会将多有的数据都加载进来,每次只会迭代SIZE大小的数据,这样JVM就不会有任何压力,百万,千万的数据都是可以撑的住的
下面来看一下下载的代码
public static <T> SXSSFWorkbook export(Iterable<T> excelData, String sheetName, TableFormatterConfigurer configurer) { // 声明一个(SXSSFWorkbook)工作薄 SXSSFWorkbook workbook = new SXSSFWorkbook(); // SXSSFWorkbook默认只在内存中存放100行 T firstItem = Iterables.getFirst(excelData, null); if (firstItem != null) { return export(sheetName, excelData, workbook, configurer, (Class<T>) firstItem.getClass()); } return workbook; } /** * 导出Excel * * @param sheetName excel表格名 * @param excelData 要导出的数据 * @param workbook 要导出的工作薄 * @return * * @author mars.mao created on 2014年10月17日下午7:36:26 */ public static <T> HSSFWorkbook export(String sheetName, List<T> excelData, HSSFWorkbook workbook) { return export(sheetName, excelData, workbook, TableFormatterConfigurer.NONE); } public static <T> HSSFWorkbook export(String sheetName, List<T> excelData, HSSFWorkbook workbook, TableFormatterConfigurer configurer) { if (excelData =http://www.mamicode.com/= null || excelData.isEmpty() || workbook == null || StringUtils.isBlank(sheetName)) {"数据量超过了Excel的容量范围!"); } for (int i = 0; i < sheetCnt; i++) { int fromIndex = i * EXCEL_MAX_ROW_NO; int toIndex = fromIndex + EXCEL_MAX_ROW_NO; toIndex = toIndex > totalDataSize ? totalDataSize : toIndex; List<T> sheetData = http://www.mamicode.com/excelData.subList(fromIndex, toIndex);"_" + i); // 生成标题行 createHeader(sheetData, sheet, font, configurer); // 遍历集合数据,产生数据行 createBody(sheetData, sheet, configurer); } logger.info("导出的数据行数(不含表头): writeCount={}", excelData.size()); return workbook; } catch (Exception e) { logger.error("导出Excel异常!", e); } return workbook; } /** * 导出Excel, 使用SXSSFWorkbook支持大数据量的导出 * * @param sheetName excel表格名 * @param excelData 要导出的数据 * @param workbook 要导出的工作薄 * @param configurer 配置数据格式化行为的对象 * @param clazz 待导出数据的类型信息 * @return */ public static <T> SXSSFWorkbook export(String sheetName, Iterable<T> excelData, SXSSFWorkbook workbook, TableFormatterConfigurer configurer, Class<T> clazz) { final Iterator<T> externalIterator; if (excelData =http://www.mamicode.com/= null || !(externalIterator = excelData.iterator()).hasNext() || workbook == null"_" + i); // 生成标题行 createHeader(sheet, font, configurer, clazz); // 遍历集合数据,产生数据行 writeCount += createBody(new Iterable<T>() { public Iterator<T> iterator() { return new Iterator<T>() { int lineCount = 0; public boolean hasNext() { return lineCount < EXCEL_MAX_ROW_NO && externalIterator.hasNext(); } public T next() { if (!hasNext()) { throw new NoSuchElementException("没有更多的元素"); } lineCount++; return externalIterator.next(); } public void remove() { throw new UnsupportedOperationException("不支持删除操作"); } }; } }, sheet, configurer, clazz); if ((++i) == EXCEL_MAX_SHEET_CNT) { throw new Exception("数据量超过了Excel的容量范围!"); } } logger.info("导出的数据行数(不含表头): writeCount={}", writeCount); return workbook; } catch (Exception e) { logger.error("导出Excel异常!", e); } return workbook; } /** * 创建表格数据 * * @param excelData 要导出的数据 * @param sheet excel表格 * @param configurer 配置数据格式化行为的对象 * @return 返回创建的数据行数 * * @author mars.mao created on 2014年10月17日下午3:43:43 */ @SuppressWarnings("unchecked") private static <T> int createBody(List<T> excelData, Sheet sheet, TableFormatterConfigurer configurer) throws Exception { if (CollectionUtils.isEmpty(excelData)) { return 0; } Class<?> dataClass = excelData.get(0).getClass(); return createBody(excelData, sheet, configurer, dataClass); } /** * 创建表格数据 * * @param excelData 要导出的数据 * @param sheet excel表格 * @param configurer 配置数据格式化行为的对象 * @param clazz 要导出的对象的类型信息, 用以获取相关注解以及字段值 * @return 返回创建的数据行数 */ private static <T> int createBody(Iterable<T> excelData, Sheet sheet, TableFormatterConfigurer configurer, Class<?> clazz) throws Exception { if (excelData =http://www.mamicode.com/= null) {"格式化方法"对属性进行格式化 formattingMethod.setAccessible(true); textValue = http://www.mamicode.com/String.valueOf(formattingMethod.invoke(data));"格式化方法"时使用@ExcelColumn指定的格式化方式 textValue = http://www.mamicode.com/getTextValue(data, field, columnHeader, configurer);"unchecked") private static <T> void createHeader(List<T> excelData, Sheet sheet, Font font, TableFormatterConfigurer configurer) { if (CollectionUtils.isEmpty(excelData)) { return; } Class<?> dataClass = excelData.get(0).getClass(); createHeader(sheet, font, configurer, dataClass); } private static void createHeader(Sheet sheet, Font font, TableFormatterConfigurer configurer, Class<?> clazz) { List<Field> fields = getExportableFields(clazz, configurer); Row headerRow = sheet.createRow(0); int columnIndex = 0; for (Field field : fields) { ExcelColumn columnHeader = field.getAnnotation(ExcelColumn.class); // 获取指定的列标题和列宽度 String columnTitle = columnHeader.headerName(); int columnWidth = columnHeader.columnWidth(); // 创建列 Cell cell = headerRow.createCell(columnIndex); // 设置列标题 if (sheet instanceof HSSFSheet) { RichTextString text = new HSSFRichTextString(columnTitle); text.applyFont(font); cell.setCellValue(text); } else { cell.setCellValue(columnTitle); } // 设置列宽度 sheet.setColumnWidth(columnIndex, columnWidth * 256); columnIndex++; } } /** * 获取格式化的文本内容 * * @param obj 输入对象 * @param field 对象域 * @param columnHeader ExcelColumn注解的配置实例 * @param configurer TableFormatterConfigurer的配置实例 * @return 格式化后的文本 */ @SuppressWarnings("unchecked") private static <T> String getTextValue(T obj, Field field, ExcelColumn columnHeader, TableFormatterConfigurer configurer) { String aimPattern = columnHeader.pattern(); Object fieldValue = http://www.mamicode.com/getFieldValue(obj, field); // 反射获取字段的值"导出Excel使用formatter格式化出错: {}", e.getMessage(), e); formatted = false; } if (!formatted) { /* * 3) [未指定formatter, 且未开启formatter的注册] 或者 [以上格式化失败]; 使用默认的处理方式... */ textValue = http://www.mamicode.com/defaultToString(fieldValue, aimPattern);" "; if (fieldValue != null) { textValue = http://www.mamicode.com/fieldValue.toString();"导出Excel日期格式化错误!", e); } } else if (fieldValue instanceof DateTime) { // 添加对DateTime类型的兼容(采用与Date类型一样的pattern设置) [by chenjiahua.chen // 2016/09/05] try { String pattern = StringUtils.isBlank(aimPattern) ? DEFAULT_DATE_FORMAT : aimPattern; DateTime dateTime = (DateTime) fieldValue; textValue = http://www.mamicode.com/dateTime.toString(pattern);"导出Excel日期格式化错误!", e); } } else if (fieldValue instanceof Money) { // 添加对Money类型的兼容, 以toMoneyString输出(两位小数) [by chenjiahua.chen // 2016/09/05] try { Money money = (Money) fieldValue; return money.toMoneyString(); } catch (Exception e) { logger.error("导出Excel金额格式化错误!", e); } } else if (fieldValue instanceof Number) { if (StringUtils.isNotBlank(aimPattern)) { try { double doubleValue = http://www.mamicode.com/Double.parseDouble(fieldValue.toString());"导出Excel数字格式化错误!", e); } } } return textValue; } /** * 反射获取字段的值 * * @param obj 对象 * @param field 字段 * @return * * @author mars.mao created on 2014年10月17日下午2:58:53 */ private static <T> Object getFieldValue(T obj, Field field) { Object fieldValuehttp://www.mamicode.com/= " "; try { field.setAccessible(true); fieldValue = http://www.mamicode.com/field.get(obj);"导出Excel动态获取字段值异常", e); } return fieldValue; } /** * 根据给定类的信息获取其所有字段 (包括继承而来的字段!) */ private static List<Field> getExportableFields(Class<?> clazz, final TableFormatterConfigurer configurer) { List<Field> fields = Lists.newArrayList(); /* 循环向上查找, 以支持父类中的属性导出 [by chenjiahua.chen 2016/09/05] */ for (Class<?> dataClass : getClassesInHierarchy(clazz)) { fields.addAll(Arrays.asList(dataClass.getDeclaredFields())); } // 过滤不需要导出的属性 fields = Lists.newArrayList(Iterables.filter(fields, new Predicate<Field>() { @Override public boolean apply(Field field) { ExcelColumn columnHeader = field.getAnnotation(ExcelColumn.class); if (columnHeader == null || configurer.getHeadersToBeFiltered().contains(columnHeader.headerName())) { return false; } return true; } })); return fields; } private static Map<String, Method> getFormattingMethods(Class<?> clazz) { List<Method> methods = Lists.newArrayList(); /* 循环向上查找, 以支持父类中的格式化方法 [by chenjiahua.chen 2017/02/09] */ for (Class<?> dataClass : getClassesInHierarchy(clazz)) { methods.addAll(Arrays.asList(dataClass.getDeclaredMethods())); } // 过滤和格式化无关的方法 methods = Lists.newArrayList(Iterables.filter(methods, new Predicate<Method>() { @Override public boolean apply(Method method) { return method.getAnnotation(FieldFormat.class) != null; } })); return Maps.uniqueIndex(methods, new Function<Method, String>() { @Override public String apply(Method method) { return method.getAnnotation(FieldFormat.class).fieldName(); } }); } /** * 获取继承体系中的类 */ private static List<Class<?>> getClassesInHierarchy(Class<?> clazz) { List<Class<?>> classes = Lists.newArrayList(); Stack<Class<?>> classStack = new Stack<Class<?>>(); Class<?> currentClass = clazz; while (currentClass != null) { classStack.push(currentClass); currentClass = currentClass.getSuperclass(); } while (!classStack.isEmpty()) { classes.add(classStack.pop()); } return classes; } public static Workbook dynamicExport(String sheetName, List<RowRecord> excelData) { if (org.apache.commons.collections4.CollectionUtils.isEmpty(excelData) || StringUtils.isBlank(sheetName)) { return null; } HSSFWorkbook workbook = new HSSFWorkbook(); // 标题行字体 Font font = workbook.createFont(); font.setBold(true); // 定义单元格格式 CellStyle cellStyle = workbook.createCellStyle(); cellStyle.setAlignment(HorizontalAlignment.CENTER); // 分割sheet int totalCnt = excelData.size(); int sheetCnt = totalCnt / EXCEL_MAX_ROW_NO + 1; if (sheetCnt > EXCEL_MAX_SHEET_CNT) { throw new RuntimeException("数据量超过了Excel的容量范围!"); } for (int i = 0; i < sheetCnt; i++) { int fromIndex = i * EXCEL_MAX_ROW_NO; int toIndex = fromIndex + EXCEL_MAX_ROW_NO; toIndex = toIndex > totalCnt ? totalCnt : toIndex; List<RowRecord> sheetData = http://www.mamicode.com/excelData.subList(fromIndex, toIndex);"_" + i); // 生成标题行 createHeader(sheetData, sheet, font, cellStyle); // 遍历集合数据,产生数据行 createBody(sheetData, sheet); } return workbook; } private static void createHeader(List<RowRecord> excelData, Sheet sheet, Font font, CellStyle cellStyle) { if (org.apache.commons.collections4.CollectionUtils.isEmpty(excelData)) { return; } RowRecord rowRecord = excelData.get(0); createHeader(sheet, rowRecord, font, cellStyle); } private static void createHeader(Sheet sheet, RowRecord rowRecord, Font font, CellStyle cellStyle) { if (org.apache.commons.collections4.CollectionUtils.isEmpty(rowRecord.getColumnRecordList())) { return; } // 获取表头节点的层数 List<ColumnRecord> columnRecordList = rowRecord.getColumnRecordList(); int HeaderLayerCnt = 1; for (ColumnRecord columnRecord : columnRecordList) { if (HeaderLayerCnt < columnRecord.getColumnNameList().size()) { HeaderLayerCnt = columnRecord.getColumnNameList().size(); } } int i = 0; int columnIndex; int preColumnIndex; String preHeader; while (i < HeaderLayerCnt) { preHeader = null; preColumnIndex = columnIndex = 0; Row headerRow = sheet.createRow(i); for (ColumnRecord columnRecord : rowRecord.getColumnRecordList()) { if (columnRecord.getColumnNameList().size() < i + 1) { columnIndex++; preColumnIndex++; continue; } if (preHeader == null || !columnRecord.getColumnNameList().get(i).equals(preHeader)) { // 获取指定的列标题和列 if (columnIndex - preColumnIndex > 1) { sheet.addMergedRegion(new CellRangeAddress(i, i, preColumnIndex, columnIndex - 1)); } preColumnIndex = columnIndex; String columnTitle = columnRecord.getColumnNameList().get(i); // 创建列 Cell cell = headerRow.createCell(columnIndex); // 设置列标题 RichTextString text = new HSSFRichTextString(columnTitle); text.applyFont(font); cell.setCellValue(text); cell.setCellStyle(cellStyle); preHeader = columnRecord.getColumnNameList().get(i); } // 最后一列判定 if ((columnIndex == rowRecord.getColumnRecordList().size() - 1) && (columnIndex - preColumnIndex >= 1)) { sheet.addMergedRegion(new CellRangeAddress(i, i, preColumnIndex, columnIndex)); } columnIndex++; } i++; } } private static void createBody(List<RowRecord> rowRecords, Sheet sheet) { int rowIndex = sheet.getLastRowNum() + 1; for (RowRecord rowRecord : rowRecords) { Row dataRow = sheet.createRow(rowIndex); int columnIndex = 0; for (ColumnRecord columnRecord : rowRecord.getColumnRecordList()) { String textValue = http://www.mamicode.com/getTextValue(columnRecord);"导出Excel使用formatter格式化出错: {}", e.getMessage(), e); } return null; }
百万甚至千万级数据下载