前言
最近做的高校平台项目中,许多模块都用到了导入导出Excel的功能,于是决定封装一个ExcelUtil类,专门用来处理Excel的导入和导出
本项目的持久化层用的是JPA(底层用hibernate实现),所以导入和导出也都是基于实体类的。
在编写ExcelUtil之前,在网上查了一些资料。java中用来处理Excel的第三方开源项目主要就是POI和JXL。poi功能强大,但是比较耗资源,对于大数据量的导入导出性能不是太好;jxl功能简单,但是性能比较好。
由于本项目的导入导出更多关注性能问题,而且jxl提供的功能基本也都够用了,于是选择了jxl作为支持。
实战
导出就是将List转化为Excel(listToExcel)
导入就是将Excel转化为List(excelToList)
导入导出中会出现各种各样的问题,比如:数据源为空、有重复行等,我自定义了一个ExcelException异常类,用来处理这些问题。
ExcelException类
[java]view plain copy packagecommon.tool.excel; publicclassExcelExceptionextendsException{ publicExcelException(){ //TODOAuto-generatedconstructorstub } publicExcelException(Stringmessage){ super(message); //TODOAuto-generatedconstructorstub } publicExcelException(Throwablecause){ super(cause); //TODOAuto-generatedconstructorstub } publicExcelException(Stringmessage,Throwablecause){ super(message,cause); //TODOAuto-generatedconstructorstub } }下面就是该文的主角 ExcelUtil 登场了,作为一个工具类,其内的所有方法都是静态的,方便使用。
ExcelUitl类
[java]view plain copy /** *@author:WH *@group:tgb8 *@Date:-1-2下午9:13:21 *@Comments:导入导出Excel工具类 *@Version:1.0.0 */ publicclassExcelUtil{ /** *@MethodName:listToExcel *@Description:导出Excel(可以导出到本地文件系统,也可以导出到浏览器,可自定义工作表大小) *@paramlist数据源 *@paramfieldMap类的英文属性和Excel中的中文列名的对应关系 *如果需要的是引用对象的属性,则英文属性使用类似于EL表达式的格式 *如:list中存放的都是student,student中又有college属性,而我们需要学院名称,则可以这样写 *fieldMap.put("college.collegeName","学院名称") *@paramsheetName工作表的名称 *@paramsheetSize每个工作表中记录的最大个数 *@paramout导出流 *@throwsExcelException */ publicstatic<T>voidlistToExcel( List<T>list, LinkedHashMap<String,String>fieldMap, StringsheetName, intsheetSize, OutputStreamout )throwsExcelException{ if(list.size()==0||list==null){ thrownewExcelException("数据源中没有任何数据"); } if(sheetSize>65535||sheetSize<1){ sheetSize=65535; } //创建工作簿并发送到OutputStream指定的地方 WritableWorkbookwwb; try{ wwb=Workbook.createWorkbook(out); //因为的Excel一个工作表最多可以有65536条记录,除去列头剩下65535条 //所以如果记录太多,需要放到多个工作表中,其实就是个分页的过程 //1.计算一共有多少个工作表 doublesheetNum=Math.ceil(list.size()/newInteger(sheetSize).doubleValue()); //2.创建相应的工作表,并向其中填充数据 for(inti=0;i<sheetNum;i++){ //如果只有一个工作表的情况 if(1==sheetNum){ WritableSheetsheet=wwb.createSheet(sheetName,i); fillSheet(sheet,list,fieldMap,0,list.size()-1); //有多个工作表的情况 }else{ WritableSheetsheet=wwb.createSheet(sheetName+(i+1),i); //获取开始索引和结束索引 intfirstIndex=i*sheetSize; intlastIndex=(i+1)*sheetSize-1>list.size()-1?list.size()-1:(i+1)*sheetSize-1; //填充工作表 fillSheet(sheet,list,fieldMap,firstIndex,lastIndex); } } wwb.write(); wwb.close(); }catch(Exceptione){ e.printStackTrace(); //如果是ExcelException,则直接抛出 if(einstanceofExcelException){ throw(ExcelException)e; //否则将其它异常包装成ExcelException再抛出 }else{ thrownewExcelException("导出Excel失败"); } } } /** *@MethodName:listToExcel *@Description:导出Excel(可以导出到本地文件系统,也可以导出到浏览器,工作表大小为支持的最大值) *@paramlist数据源 *@paramfieldMap类的英文属性和Excel中的中文列名的对应关系 *@paramout导出流 *@throwsExcelException */ publicstatic<T>voidlistToExcel( List<T>list, LinkedHashMap<String,String>fieldMap, StringsheetName, OutputStreamout )throwsExcelException{ listToExcel(list,fieldMap,sheetName,65535,out); } /** *@MethodName:listToExcel *@Description:导出Excel(导出到浏览器,可以自定义工作表的大小) *@paramlist数据源 *@paramfieldMap类的英文属性和Excel中的中文列名的对应关系 *@paramsheetSize每个工作表中记录的最大个数 *@paramresponse使用response可以导出到浏览器 *@throwsExcelException */ publicstatic<T>voidlistToExcel( List<T>list, LinkedHashMap<String,String>fieldMap, StringsheetName, intsheetSize, HttpServletResponseresponse )throwsExcelException{ //设置默认文件名为当前时间:年月日时分秒 StringfileName=newSimpleDateFormat("yyyyMMddhhmmss").format(newDate()).toString(); //设置response头信息 response.reset(); response.setContentType("application/vnd.ms-excel");//改成输出excel文件 response.setHeader("Content-disposition","attachment;filename="+fileName+".xls"); //创建工作簿并发送到浏览器 try{ OutputStreamout=response.getOutputStream(); listToExcel(list,fieldMap,sheetName,sheetSize,out); }catch(Exceptione){ e.printStackTrace(); //如果是ExcelException,则直接抛出 if(einstanceofExcelException){ throw(ExcelException)e; //否则将其它异常包装成ExcelException再抛出 }else{ thrownewExcelException("导出Excel失败"); } } } /** *@MethodName:listToExcel *@Description:导出Excel(导出到浏览器,工作表的大小是支持的最大值) *@paramlist数据源 *@paramfieldMap类的英文属性和Excel中的中文列名的对应关系 *@paramresponse使用response可以导出到浏览器 *@throwsExcelException */ publicstatic<T>voidlistToExcel( List<T>list, LinkedHashMap<String,String>fieldMap, StringsheetName, HttpServletResponseresponse )throwsExcelException{ listToExcel(list,fieldMap,sheetName,65535,response); } /** *@MethodName:excelToList *@Description:将Excel转化为List *@paramin:承载着Excel的输入流 *@paramsheetIndex:要导入的工作表序号 *@paramentityClass:List中对象的类型(Excel中的每一行都要转化为该类型的对象) *@paramfieldMap:Excel中的中文列头和类的英文属性的对应关系Map *@paramuniqueFields:指定业务主键组合(即复合主键),这些列的组合不能重复 *@return:List *@throwsExcelException */ publicstatic<T>List<T>excelToList( InputStreamin, StringsheetName, Class<T>entityClass, LinkedHashMap<String,String>fieldMap, String[]uniqueFields )throwsExcelException{ //定义要返回的list List<T>resultList=newArrayList<T>(); try{ //根据Excel数据源创建WorkBook Workbookwb=Workbook.getWorkbook(in); //获取工作表 Sheetsheet=wb.getSheet(sheetName); //获取工作表的有效行数 intrealRows=0; for(inti=0;i<sheet.getRows();i++){ intnullCols=0; for(intj=0;j<sheet.getColumns();j++){ CellcurrentCell=sheet.getCell(j,i); if(currentCell==null||"".equals(currentCell.getContents().toString())){ nullCols++; } } if(nullCols==sheet.getColumns()){ break; }else{ realRows++; } } //如果Excel中没有数据则提示错误 if(realRows<=1){ thrownewExcelException("Excel文件中没有任何数据"); } Cell[]firstRow=sheet.getRow(0); String[]excelFieldNames=newString[firstRow.length]; //获取Excel中的列名 for(inti=0;i<firstRow.length;i++){ excelFieldNames[i]=firstRow[i].getContents().toString().trim(); } //判断需要的字段在Excel中是否都存在 booleanisExist=true; List<String>excelFieldList=Arrays.asList(excelFieldNames); for(StringcnName:fieldMap.keySet()){ if(!excelFieldList.contains(cnName)){ isExist=false; break; } } //如果有列名不存在,则抛出异常,提示错误 if(!isExist){ thrownewExcelException("Excel中缺少必要的字段,或字段名称有误"); } //将列名和列号放入Map中,这样通过列名就可以拿到列号 LinkedHashMap<String,Integer>colMap=newLinkedHashMap<String,Integer>(); for(inti=0;i<excelFieldNames.length;i++){ colMap.put(excelFieldNames[i],firstRow[i].getColumn()); } //判断是否有重复行 //1.获取uniqueFields指定的列 Cell[][]uniqueCells=newCell[uniqueFields.length][]; for(inti=0;i<uniqueFields.length;i++){ intcol=colMap.get(uniqueFields[i]); uniqueCells[i]=sheet.getColumn(col); } //2.从指定列中寻找重复行 for(inti=1;i<realRows;i++){ intnullCols=0; for(intj=0;j<uniqueFields.length;j++){ StringcurrentContent=uniqueCells[j][i].getContents(); CellsameCell=sheet.findCell(currentContent, uniqueCells[j][i].getColumn(), uniqueCells[j][i].getRow()+1, uniqueCells[j][i].getColumn(), uniqueCells[j][realRows-1].getRow(), true); if(sameCell!=null){ nullCols++; } } if(nullCols==uniqueFields.length){ thrownewExcelException("Excel中有重复行,请检查"); } } //将sheet转换为list for(inti=1;i<realRows;i++){ //新建要转换的对象 Tentity=entityClass.newInstance(); //给对象中的字段赋值 for(Entry<String,String>entry:fieldMap.entrySet()){ //获取中文字段名 StringcnNormalName=entry.getKey(); //获取英文字段名 StringenNormalName=entry.getValue(); //根据中文字段名获取列号 intcol=colMap.get(cnNormalName); //获取当前单元格中的内容 Stringcontent=sheet.getCell(col,i).getContents().toString().trim(); //给对象赋值 setFieldValueByName(enNormalName,content,entity); } resultList.add(entity); } }catch(Exceptione){ e.printStackTrace(); //如果是ExcelException,则直接抛出 if(einstanceofExcelException){ throw(ExcelException)e; //否则将其它异常包装成ExcelException再抛出 }else{ e.printStackTrace(); thrownewExcelException("导入Excel失败"); } } returnresultList; }/*<-------------------------辅助的私有方法----------------------------------------------->*/ /** *@MethodName:getFieldValueByName *@Description:根据字段名获取字段值 *@paramfieldName字段名 *@paramo对象 *@return字段值 */ privatestaticObjectgetFieldValueByName(StringfieldName,Objecto)throwsException{ Objectvalue=null; Fieldfield=getFieldByName(fieldName,o.getClass()); if(field!=null){ field.setAccessible(true); value=field.get(o); }else{ thrownewExcelException(o.getClass().getSimpleName()+"类不存在字段名"+fieldName); } returnvalue; } /** *@MethodName:getFieldByName *@Description:根据字段名获取字段 *@paramfieldName字段名 *@paramclazz包含该字段的类 *@return字段 */ privatestaticFieldgetFieldByName(StringfieldName,Class<?>clazz){ //拿到本类的所有字段 Field[]selfFields=clazz.getDeclaredFields(); //如果本类中存在该字段,则返回 for(Fieldfield:selfFields){ if(field.getName().equals(fieldName)){ returnfield; } } //否则,查看父类中是否存在此字段,如果有则返回 Class<?>superClazz=clazz.getSuperclass(); if(superClazz!=null&&superClazz!=Object.class){ returngetFieldByName(fieldName,superClazz); } //如果本类和父类都没有,则返回空 returnnull; } /** *@MethodName:getFieldValueByNameSequence *@Description: *根据带路径或不带路径的属性名获取属性值 *即接受简单属性名,如userName等,又接受带路径的属性名,如student.department.name等 * *@paramfieldNameSequence带路径的属性名或简单属性名 *@paramo对象 *@return属性值 *@throwsException */ privatestaticObjectgetFieldValueByNameSequence(StringfieldNameSequence,Objecto)throwsException{ Objectvalue=null; //将fieldNameSequence进行拆分 String[]attributes=fieldNameSequence.split("\\."); if(attributes.length==1){ value=getFieldValueByName(fieldNameSequence,o); }else{ //根据属性名获取属性对象 ObjectfieldObj=getFieldValueByName(attributes[0],o); StringsubFieldNameSequence=fieldNameSequence.substring(fieldNameSequence.indexOf(".")+1); value=getFieldValueByNameSequence(subFieldNameSequence,fieldObj); } returnvalue; } /** *@MethodName:setFieldValueByName *@Description:根据字段名给对象的字段赋值 *@paramfieldName字段名 *@paramfieldValue字段值 *@paramo对象 */ privatestaticvoidsetFieldValueByName(StringfieldName,ObjectfieldValue,Objecto)throwsException{ Fieldfield=getFieldByName(fieldName,o.getClass()); if(field!=null){ field.setAccessible(true); //获取字段类型 Class<?>fieldType=field.getType(); //根据字段类型给字段赋值 if(String.class==fieldType){ field.set(o,String.valueOf(fieldValue)); }elseif((Integer.TYPE==fieldType) ||(Integer.class==fieldType)){ field.set(o,Integer.parseInt(fieldValue.toString())); }elseif((Long.TYPE==fieldType) ||(Long.class==fieldType)){ field.set(o,Long.valueOf(fieldValue.toString())); }elseif((Float.TYPE==fieldType) ||(Float.class==fieldType)){ field.set(o,Float.valueOf(fieldValue.toString())); }elseif((Short.TYPE==fieldType) ||(Short.class==fieldType)){ field.set(o,Short.valueOf(fieldValue.toString())); }elseif((Double.TYPE==fieldType) ||(Double.class==fieldType)){ field.set(o,Double.valueOf(fieldValue.toString())); }elseif(Character.TYPE==fieldType){ if((fieldValue!=null)&&(fieldValue.toString().length()>0)){ field.set(o,Character .valueOf(fieldValue.toString().charAt(0))); } }elseif(Date.class==fieldType){ field.set(o,newSimpleDateFormat("yyyy-MM-ddHH:mm:ss").parse(fieldValue.toString())); }else{ field.set(o,fieldValue); } }else{ thrownewExcelException(o.getClass().getSimpleName()+"类不存在字段名"+fieldName); } } /** *@MethodName:setColumnAutoSize *@Description:设置工作表自动列宽和首行加粗 *@paramws */ privatestaticvoidsetColumnAutoSize(WritableSheetws,intextraWith){ //获取本列的最宽单元格的宽度 for(inti=0;i<ws.getColumns();i++){ intcolWith=0; for(intj=0;j<ws.getRows();j++){ Stringcontent=ws.getCell(i,j).getContents().toString(); intcellWith=content.length(); if(colWith<cellWith){ colWith=cellWith; } } //设置单元格的宽度为最宽宽度+额外宽度 ws.setColumnView(i,colWith+extraWith); } } /** *@MethodName:fillSheet *@Description:向工作表中填充数据 *@paramsheet工作表 *@paramlist数据源 *@paramfieldMap中英文字段对应关系的Map *@paramfirstIndex开始索引 *@paramlastIndex结束索引 */ privatestatic<T>voidfillSheet( WritableSheetsheet, List<T>list, LinkedHashMap<String,String>fieldMap, intfirstIndex, intlastIndex )throwsException{ //定义存放英文字段名和中文字段名的数组 String[]enFields=newString[fieldMap.size()]; String[]cnFields=newString[fieldMap.size()]; //填充数组 intcount=0; for(Entry<String,String>entry:fieldMap.entrySet()){ enFields[count]=entry.getKey(); cnFields[count]=entry.getValue(); count++; } //填充表头 for(inti=0;i<cnFields.length;i++){ Labellabel=newLabel(i,0,cnFields[i]); sheet.addCell(label); } //填充内容 introwNo=1; for(intindex=firstIndex;index<=lastIndex;index++){ //获取单个对象 Titem=list.get(index); for(inti=0;i<enFields.length;i++){ ObjectobjValue=getFieldValueByNameSequence(enFields[i],item); StringfieldValue=objValue==null?"":objValue.toString(); Labellabel=newLabel(i,rowNo,fieldValue); sheet.addCell(label); } rowNo++; } //设置自动列宽 setColumnAutoSize(sheet,5); } }该工具类有4个重载的导出方法和1个导入方法,大家可以根据实际情况进行选择。
总结
导入和导出方法都是通过传一个fieldMap参数(类的英文属性和Excel的中文列头的对应关系)来连接实体类和Excel的
导出的时候可以选择导出到本地文件系统或导出到浏览器,也可以自定义每个工作表的大小
导入的时候可以自定义业务主键组合uniqueFields,这样就可以检测Excel中是否有重复行了