zoukankan      html  css  js  c++  java
  • java Workbook 读写excel

    1. 读取具体位置的excel数据

    复制代码
    package com.royan.weakey.platform.common.excel;
    
    import lombok.extern.slf4j.Slf4j;
    import org.apache.poi.hssf.usermodel.HSSFWorkbook;
    import org.apache.poi.ss.usermodel.Cell;
    import org.apache.poi.ss.usermodel.Sheet;
    import org.apache.poi.ss.usermodel.Workbook;
    
    import java.io.File;
    import java.io.FileInputStream;
    import java.io.InputStream;
    import java.util.ArrayList;
    import java.util.List;
    
    /**
     * read  a excel file of the definite path
     */
    @Slf4j
    public class ReadExcel<T> {
    
    
        public static void main(String[] args) {
            ReadExcel<String> obj = new ReadExcel<>();
            File file = new File("path");
            List<List<List>> excelList = obj.readExcel(file);
            excelList.forEach(sheet -> {
                sheet.forEach(row -> {
                    row.forEach(column -> {
                        System.out.print(column + " ");
                    });
                    System.out.println();
                });
    
            });
    
        }
    
    
        public List<List<List>> readExcel(File file) {
            try {
                //create input stream ,get data
                InputStream stream = new FileInputStream(file);
                Workbook wb = new HSSFWorkbook(stream);
                int sheetSize = wb.getNumberOfSheets();
                List<List<List>> sheetList = new ArrayList<>();
                for (int i = 0; i < sheetSize; i++) {
                    List<List> outerList = new ArrayList<>();
                    Sheet sheet = wb.getSheetAt(i);
                    for (int j = 0; j <= sheet.getLastRowNum(); j++) {
                        List<String> list = new ArrayList<>();
                        for (int c = 0; c < sheet.getRow(j).getLastCellNum(); c++) {
                            Cell cell = sheet.getRow(j).getCell(c);
                            String str = cell.getStringCellValue();
                            list.add(str);
                        }
                        outerList.add(list);
                    }
                    sheetList.add(outerList);
                }
                return sheetList;
    
            } catch (Exception e) {
                log.info(e.getMessage());
            }
            return null;
    
        }
    
    
    }
    复制代码

    2.将数据写入指定的位置的excel

    复制代码
    package com.royan.weakey.platform.common.excel;
    
    import com.google.common.collect.Lists;
    import com.sun.xml.internal.ws.api.streaming.XMLStreamReaderFactory;
    import lombok.extern.slf4j.Slf4j;
    import org.apache.poi.hssf.usermodel.HSSFWorkbook;
    import org.apache.poi.ss.usermodel.Cell;
    import org.apache.poi.ss.usermodel.Row;
    import org.apache.poi.ss.usermodel.Sheet;
    import org.apache.poi.ss.usermodel.Workbook;
    import org.apache.poi.xssf.usermodel.XSSFWorkbook;
    
    import java.io.*;
    import java.util.*;
    
    /**
     * writeExcel
     */
    @Slf4j
    public class WriteExcel {
        private static final String EXEL_XLS = "xls";
        private static final String EXEL_XLSX = "xlsx";
    
    
        public static void writeExcel(List<Map> dataList, String filePath, List<String> header) {
            OutputStream out = null;
            try {
    
                /**
                 read  excel document
                 */
                File file = new File(filePath);
                Workbook workbook = null;
                if (file.isDirectory()) {
                    // default create .xls
                    workbook = new HSSFWorkbook();
                } else if (file.getName().endsWith(".xls")) {
                    workbook = new HSSFWorkbook();
                } else if (file.getName().endsWith(".xlsx")) {
                    workbook = new XSSFWorkbook();
                } else
                    throw new Exception("please get  .xls .xlsx end or directory,[" + filePath + "]");
                //if only support  single sheet
                Sheet sheet = workbook.createSheet("data");
                //set excel header
                Row fisrtRow = sheet.createRow(0);
                int columnCount = header.size();
                for (int h = 0; h < columnCount; h++) {
                    fisrtRow.createCell(h).setCellValue(header.get(h));
                }
                /**
                 * fill up  data in excel
                 */
                int i = 1;
                for (Map data : dataList) {
                    Row row = sheet.createRow(i);
                    Object[] arrays = data.values().toArray();
                    for (int j = 0; j < columnCount; j++) {
                        Object obj = arrays[j];
                        row.createCell(j).setCellValue(obj != null ? obj.toString() : "");
                    }
                    i++;
                }
                out = new FileOutputStream(file.isDirectory() ? filePath + "/" + header.get(0) + ".xls" : filePath);
                workbook.write(out);
    
            } catch (Exception e) {
                log.info("exception {}", e);
            } finally {
    
                try {
                    if (out != null) {
                        out.flush();
                        out.close();
                    }
                } catch (IOException o) {
                    log.info("exception {}", o);
                }
                log.info("data output success");
            }
    
    
        }
    
    
        public static void main(String[] args) {
            Map<String, Object> dataMap = new HashMap<>();
            dataMap.put("BankName", "BankName");
            dataMap.put("Addr", "Addr");
            dataMap.put("Phone", "Phone");
            Map<String, Object> dataMap1 = new HashMap<>();
            dataMap1.put("BankName", "BankName1");
            dataMap1.put("Addr", "Addr1");
            dataMap1.put("Phone", "Phone1");
            List<String> header = new LinkedList<>();
            header.add("BankName");
            header.add("Addr");
            header.add("Phone");
            List<Map> list = new ArrayList<>();
            list.add(dataMap);
            list.add(dataMap1);
            writeExcel(list, "/Users/caibixiang/workSoftware/log", header);
    
        }
    
    }

    转自:https://www.cnblogs.com/caibixiang123/p/10164423.html

  • 相关阅读:
    bug管理工具之禅道的测试模块的使用
    bug优先级别
    Python ——报错集锦
    Python 中if __name__ == '__main__': 的作用和原理
    selenium驱动程序下载和使用流程
    Python介绍及环境搭建
    自动化测试基础篇--Selenium发送测试报告邮件
    自动化测试基础篇--Selenium cookie操作
    自动化测试基础篇--Selenium等待时间
    批量建库脚本
  • 原文地址:https://www.cnblogs.com/javalinux/p/15693005.html
Copyright © 2011-2022 走看看