Java 数据导入导出

最新推荐文章于 2024-08-02 19:01:33 发布

如漩涡

最新推荐文章于 2024-08-02 19:01:33 发布

阅读量2.9k

点赞数 1

分类专栏： Java 文章标签： java

本文链接：https://blog.csdn.net/m0_37701381/article/details/80256135

版权

Java 专栏收录该内容

28 篇文章 0 订阅

订阅专栏

前言

之前在博客中有一篇是利用POI封装的一个Excel导入导出的框架，由于最近自己也在弄。只是封装的框架用不上，因为字段这些都是动态生成的，没有实体类加注解，所以重新做了一个，比较简单。

本来不太喜欢把介绍什么的说的太多，大家可以去POI官网看文档，也可以去这个博客说的还是比较明白的《JavaWEB–POI之EXCEL操作、优化、封装详解系列（一）–概述与原理》

直接上工具类

import com.uhope.data.export.constants.ExcelConstant;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;

/**
 * @Author: ChenBin
 * @Date: 2018/5/3/0003 11:37
 */
public class ExportUtil {


    /**
     * 生成excel模板
     *
     * @param list 模板字段集合
     * @return excel
     */
    public static Workbook export(List<HashMap<String, Object>> list) {
        Workbook workbook = new HSSFWorkbook();
        Sheet sheet = workbook.createSheet();
        Row row = sheet.createRow(0);
        CellStyle cellStyle = workbook.createCellStyle();
        DataFormat format = workbook.createDataFormat();
        cellStyle.setDataFormat(format.getFormat("@"));
        for (int i = 0; i < list.size(); i++) {
            sheet.setDefaultColumnWidth((short) 15);
            sheet.setDefaultColumnStyle(i, cellStyle);
            Cell cell = row.createCell(i);
            cell.setCellValue(list.get(i).get("name").toString());
            isRed(workbook, cell, (Boolean) list.get(i).get("bool"));
        }
        return workbook;
    }

/**
     * 导出Excel模块
     * @param response
     * @param templateName
     * @param list
     */
    public static void export(HttpServletResponse response, String templateName, List<HashMap<String, Object>> list){
        Workbook workbook = export(list);
        try {
            response.setContentType("application/vnd.ms-excel;charset=UTF-8");
            response.setCharacterEncoding("UTF-8");
            response.addHeader("Content-Disposition", "attachment;filename=" + new String((templateName + "模版").getBytes(), "ISO-8859-1") + ".xls");
            OutputStream os = response.getOutputStream();
            workbook.write(os);
            os.flush();
            os.close();
        } catch (IOException e) {
            e.printStackTrace();
        }

    }

    /**
     * 字段飞空则字体颜色为红色
     *
     * @param workbook 当前excel
     * @param cell     单元格
     * @param flag     是否非空校验
     */
    private static void isRed(Workbook workbook, Cell cell, boolean flag) {
        if (flag) {
            setFont(workbook, cell, null, ExcelConstant.COLOR_RED);
        } else {
            setFont(workbook, cell);
        }
    }

    /**
     * 设置字体格式和颜色 默认微软雅黑 默认颜色黑色
     *
     * @param workbook 当前excel
     * @param cell     单元格
     */
    private static void setFont(Workbook workbook, Cell cell) {
        setFont(workbook, cell, null, null);
    }

    /**
     * 设置字体格式和颜色
     *
     * @param workbook 当前excel
     * @param cell     单元格
     * @param fontName 字体格式 默认微软雅黑
     * @param color    字体颜色    默认黑色
     */
    private static void setFont(Workbook workbook, Cell cell, String fontName, Short color) {
        fontName = fontName == null ? ExcelConstant.FONTNAME : fontName;
        color = color == null ? ExcelConstant.COLOR_BLACK : color;
        CellStyle style = workbook.createCellStyle();
        DataFormat format = workbook.createDataFormat();
        style.setDataFormat(format.getFormat("@"));
        style.setFillForegroundColor(IndexedColors.AQUA.getIndex());
        style.setAlignment(HorizontalAlignment.CENTER);
        Font font = workbook.createFont();
        font.setFontName(fontName);
        font.setFontHeightInPoints((short) 12);
        font.setColor(color);
        style.setFont(font);
        cell.setCellStyle(style);
    }

    /**
     * 数据导入
     *
     * @param multipartFile 文件流
     * @param size   有多少单元格大小
     * @return 数据集合
     */
    public static List<List<String>> importExcel(MultipartFile multipartFile, int size) {
        Workbook workbook = null;
        try {
            workbook = getWorkBook(multipartFile.getInputStream(), multipartFile.getOriginalFilename());
        } catch (IOException e) {
            e.printStackTrace();
        }
        List<Sheet> sheets = getSheets(workbook);
        return readExcel(sheets,size);
    }

    /**
     * 判断获取不同版本的Workbook实现类
     *
     * @param stream 文件流
     * @param path   文件全称
     * @return Workbook对象
     */
    private static Workbook getWorkBook(InputStream stream, String path) {
        Workbook workbook = null;
        try {
            workbook = path.endsWith(".xls") ? (new HSSFWorkbook(stream))
                    : (path.endsWith(".xlsx") ? (new XSSFWorkbook(stream)) : (null));
        } catch (IOException e) {
            e.printStackTrace();
        }
        return workbook;
    }

    /**
     * 获取每一个sheet，以防有多sheet情况
     *
     * @param book Workbook对象
     * @return sheet集合
     */
    private static List<Sheet> getSheets(Workbook book) {
        int numberOfSheets = book.getNumberOfSheets();
        List<Sheet> sheets = new ArrayList<>();
        for (int i = 0; i < numberOfSheets; i++) {
            sheets.add(book.getSheetAt(i));
        }
        return sheets;
    }

    /**
     * 解析导入的文件数据
     *
     * @param sheets sheet集合
     * @return 解析后的数据
     */
    private static List<List<String>> readExcel(List<Sheet> sheets,int size) {
        List<List<String>> dataList = new ArrayList<>();
        for (int i = 0; i < sheets.size(); i++) {
            Sheet sheet = sheets.get(i);
            if (sheet.getLastRowNum() >= 1) {
                System.out.println("=======" + sheet.getSheetName() + "=======");
            }

            Iterator<Row> sheetIterator = sheet.rowIterator();
            while (sheetIterator.hasNext()) {
                Row row = sheetIterator.next();
                if (row == null) {
                    continue;
                }
                List<String> data = new ArrayList<>();
                for (int j = row.getFirstCellNum(); j < size; j++) {
                    Cell cell = row.getCell(j);
                    if (cell != null){
                        cell.setCellType(CellType.STRING);
                        data.add(cell.getStringCellValue());
                    }else {
                        data.add("");
                    }

                }
                dataList.add(data);
            }
        }
        return dataList;
    }

}

说明

工具类是动态生成标题以及标题字体的，为什么呢，因为数据导出之后，有些必填项什么的就是红色的字体，标题是根据传入的值进行生成，这个工具类还可以直接实现导出代码，不必再写请求头，导出的默认是.xls的格式，有数据的单元格都是文本格式；

导入的方法会根据文件后缀进行判断是做哪个格式的方法操作，遍历所有sheet，在readExcel方法中有个size参数，这个参数用来做数据行最后一个单元用的，防止有些格式少了一个单元格数据而长度超出，可以根据业务逻辑装入一个list.size也行，导入里没有格式转换，因为导出的时候全都是文本格式了，为了业务需求，需要将标题和数据做个区分，所以多创建了一个List data，将解析出来的每一行每个单元格数据都add，这样通过下标取出，get(0)就是第一行的所有标题，其余的就是数据。