借助jxl将Excel中的数据注入到Bean中

前言

使用了Hibernate的项目中需要导入多张表的数据,但是我又不想写多次取出Excle数据放到Bean里的代码,于是写了个ExcleUtils来帮助我做这件事。

基本思路

技术上,首先肯定是要借助反射的,然后选择了jxl来操作Excle。

需要的参数上,Excle文件不能少;好像没有什么方法能够在Excel隐藏地放入对应属性名的信息(就是属性名字不会在Excle中显示出来,如果有方法能够做到,请给我留言,多谢!),于是我会需要按照Excle顺序的属性名称数组;然后我需要在Sheet页的序号以及数据从第几行开始;对于Date类型,可能需要SimpleDateFormat对象来进行转换,如果没有就使用默认的。因此这个方法应该是这样的(并且有一个重载版本):

public static <T> List<T> getDataList(File excelFile, int sheetindex,
            int start, String[] props, Class<T> clazz) throws Exception {
    return getDataList(excelFile, sheetindex, start, props, clazz, null);
}

public static <T> List<T> getDataList(File excelFile, int sheetindex,
            int start, String[] props, Class<T> clazz, SimpleDateFormat sdf) {
    //TODO
}

实现

本来想把代码拆开说说自己的思路,想了想还是整块代码贴上然后写注释把(只贴相关部分的),其实都蛮简单的。

private static final Map<Class<?>, Class<?>> pwMap = new HashMap<Class<?>, Class<?>>();

    static {
        pwMap.put(byte.class, Byte.class);
        pwMap.put(short.class, Short.class);
        pwMap.put(int.class, Integer.class);
        pwMap.put(long.class, Long.class);
        pwMap.put(float.class, Float.class);
        pwMap.put(double.class, Double.class);
        pwMap.put(boolean.class, Boolean.class);
        pwMap.put(char.class, Character.class);
    }

    private ExcelUtils() {
    }

    /**
     * <p>
     * 从Excel中获取对象列
     * </p>
     * 
     * @param excelFile
     *            文件
     * @param sheetindex
     *            第几个sheet
     * @param props
     *            属性名(按顺序) (如果有非基本类型需要实现valueOf方法)
     * @param clazz
     *            类型
     * @return
     * @throws Exception
     *      String[], Class, SimpleDateFormat)
     */
    public static <T> List<T> getDataList(File excelFile, int sheetindex,
            int startRow, int startCol, String[] props, Class<T> clazz)
            throws Exception {
        return getDataList(excelFile, sheetindex, startRow, startCol, props,
                clazz, null);
    }

    /**
     * 
     * @param excelFile
     *            Excel文件
     * @param sheetindex
     * @param start
     * @param props
     * @param clazz
     * @param sdf
     * @return
     * @throws Exception
     */
    public static <T> List<T> getDataList(File excelFile, int sheetindex,
            int startRow, int startCol, String[] props, Class<T> clazz,
            SimpleDateFormat sdf) throws Exception {
        Workbook excel = null;
        try {
            excel = Workbook.getWorkbook(excelFile);
            Sheet sheet = excel.getSheet(sheetindex);
            int n = sheet.getRows();
            List<T> result = new ArrayList<T>(n);
            for (int i = startRow; i < n; i++) {
                Cell[] row = sheet.getRow(i);
                result.add(getRowData(props, row, clazz, sdf, startCol));
            }
            return result;
        } catch (BiffException | IOException e) {
            throw new Exception("读取Excel文件出错!");
        } finally {
            excel.close();
        }
    }

    /**
     * <p>
     * 获取其中一行的数据注入到对象中
     * </p>
     * 
     * @param props
     * @param row
     * @param clazz
     * @return
     * @throws Exception
     */
    public static <T> T getRowData(String[] props, Cell[] row, Class<T> clazz,
            int startCol) throws Exception {
        return getRowData(props, row, clazz, null, startCol);
    }

    /**
     * 
     * @param props
     * @param row
     * @param clazz
     * @param sdf
     * @return
     * @throws Exception
     */
    public static <T> T getRowData(String[] props, Cell[] row, Class<T> clazz,
            SimpleDateFormat sdf, int startCol) throws Exception {
        // 属性首字母大写
        for (int i = 0; i < props.length; i++) {
            if (props[i] == null)
                continue;
            StringBuilder sb = new StringBuilder(props[i]);
            sb.setCharAt(0, Character.toUpperCase(sb.charAt(0)));
            props[i] = sb.toString();
        }
        // 实例化一个对象
        T result = clazz.newInstance();
        for (int i = 0; i < props.length; i++) {
            String propName = props[i];
            if (propName == null) {
                continue;
            }
            Class<?> type = getPropType(clazz, propName);
            // 把getter的返回类型作为参数类型获取setter
            Method setter = clazz.getMethod("set" + propName, type);
            String contents = row[i + startCol].getContents();

            if (contents != null) {
                Object val = string2Type(contents, type, sdf);
                // 执行setter
                setter.invoke(result, val);
            }
        }
        return result;
    }

    private static final SimpleDateFormat DEFALUT_SIMPLEDATEF = new SimpleDateFormat(
            "yyyy-MM-dd");

    @SuppressWarnings("unchecked")
    private static <T> T string2Type(String val, Class<T> clazz,
            SimpleDateFormat sdf) throws Exception {
        Method valueOf = null;
        // 对Date和String特殊处理
        if (String.class == clazz) {
            return (T) val;
        }
        if (Date.class == clazz) {
            return (T) (sdf != null ? sdf.parse(val) : DEFALUT_SIMPLEDATEF
                    .parse(val));
        }
        if (char.class == clazz || Character.class == clazz) {
            Character c = val.toCharArray().length > 0 ? val.toCharArray()[0]
                    : (Character.class == clazz ? null : (char) 0);
            return (T) c;
        }
        // 对基本类型做处理
        Class<?> finalclazz = clazz.isPrimitive() ? pwMap.get(clazz) : clazz;
        try {
            valueOf = finalclazz.getMethod("valueOf", String.class);
        } catch (NoSuchMethodException e) {
            throw new Exception("成员类型需要有T valueOf(String)静态方法");
        }
        return (T) valueOf.invoke(null, val);
    }

    private static Class<?> getPropType(Class<?> clazz, String propName)
            throws NoSuchMethodException, SecurityException {
        Method getter = clazz.getMethod("get" + propName);
        Class<?> type = getter.getReturnType();
        return type;
    }
posted @ 2017-05-16 13:00  -.-|  阅读(...)  评论(... 编辑 收藏