前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Java导入导出Excel工具类ExcelUtil

Java导入导出Excel工具类ExcelUtil

原创
作者头像
用户7705674
修改2021-09-24 10:32:11
3K0
修改2021-09-24 10:32:11
举报
文章被收录于专栏:css小迷妹css小迷妹

前段时间做的分布式集成平台项目中,许多模块都用到了导入导出Excel的功能,于是决定封装一个ExcelUtil类,专门用来处理Excel的导入和导出

本项目的持久化层用的是JPA(底层用hibernate实现),所以导入和导出也都是基于实体类的。

在编写ExcelUtil之前,在网上查了一些资料。java中用来处理Excel的第三方开源项目主要就是POI和JXL。poi功能强大,但是比较耗资源,对于大数据量的导入导出性能不是太好;jxl功能简单,但是性能比较好。

由于本项目的导入导出更多关注性能问题,而且jxl提供的功能基本也都够用了,于是选择了jxl作为支持。

实战

导出就是将List转化为Excel(listToExcel)

导入就是将Excel转化为List(excelToList)

导入导出中会出现各种各样的问题,比如:数据源为空、有重复行等,我自定义了一个ExcelException异常类,用来处理这些问题。

ExcelException类

代码语言:javascript
复制
package common.tool.excel;
public class ExcelException extends Exception {

public ExcelException() {
    // TODO Auto-generated constructor stub
}

public ExcelException(String message) {
    super(message);
    // TODO Auto-generated constructor stub
}

public ExcelException(Throwable cause) {
    super(cause);
    // TODO Auto-generated constructor stub
}

public ExcelException(String message, Throwable cause) {
    super(message, cause);
    // TODO Auto-generated constructor stub
}



}</pre> 

 

 下面就是该文的主角ExcelUtil登场了,作为一个工具类,其内的所有方法都是静态的,方便使用。


 
 
 
 
 ExcelUitl类 
 
 
/**

@author     : WH
@group      : tgb8
@Date       : 2014-1-2 下午9:13:21
@Comments   : 导入导出Excel工具类
@Version    : 1.0.0
*/

public class ExcelUtil  {

/**
 * @MethodName  : listToExcel
 * @Description : 导出Excel(可以导出到本地文件系统,也可以导出到浏览器,可自定义工作表大小)
 * @param list      数据源
 * @param fieldMap      类的英文属性和Excel中的中文列名的对应关系
 * 如果需要的是引用对象的属性,则英文属性使用类似于EL表达式的格式
 * 如:list中存放的都是student,student中又有college属性,而我们需要学院名称,则可以这样写
 * fieldMap.put("college.collegeName","学院名称")
 * @param sheetName 工作表的名称
 * @param sheetSize 每个工作表中记录的最大个数
 * @param out       导出流
 * @throws ExcelException
 */
public static <T>  void   listToExcel (
        List<T> list ,
        LinkedHashMap<String,String> fieldMap,
        String sheetName,
        int sheetSize,
        OutputStream out
        ) throws ExcelException{


    if(list.size()==0 || list==null){
        throw new ExcelException("数据源中没有任何数据");
    }

    if(sheetSize>65535 || sheetSize<1){
        sheetSize=65535;
    }

    //创建工作簿并发送到OutputStream指定的地方
    WritableWorkbook wwb;
    try {
        wwb = Workbook.createWorkbook(out);

        //因为2003的Excel一个工作表最多可以有65536条记录,除去列头剩下65535条
        //所以如果记录太多,需要放到多个工作表中,其实就是个分页的过程
        //1.计算一共有多少个工作表
        double sheetNum=Math.ceil(list.size()/new Integer(sheetSize).doubleValue());

        //2.创建相应的工作表,并向其中填充数据
        for(int i=0; i<sheetNum; i++){
            //如果只有一个工作表的情况
            if(1==sheetNum){
                WritableSheet sheet=wwb.createSheet(sheetName, i);
                fillSheet(sheet, list, fieldMap, 0, list.size()-1);

            //有多个工作表的情况
            }else{
                WritableSheet sheet=wwb.createSheet(sheetName+(i+1), i);

                //获取开始索引和结束索引
                int firstIndex=i*sheetSize;
                int lastIndex=(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 (Exception e) {
        e.printStackTrace();
        //如果是ExcelException,则直接抛出
        if(e instanceof ExcelException){
            throw (ExcelException)e;

        //否则将其它异常包装成ExcelException再抛出
        }else{
            throw new ExcelException("导出Excel失败");
        }
    }

}

/**
 * @MethodName  : listToExcel
 * @Description : 导出Excel(可以导出到本地文件系统,也可以导出到浏览器,工作表大小为2003支持的最大值)
 * @param list      数据源
 * @param fieldMap      类的英文属性和Excel中的中文列名的对应关系
 * @param out       导出流
 * @throws ExcelException
 */
public static  <T>  void   listToExcel (
        List<T> list ,
        LinkedHashMap<String,String> fieldMap,
        String sheetName,
        OutputStream out
        ) throws ExcelException{

    listToExcel(list, fieldMap, sheetName, 65535, out);

}


/**
 * @MethodName  : listToExcel
 * @Description : 导出Excel(导出到浏览器,可以自定义工作表的大小)
 * @param list      数据源
 * @param fieldMap      类的英文属性和Excel中的中文列名的对应关系
 * @param sheetSize    每个工作表中记录的最大个数
 * @param response  使用response可以导出到浏览器
 * @throws ExcelException
 */
public static  <T>  void   listToExcel (
        List<T> list ,
        LinkedHashMap<String,String> fieldMap,
        String sheetName,
        int sheetSize,
        HttpServletResponse response 
        ) throws ExcelException{

    //设置默认文件名为当前时间:年月日时分秒
    String fileName=new SimpleDateFormat("yyyyMMddhhmmss").format(new Date()).toString();

    //设置response头信息
    response.reset();          
    response.setContentType("application/vnd.ms-excel");        //改成输出excel文件
    response.setHeader("Content-disposition","attachment; filename="+fileName+".xls" );

    //创建工作簿并发送到浏览器
    try {

        OutputStream out=response.getOutputStream();
        listToExcel(list, fieldMap, sheetName, sheetSize,out );

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

        //如果是ExcelException,则直接抛出
        if(e instanceof ExcelException){
            throw (ExcelException)e;

        //否则将其它异常包装成ExcelException再抛出
        }else{
            throw new ExcelException("导出Excel失败");
        }
    }
}


/**
 * @MethodName  : listToExcel
 * @Description : 导出Excel(导出到浏览器,工作表的大小是2003支持的最大值)
 * @param list      数据源
 * @param fieldMap      类的英文属性和Excel中的中文列名的对应关系
 * @param response  使用response可以导出到浏览器
 * @throws ExcelException
 */
public static <T>  void   listToExcel (
        List<T> list ,
        LinkedHashMap<String,String> fieldMap,
        String sheetName,
        HttpServletResponse response 
        ) throws ExcelException{

    listToExcel(list, fieldMap, sheetName, 65535, response);
}

/**
 * @MethodName          : excelToList
 * @Description             : 将Excel转化为List
 * @param in                    :承载着Excel的输入流
 * @param sheetIndex        :要导入的工作表序号
 * @param entityClass       :List中对象的类型(Excel中的每一行都要转化为该类型的对象)
 * @param fieldMap          :Excel中的中文列头和类的英文属性的对应关系Map
 * @param uniqueFields  :指定业务主键组合(即复合主键),这些列的组合不能重复
 * @return                      :List
 * @throws ExcelException
 */
public static <T>  List<T>  excelToList(
        InputStream in,
        String sheetName,
        Class<T> entityClass,
        LinkedHashMap<String, String> fieldMap,
        String[] uniqueFields
        ) throws ExcelException{

    //定义要返回的list
    List<T> resultList=new ArrayList<T>();

    try {

        //根据Excel数据源创建WorkBook
        Workbook wb=Workbook.getWorkbook(in);
        //获取工作表
        Sheet sheet=wb.getSheet(sheetName);

        //获取工作表的有效行数
        int realRows=0;
        for(int i=0;i<sheet.getRows();i++){

            int nullCols=0;
            for(int j=0;j<sheet.getColumns();j++){
                Cell currentCell=sheet.getCell(j,i);
                if(currentCell==null || "".equals(currentCell.getContents().toString())){
                    nullCols++;
                }
            }

            if(nullCols==sheet.getColumns()){
                break;
            }else{
                realRows++;
            }
        }


        //如果Excel中没有数据则提示错误
        if(realRows<=1){
            throw new ExcelException("Excel文件中没有任何数据");
        }


        Cell[] firstRow=sheet.getRow(0);

        String[] excelFieldNames=new String[firstRow.length];

        //获取Excel中的列名
        for(int i=0;i<firstRow.length;i++){
            excelFieldNames[i]=firstRow[i].getContents().toString().trim();
        }

        //判断需要的字段在Excel中是否都存在
        boolean isExist=true;
        List<String> excelFieldList=Arrays.asList(excelFieldNames);
        for(String cnName : fieldMap.keySet()){
            if(!excelFieldList.contains(cnName)){
                isExist=false;
                break;
            }
        }

        //如果有列名不存在,则抛出异常,提示错误
        if(!isExist){
            throw new ExcelException("Excel中缺少必要的字段,或字段名称有误");
        }


        //将列名和列号放入Map中,这样通过列名就可以拿到列号
        LinkedHashMap<String, Integer> colMap=new LinkedHashMap<String, Integer>();
        for(int i=0;i<excelFieldNames.length;i++){
            colMap.put(excelFieldNames[i], firstRow[i].getColumn());
        } 



        //判断是否有重复行
        //1.获取uniqueFields指定的列
        Cell[][] uniqueCells=new Cell[uniqueFields.length][];
        for(int i=0;i<uniqueFields.length;i++){
            int col=colMap.get(uniqueFields[i]);
            uniqueCells[i]=sheet.getColumn(col);
        }

        //2.从指定列中寻找重复行
        for(int i=1;i<realRows;i++){
            int nullCols=0;
            for(int j=0;j<uniqueFields.length;j++){
                String currentContent=uniqueCells[j][i].getContents();
                Cell sameCell=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){
                throw new ExcelException("Excel中有重复行,请检查");
            }
        }

        //将sheet转换为list
        for(int i=1;i<realRows;i++){
            //新建要转换的对象
            T entity=entityClass.newInstance();

            //给对象中的字段赋值
            for(Entry<String, String> entry : fieldMap.entrySet()){
                //获取中文字段名
                String cnNormalName=entry.getKey();
                //获取英文字段名
                String enNormalName=entry.getValue();
                //根据中文字段名获取列号
                int col=colMap.get(cnNormalName);

                //获取当前单元格中的内容
                String content=sheet.getCell(col, i).getContents().toString().trim();

                //给对象赋值
                setFieldValueByName(enNormalName, content, entity);
            }

            resultList.add(entity);
        }
    } catch(Exception e){
        e.printStackTrace();
        //如果是ExcelException,则直接抛出
        if(e instanceof ExcelException){
            throw (ExcelException)e;

        //否则将其它异常包装成ExcelException再抛出
        }else{
            e.printStackTrace();
            throw new ExcelException("导入Excel失败");
        }
    }
    return resultList;
}





/*<-------------------------辅助的私有方法----------------------------------------------->*/
/**
 * @MethodName  : getFieldValueByName
 * @Description : 根据字段名获取字段值
 * @param fieldName 字段名
 * @param o 对象
 * @return  字段值
 */
private static  Object getFieldValueByName(String fieldName, Object o) throws Exception{

    Object value=null;
    Field field=getFieldByName(fieldName, o.getClass());

    if(field !=null){
        field.setAccessible(true);
        value=field.get(o);
    }else{
        throw new ExcelException(o.getClass().getSimpleName() + "类不存在字段名 "+fieldName);
    }

    return value;
}

/**
 * @MethodName  : getFieldByName
 * @Description : 根据字段名获取字段
 * @param fieldName 字段名
 * @param clazz 包含该字段的类
 * @return 字段
 */
private static Field getFieldByName(String fieldName, Class<?>  clazz){
    //拿到本类的所有字段
    Field[] selfFields=clazz.getDeclaredFields();

    //如果本类中存在该字段,则返回
    for(Field field : selfFields){
        if(field.getName().equals(fieldName)){
            return field;
        }
    }

    //否则,查看父类中是否存在此字段,如果有则返回
    Class<?> superClazz=clazz.getSuperclass();
    if(superClazz!=null  &&  superClazz !=Object.class){
        return getFieldByName(fieldName, superClazz);
    }

    //如果本类和父类都没有,则返回空
    return null;
}



/**
 * @MethodName  : getFieldValueByNameSequence
 * @Description : 
 * 根据带路径或不带路径的属性名获取属性值
 * 即接受简单属性名,如userName等,又接受带路径的属性名,如student.department.name等
 * 
 * @param fieldNameSequence  带路径的属性名或简单属性名
 * @param o 对象
 * @return  属性值
 * @throws Exception
 */
private static  Object getFieldValueByNameSequence(String fieldNameSequence, Object o) throws Exception{

    Object value=null;

    //将fieldNameSequence进行拆分
    String[] attributes=fieldNameSequence.split("\\.");
    if(attributes.length==1){
        value=getFieldValueByName(fieldNameSequence, o);
    }else{
        //根据属性名获取属性对象
        Object fieldObj=getFieldValueByName(attributes[0], o);
        String subFieldNameSequence=fieldNameSequence.substring(fieldNameSequence.indexOf(".")+1);
        value=getFieldValueByNameSequence(subFieldNameSequence, fieldObj);
    }
    return value; 

} 


/**
 * @MethodName  : setFieldValueByName
 * @Description : 根据字段名给对象的字段赋值
 * @param fieldName  字段名
 * @param fieldValue    字段值
 * @param o 对象
 */
private static void setFieldValueByName(String fieldName,Object fieldValue,Object o) throws Exception{

        Field field=getFieldByName(fieldName, o.getClass());
        if(field!=null){
            field.setAccessible(true);
            //获取字段类型
            Class<?> fieldType = field.getType();  

            //根据字段类型给字段赋值
            if (String.class == fieldType) {  
                field.set(o, String.valueOf(fieldValue));  
            } else if ((Integer.TYPE == fieldType)  
                    || (Integer.class == fieldType)) {  
                field.set(o, Integer.parseInt(fieldValue.toString()));  
            } else if ((Long.TYPE == fieldType)  
                    || (Long.class == fieldType)) {  
                field.set(o, Long.valueOf(fieldValue.toString()));  
            } else if ((Float.TYPE == fieldType)  
                    || (Float.class == fieldType)) {  
                field.set(o, Float.valueOf(fieldValue.toString()));  
            } else if ((Short.TYPE == fieldType)  
                    || (Short.class == fieldType)) {  
                field.set(o, Short.valueOf(fieldValue.toString()));  
            } else if ((Double.TYPE == fieldType)  
                    || (Double.class == fieldType)) {  
                field.set(o, Double.valueOf(fieldValue.toString()));  
            } else if (Character.TYPE == fieldType) {  
                if ((fieldValue!= null) && (fieldValue.toString().length() > 0)) {  
                    field.set(o, Character  
                            .valueOf(fieldValue.toString().charAt(0)));  
                }  
            }else if(Date.class==fieldType){
                field.set(o, new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").parse(fieldValue.toString()));
            }else{
                field.set(o, fieldValue);
            }
        }else{
            throw new ExcelException(o.getClass().getSimpleName() + "类不存在字段名 "+fieldName);
        }
}


/**
 * @MethodName  : setColumnAutoSize
 * @Description : 设置工作表自动列宽和首行加粗
 * @param ws
 */
private static void setColumnAutoSize(WritableSheet ws,int extraWith){
    //获取本列的最宽单元格的宽度
    for(int i=0;i<ws.getColumns();i++){
        int colWith=0;
        for(int j=0;j<ws.getRows();j++){
            String content=ws.getCell(i,j).getContents().toString();
            int cellWith=content.length();
            if(colWith<cellWith){
                colWith=cellWith;
            }
        }
        //设置单元格的宽度为最宽宽度+额外宽度
        ws.setColumnView(i, colWith+extraWith);
    }

}

/**
 * @MethodName  : fillSheet
 * @Description : 向工作表中填充数据
 * @param sheet     工作表 
 * @param list  数据源
 * @param fieldMap 中英文字段对应关系的Map
 * @param firstIndex    开始索引
 * @param lastIndex 结束索引
 */
private static <T> void fillSheet(
        WritableSheet sheet,
        List<T> list,
        LinkedHashMap<String,String> fieldMap,
        int firstIndex,
        int lastIndex
        )throws Exception{

    //定义存放英文字段名和中文字段名的数组
    String[] enFields=new String[fieldMap.size()];
    String[] cnFields=new String[fieldMap.size()];

    //填充数组
    int count=0;
    for(Entry<String,String> entry:fieldMap.entrySet()){
        enFields[count]=entry.getKey();
        cnFields[count]=entry.getValue();
        count++;
    }
    //填充表头
    for(int i=0;i<cnFields.length;i++){
        Label label=new Label(i,0,cnFields[i]);
        sheet.addCell(label);
    }

    //填充内容
    int rowNo=1;
    for(int index=firstIndex;index<=lastIndex;index++){
        //获取单个对象
        T item=list.get(index);
        for(int i=0;i<enFields.length;i++){
            Object objValue=getFieldValueByNameSequence(enFields[i], item);
            String fieldValue=objValue==null ? "" : objValue.toString();
            Label label =new Label(i,rowNo,fieldValue);
            sheet.addCell(label);
        }

        rowNo++;
    }

    //设置自动列宽
    setColumnAutoSize(sheet, 5);
}


}</pre> 

 


 
 
 该工具类有4个重载的导出方法和1个导入方法,大家可以根据实际情况进行选择。 
   
 总结 
 导入和导出方法都是通过传一个fieldMap参数(类的英文属性和Excel的中文列头的对应关系)来连接实体类和Excel的 
 导出的时候可以选择导出到本地文件系统或导出到浏览器,也可以自定义每个工作表的大小 
 导入的时候可以自定义业务主键组合uniqueFields,这样就可以检测Excel中是否有重复行了  

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 实战
    • ExcelException类
    领券
    问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档