tobacco-back.git

package cn.exrick.xboot.core.common.utils;
 
import cn.hutool.core.util.StrUtil;
import jxl.Cell;
import jxl.Sheet;
import jxl.Workbook;
import jxl.read.biff.BiffException;
import jxl.write.Label;
import jxl.write.WritableSheet;
import jxl.write.WritableWorkbook;
 
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.lang.reflect.Field;
import java.text.SimpleDateFormat;
import java.util.*;
import java.util.Map.Entry;
 
/**
 * 通用的excel导入导出工具,由于时间仓促所以这些代码中有许多不尽人意的地方。
 * 现在只能说是实现的功能。后期应该对逻辑和方法代码进行细致的优化。
 */
// FIXME: 2017/9/12 这个类的代码自认为写的很烂,先凑活用吧!!!!!!!!!!
public final class ExcelUtil {
 
    private ExcelUtil() {
    }
 
    /**
     * function 用于实现将后台数据导入到Excel中
     * 
     * @param <T>
     *            模板中的类型
     * @param list
     *            装在数据的list
     * @param fieldMap
     *            Excel中的数据容器，！！值得注意的是fileMap中的key要和表格中的表头的英文名数据一一对应才行
     *            如果需要的是引用对象的某个属性，则英文属性使用的是类似于EL表达式的格式
     *            如我们存储的是student，但是在student中要存储班级名称，班级是个对象，这个时候我们就可以这样实现
     *            fileMap.put("Calzz.name","学生所在班级");
     * @param sheetName
     *            Excel中工作表的名称
     * @param sheetSize
     *            工作也得数量
     * @param out
     *            字节流输出流
     * @throws RuntimeException
     *             1--数据源没有数据
     */
    public static <T> void listToExcel(List<T> list, LinkedHashMap<String, String> fieldMap, String sheetName,
            int sheetSize, OutputStream out) throws RuntimeException {
        // 数据为空或者是数据不存在
        if (list == null || list.size() == 0) {
            throw new RuntimeException("数据源中没有任何数据！请核实");
        }
        // 一次性处理的表格sheet的范围
        if (sheetSize > 65535 || sheetSize < 1) {
            // 不符合规范
            sheetSize = 65535;
        }
        // 创建工作部， 并通过out发送
        WritableWorkbook wwb = null;
        try {
            wwb = Workbook.createWorkbook(out);// 这里的workBook包是jxl里面的
            // 计算一共有多少个工作表
            double sheetNum = Math.ceil(list.size() / new Integer(sheetSize).doubleValue());
            // 创建相应的工作表，并向其中填充数据
            for (int i = 0; i < sheetNum; i++) {
                // 判断如果只有一张工作表的情况
                if (sheetNum == 1) {
                    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) {
            // TODO: handle exception
            e.printStackTrace();
            // 如果是ExcelExpection直接抛出
            if (e instanceof RuntimeException) {
                throw (RuntimeException) e;
            } else {
                throw new RuntimeException("Excel导入失败");
            }
        }
    }
 
    /**
     * 重载的函数
     */
    public static <T> void listToExcel(List<T> list, LinkedHashMap<String, String> fieldMap, String sheetName,
            OutputStream out) throws RuntimeException {
        listToExcel(list, fieldMap, sheetName, 65535, out);
    }
 
    /**
     * @MethodName : listToExcel
     * @Description : 导出Excel（导出到浏览器，可以自定义工作表的大小）
     * @param list
     *            数据源
     * @param fieldMap
     *            类的英文属性和Excel中的中文列名的对应关系
     * @param sheetSize
     *            每个工作表中记录的最大个数
     * @param response
     *            使用response可以导出到浏览器
     * @throws RuntimeException
     */
    public static <T> void listToExcel(List<T> list, LinkedHashMap<String, String> fieldMap, String sheetName,
            int sheetSize, HttpServletResponse response) throws RuntimeException {
 
        // 设置默认文件名为当前时间：年月日时分秒
        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();
 
            // 如果是RuntimeException，则直接抛出
            if (e instanceof RuntimeException) {
                throw (RuntimeException) e;
 
                // 否则将其它异常包装成RuntimeException再抛出
            } else {
                throw new RuntimeException("导出Excel失败");
            }
        }
    }
 
    /**
     * @MethodName : listToExcel
     * @Description : 导出Excel（导出到浏览器，工作表的大小是2003支持的最大值）
     * @param list
     *            数据源
     * @param fieldMap
     *            类的英文属性和Excel中的中文列名的对应关系
     * @param response
     *            使用response可以导出到浏览器
     * @throws RuntimeException
     */
    public static <T> void listToExcel(List<T> list, LinkedHashMap<String, String> fieldMap, String sheetName,
            HttpServletResponse response) throws RuntimeException {
 
        listToExcel(list, fieldMap, sheetName, 65535, response);
    }
 
    /**
     * 
     * @param <T>
     *            模型T
     * @param in
     *            Excel的输入流
     * @param sheetName
     *            Excel文件名
     * @param entityClass
     *            实体对象
     * @param fieldMap
     *            Excel中的中文猎头和类的英文属性的对应关系MAP
     * @param uniqueFields
     *            指定业务主键组合(既复合主键) ， 这些列的组合不能重复
     * @return
     * @throws RuntimeException
     */
    public static <T> List<T> excelToList(InputStream in, String sheetName, Class<T> entityClass,
            LinkedHashMap<String, String> fieldMap, String[] uniqueFields) throws RuntimeException {
        // 定义要返回的list
        List<T> resultList = new ArrayList<T>();
        try {
            // 根据Excel数据源来创建WorkBook
            Workbook wb = Workbook.getWorkbook(in);
            // 获取工作表
            Sheet sheet = null;
            if(StrUtil.isNotBlank(sheetName)){
                sheet = wb.getSheet(sheetName);
            }else{
                sheet = wb.getSheets()[0];
            }
            // 获取工作表的有效行数 默认为0
            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++;
                }
            }
            if (realRows <= 1) {
                // 如果Excel中没有数据则提示报错
                throw new RuntimeException("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集合
            List<String> excelFieldList = Arrays.asList(excelFieldNames);
            for (String cnName : fieldMap.values()) {
                if (!excelFieldList.contains(cnName)) {
                    isExist = false;
                    break;
                }
            }
            // 如果有列名不匹配 报错
            if (!isExist) {
                throw new RuntimeException("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());
            }
            if (uniqueFields != null && uniqueFields.length > 0) {
                // 判断是否有重复行
                // 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);
                        /*
                         * 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 RuntimeException("Excel中有重复行，请检查");
                    }
                }
            }
            // 将sheet转换为list
            for (int i = 1; i < realRows; i++) {
                // 新建要转换的对象
                T entity = entityClass.newInstance();
                // 给对象中的字段赋值
                for (Entry<String, String> entry : fieldMap.entrySet()) {
                    // 获取中文字段名
                    String cnNormalName = entry.getValue();
                    // 获取英文字段名
                    String enNormalName = entry.getKey();
                    // 根据中文字段名获取列
                    int col = colMap.get(cnNormalName);
                    // 获取当前单元格的内容 去空格
                    String content = sheet.getCell(col, i).getContents().toString().trim();
                    // 给对象赋值
                    if(content!=null && !"".equals(content) && content.length()>0){
                        setFieldValueByName(enNormalName, content, entity);
                    }
                }
                resultList.add(entity);
            }
        } catch (BiffException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
            throw new RuntimeException("导入Excel失败");
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
            throw new RuntimeException("导入Excel失败");
        } catch (InstantiationException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
            throw new RuntimeException("导入Excel失败");
        } catch (IllegalAccessException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
            throw new RuntimeException("导入Excel失败");
        } catch (Exception e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
            throw new RuntimeException("导入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 RuntimeException(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) {
                field.set(o, Character.valueOf(fieldValue.toString().charAt(0)));
            } else if (Date.class == fieldType) {
                String dateStr = fieldValue.toString();
                if(dateStr.length() > 10){
                    field.set(o, new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").parse(dateStr));
                }else{
                    field.set(o, new SimpleDateFormat("yyyy-MM-dd").parse(dateStr));
                }
            } else {
                field.set(o, fieldValue);
            }
        } else {
            throw new RuntimeException(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);
    }
}