JAVA 导入EXCEL数据 图片 工具类

JAVA 导入EXCEL数据 图片 工具类

仅用于记录附源码如下

import org.apache.poi.ss.usermodel.*;
import org.apache.poi.ss.usermodel.DateUtil;
import org.springframework.web.multipart.MultipartFile;

import java.io.InputStream;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.text.DateFormat;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class ExcelUtil {

    /**
     * 导入excel表格.返回的是List<Object> 的数据结构,使用反射去赋值
     *
     * @param file  导入的文件
     * @param clazz 存储数据的实体
     * @param <T>   表示这个方法是泛型方法.
     * @return
     */
    public static <T> List<T> readExcel(MultipartFile file, Class<T> clazz) {
        try {
            //最终返回数据集合
            List<T> resultList = new ArrayList<T>();
            //工作簿
            Workbook workbook = null;
            InputStream is = file.getInputStream();
            String name = file.getOriginalFilename().toLowerCase();
            //创建excel操作对象
            if (name.contains(".xlsx") || name.contains(".xls")) {
                //使用工厂方法创建.
                workbook = WorkbookFactory.create(is);
            } else {
                //非Excel类型文件
                return null;
            }
            //获得一个工作表
            Sheet sheet = workbook.getSheetAt(0);
            //获得数据的总行数
            int totalRowNum = sheet.getLastRowNum();
            //获得总列数
            int cellLength = sheet.getRow(0).getPhysicalNumberOfCells();
            //获取表头
            Row firstRow = sheet.getRow(0);
            //获取反射类的所有字段
            Field[] fields = clazz.getDeclaredFields();
            //创建字段数组 和excel行顺序一致
            Field[] newFields = new Field[cellLength];
            //获取静态方法 该方法是用于把excel的表头映射成实体字段名
            //convert方法是在反射的类中的静态方法 防止Excel中的表头和反射的类的字段顺序不一致 导致数据读取在不正确的属性上
            Method m = clazz.getDeclaredMethod("convert", String.class);
            Object ob = clazz.newInstance();
            for (int i = 0; i < cellLength; i++) {
                for (Field field : fields) {
                    Cell cell = firstRow.getCell(i);
                    //按照excel中的存储存放数组 以便后面遍历excel表格 数据一一对应
                    if (m.invoke(ob, getXCellVal(cell)).equals(field.getName())) {
                        newFields[i] = field;
                    }
                }
            }
            //从第x行开始获取
            for (int x = 1; x <= totalRowNum; x++) {
                T object = clazz.newInstance();
                //获得第i行对象
                Row row = sheet.getRow(x);
                //如果一行里的所有单元格都为空则不放进list里面
                int a = 0;
                for (int y = 0; y < cellLength; y++) {
                    if (!(row == null)) {
                        Cell cell = row.getCell(y);
                        if (cell == null) {
                            a++;
                        } else {
                            Field field = newFields[y];
                            String value = getXCellVal(cell).replace(" ", "");
                            if (value != null && !value.equals("")) {
                                //给字段设置值
                                setValue(field, value, object);
                            }
                        }
                    }
                }
                if (a != cellLength && row != null) {
                    resultList.add(object);
                }
            }
            return resultList;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

    /**
     * 给字段赋值,判断值的类型,然后转化成实体需要的类型值.
     *
     * @param field  字段
     * @param value  值
     * @param object 对象
     */
    private static void setValue(Field field, String value, Object object) {
        try {
            field.setAccessible(true);
            DateFormat fmt = new SimpleDateFormat("yyyy-MM-dd");
            if (field.getGenericType().toString().contains("Integer")) {
                field.set(object, Integer.valueOf(value));
            } else if (field.getGenericType().toString().contains("String")) {
                field.set(object, value);
            } else if (field.getGenericType().toString().contains("Date")) {
                field.set(object, fmt.parse(value));
            }
            field.setAccessible(false);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * @param cell
     * @return String
     * 获取单元格中的值
     */
    private static String getXCellVal(Cell cell) {
        DateFormat fmt = new SimpleDateFormat("yyyy-MM-dd");
        DecimalFormat df = new DecimalFormat("0.0000");
        String val = "";
        switch (cell.getCellType()) {
            case NUMERIC:
                if (DateUtil.isCellDateFormatted(cell)) {
                    //日期型
                    val = fmt.format(cell.getDateCellValue());
                } else {
                    //数字型
                    val = df.format(cell.getNumericCellValue());
                    //去掉多余的0，如最后一位是.则去掉
                    val = val.replaceAll("0+?$", "").replaceAll("[.]$", "");
                }
                break;
            case STRING:
                //文本类型
                val = cell.getStringCellValue();
                break;
            case BOOLEAN:
                //布尔型
                val = String.valueOf(cell.getBooleanCellValue());
                break;
            case BLANK:
                //空白
                val = cell.getStringCellValue();
                break;
            case ERROR:
                //错误
                val = "";
                break;
            case FORMULA:
                //公式
                try {
                    val = String.valueOf(cell.getStringCellValue());
                } catch (IllegalStateException e) {
                    val = String.valueOf(cell.getNumericCellValue());
                }
                break;
            default:
                val = cell.getRichStringCellValue() == null ? null : cell.getRichStringCellValue().toString();
        }
        return val;
    }

    /**
     * 导入excel表格,返回的List<Map<Integer, String>>数据结构.
     *
     * @param file
     * @return 返回list集合
     * @throws Exception
     */
    public static List<Map<Integer, String>> read(MultipartFile file) {
        try {
            //最终返回数据列表
            List<Map<Integer, String>> list = new ArrayList<Map<Integer, String>>();

            Workbook workbook = null;
            InputStream is = file.getInputStream();
            String name = file.getOriginalFilename().toLowerCase();
            //创建excel操作对象
            if (name.contains(".xlsx") || name.contains(".xls")) {
                workbook = WorkbookFactory.create(is);
            }
            //得到一个工作表
            Sheet sheet = workbook.getSheetAt(0);
            //获得数据的总行数
            int totalRowNum = sheet.getLastRowNum();

            //获得总列数
            int cellLength = sheet.getRow(0).getPhysicalNumberOfCells();
            Map<Integer, String> map = null;
            //获得所有数据
            //从第x行开始获取
            for (int x = 0; x <= totalRowNum; x++) {
                map = new HashMap<Integer, String>();
                //获得第i行对象
                Row row = sheet.getRow(x);
                //如果一行里的所有单元格都为空则不放进list里面
                int a = 0;
                for (int y = 0; y < cellLength; y++) {
                    if (!(row == null)) {
                        Cell cell = row.getCell(y);
                        if (cell == null) {
                            map.put(y, "");
                        } else {
                            map.put(y, getXCellVal(cell));
                        }
                        if (map.get(y) == null || "".equals(map.get(y))) {
                            a++;
                        }
                    }
                }
                if (a != cellLength && row != null) {
                    list.add(map);
                }
            }
            return list;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }
}