[POI]主管:你尽快封装一套读、写excel的方法,尽量简单

简介: 本文介绍了使用Java读取和写入Excel文件的方法,包括文件类型检查、数据读取、数据有效性验证及数据导出。代码简洁明了,适合快速上手。阅读更多相关文章可进一步了解POI和EasyExcel的使用技巧。

【版权声明】未经博主同意,谢绝转载!(请尊重原创,博主保留追究权)
https://www.cnblogs.com/cnb-yuchen/p/18321864
出自【进步*于辰的博客

方法简单,也出于个人时间考虑,就未作过多注释和说明,可谓,本文只有代码,但相信你一看便懂。

1、读

1、文件兼容类型。

// 兼容文件后缀列表
private static final String FILE_TYPE;
static {
   
    // 注:XSSFWorkBook 类支持这三种后缀的文件,不过必须由xlsx文件修改后缀而来
    FILE_TYPE = ".xls/.xlsx/.csv";
}

2、文件检查。

/**
 * 文本薄文件有效性检查
 *
 * @param excelPath 文本薄文件路径
 */
public static void checkFile(String excelPath) throws Exception {
   
    /**
     * XSSFWorkBook 底层会检查:
     * 1、文件是否存在;
     * 2、文件类型:不能是目录,仅支持xls、xlsx、csv等类型文件。
     * (不过,底层异常提示不够人性化,故自主开发)
     */

    if (excelPath == null || excelPath.equals(""))
        throw new Exception("文件路径不能为空");
    File f = new File(excelPath);
    if (f.exists()) {
   
        if (f.isDirectory())
            throw new Exception("指定文件不能是目录");
    } else
        throw new Exception("文件不存在");

    // 注:程序能运行到此,说明文件存在且不是目录,即 excelPath 一定是一个文件的绝对路径,故无需再构造 File 对象获取绝对路径
    // 检查文件后缀
    String suffix = excelPath.substring(excelPath.lastIndexOf("."));
    if (FILE_TYPE.indexOf(suffix) == -1) {
   
        throw new Exception("不是文本薄文件");
    }
}

3、读取文本薄。

/**
 * 读取excelPath所指的文本薄文件
 *
 * @param excelPath  文本薄文件路径
 * @param sheetIndex 文本簿索引
 */
public static List<Map<String, String>> readExcel(String excelPath, int sheetIndex) throws Exception {
   
    List<Map<String, String>> dataList = new ArrayList<>();

    // 检查文件
    checkFile(excelPath);

    XSSFWorkbook workbook = new XSSFWorkbook(excelPath);// 获取数据到工作簿
    if (sheetIndex < 0 || sheetIndex >= workbook.getNumberOfSheets())
        throw new Exception("此文本薄条目不存在");
    XSSFSheet sheet = workbook.getSheetAt(sheetIndex);// 获取第n张表

    XSSFRow titleRow = sheet.getRow(0);// 标题行
    for (int i = 1; i < sheet.getPhysicalNumberOfRows(); i++) {
   // 数据从第二行开始
        Map<String, String> dataMap = new HashMap<>();

        XSSFRow dataRow = sheet.getRow(i);// 数据行
        if (dataRow == null) // 排除空行(当当行所有列全为空时,此行不存在)
            continue;
        if (!isValidRow(dataRow)) // 排除无效行
            continue;

        for (int j = 0; j < dataRow.getPhysicalNumberOfCells(); j++) {
   
            XSSFCell titleCell = titleRow.getCell(j);// 表头
            XSSFCell dataCell = dataRow.getCell(j);// 数据
            dataMap.put(titleCell.getStringCellValue(), getCellValue(dataCell));
        }
        dataList.add(dataMap);
    }
    workbook.close();

    return dataList;
}

4、判断是否是无效行。

/**
 * 判断数据行是否有效,
 * 判断依据:第一列是否有内容
 *
 * @param dataRow 数据行
 */
private static boolean isValidRow(XSSFRow dataRow) {
   
    XSSFCell cell0 = dataRow.getCell(0);
    if (cell0 != null)
        return true;
    return false;
}

5、获取单元格数据。

/**
 * 根据cell值类型获取值
 * 注:无论文本列是何种类型,都转为String
 *
 * @param cell 文本列
 */
protected static String getCellValue(XSSFCell cell) {
   
    CellType type = cell.getCellType();
    if (type == CellType.BLANK) {
   
        return "";
    } else if (type == CellType.NUMERIC) {
   // 数字、时间
        return cell.getNumericCellValue() + "";
    } else if (type == CellType.BOOLEAN) {
   
        return cell.getBooleanCellValue() + "";
    } else if (type == CellType.STRING) {
   
        return cell.getStringCellValue().trim();
    } else if (type == CellType.ERROR) {
   
        return cell.getErrorCellString();
    } else {
   
        return "";
    }
}

2、写

1、将封装在List<Map<String, String>>内的数据导出excel。

/**
 * 导出excel
 * @param dataList  数据集
 * @param sheetName 文本薄名称
 * @param titleArr  表头数组
 * @return 导出行数
 */
public static int writeExcel0(List<Map<String, String>> dataList, String excelPath, String sheetName, String... titleArr)
        throws Exception {
   
    XSSFWorkbook workbook = new XSSFWorkbook();
    XSSFSheet sheet0 = workbook.createSheet(sheetName);

    // 设置表头
    XSSFRow row0 = sheet0.createRow(0);
    for (int i = 0; i < titleArr.length; i++) {
   
        XSSFCell cell = row0.createCell(i);
        cell.setCellValue(titleArr[i]);
    }

    int rowN = 1;
    for (Map<String, String> map : dataList) {
   
        XSSFRow row = sheet0.createRow(rowN++);
        int colN = -1;
        while (colN++ < titleArr.length - 1) {
   
            XSSFCell cell = row.createCell(colN);
            cell.setCellValue(map.get(getTitle(row0, colN)));// 根据列索引获取列名,再获取内容
        }
    }

    workbook.write(new FileOutputStream(excelPath));// 覆盖
    workbook.close();

    return dataList.size();
}

2、返回表头名称。

/**
 * 返回表头名称
 *
 * @param titleRow 表头行
 * @param colN     列索引
 * @return
 */
private static String getTitle(XSSFRow titleRow, int colN) {
   
    return titleRow.getCell(colN).getStringCellValue();
}

3、将封装在List<String[]>内的数据导出excel。

/**
 * 导出excel
 *  注:若实参类型为List<List<String>>,源码相同,未保留是因为认为实用性不大
 */
public static int writeExcel(List<String[]> dataList, String excelPath, String sheetName, String... titleArr)
        throws Exception {
   
    List<Map<String, String>> newList = new ArrayList<>();
    dataList.forEach(e -> {
   
        Map<String, String> map = new HashMap<>();
        int i = 0;
        for (String str: e) {
   
            map.put(titleArr[i++], str);
        }
        newList.add(map);
    });
    return writeExcel0(newList, excelPath, sheetName, titleArr);
}

最后

推荐阅读:(转发)

  1. 使用POI和EasyExcel实现Excel导入和导出功能
  2. SXSSFWorkbook:生成大Excel的低内存实现

还有一篇实用博文:《公司大佬对excel导入、导出的封装,那叫一个秒啊》(转发)。

本文完结。

相关文章
|
28天前
|
数据处理 Python
Python 高级技巧:深入解析读取 Excel 文件的多种方法
在数据分析中,从 Excel 文件读取数据是常见需求。本文介绍了使用 Python 的三个库:`pandas`、`openpyxl` 和 `xlrd` 来高效处理 Excel 文件的方法。`pandas` 提供了简洁的接口,而 `openpyxl` 和 `xlrd` 则针对不同版本的 Excel 文件格式提供了详细的数据读取和处理功能。此外,还介绍了如何处理复杂格式(如合并单元格)和进行性能优化(如分块读取)。通过这些技巧,可以轻松应对各种 Excel 数据处理任务。
84 16
|
16天前
|
前端开发 JavaScript Java
导出excel的两个方式:前端vue+XLSX 导出excel,vue+后端POI 导出excel,并进行分析、比较
这篇文章介绍了使用前端Vue框架结合XLSX库和后端结合Apache POI库导出Excel文件的两种方法,并对比分析了它们的优缺点。
173 0
|
24天前
|
Java Apache
Apache POI java对excel表格进行操作(读、写) 有代码!!!
文章提供了使用Apache POI库在Java中创建和读取Excel文件的详细代码示例,包括写入数据到Excel和从Excel读取数据的方法。
28 0
|
2月前
|
存储 数据挖掘 测试技术
Python接口自动化中操作Excel文件的技术方法
通过上述方法和库,Python接口自动化中的Excel操作变得既简单又高效,有助于提升自动化测试的整体质量和效率。
32 0
|
4月前
|
存储 开发工具 git
好的git管理方法,标明项目_编号_(功能,不过还是在没有bug出现时就提交为好)+Excel表管理的格式
好的git管理方法,标明项目_编号_(功能,不过还是在没有bug出现时就提交为好)+Excel表管理的格式
|
5月前
|
Java API Spring
集成EasyPoi(一个基于POI的Excel导入导出工具)到Spring Boot项目中
集成EasyPoi(一个基于POI的Excel导入导出工具)到Spring Boot项目中
436 1
|
5月前
|
easyexcel Java API
Apache POI与easyExcel:Excel文件导入导出的技术深度分析
Apache POI与easyExcel:Excel文件导入导出的技术深度分析
|
18天前
|
数据采集 存储 JavaScript
自动化数据处理:使用Selenium与Excel打造的数据爬取管道
本文介绍了一种使用Selenium和Excel结合代理IP技术从WIPO品牌数据库(branddb.wipo.int)自动化爬取专利信息的方法。通过Selenium模拟用户操作,处理JavaScript动态加载页面,利用代理IP避免IP封禁,确保数据爬取稳定性和隐私性。爬取的数据将存储在Excel中,便于后续分析。此外,文章还详细介绍了Selenium的基本设置、代理IP配置及使用技巧,并探讨了未来可能采用的更多防反爬策略,以提升爬虫效率和稳定性。
|
3月前
|
关系型数据库 MySQL Shell
不通过navicat工具怎么把查询数据导出到excel表中
不通过navicat工具怎么把查询数据导出到excel表中
39 0