package com.zhx.base.utils; import org.apache.poi.hssf.usermodel.HSSFWorkbook; import org.apache.poi.ss.usermodel.*; import org.apache.poi.xssf.usermodel.XSSFWorkbook; import java.io.FileInputStream; import java.io.IOException; import java.io.InputStream; import java.util.ArrayList; import java.util.List; /** * POI解析Excel */ public class ExcelReaderUtil { /** * 根據fileType不同讀取excel文件 * * @param path * @param path * @throws IOException */ public static List<List<String>> readExcel(String path) { String fileType = path.substring(path.lastIndexOf(".") + 1); // return a list contains many list List<List<String>> lists = new ArrayList<List<String>>(); //讀取excel文件 InputStream is = null; try { is = new FileInputStream(path); //獲取工作薄 Workbook wb = null; if (fileType.equals("xls")) { wb = new HSSFWorkbook(is); } else if (fileType.equals("xlsx")) { wb = new XSSFWorkbook(is); } else { return null; } //讀取第一個工作頁sheet Sheet sheet = wb.getSheetAt(0); //第一行為標題 for (Row row : sheet) { ArrayList<String> list = new ArrayList<String>(); for (Cell cell : row) { //根據不同類型轉化成字符串 cell.setCellType(Cell.CELL_TYPE_STRING); list.add(cell.getStringCellValue()); } lists.add(list); } } catch (IOException e) { e.printStackTrace(); } finally { try { if (is != null) is.close(); } catch (IOException e) { e.printStackTrace(); } } return lists; } /** * 創建Excel.xls * @param lists 需要寫入xls的數據 * @param titles 列標題 * @param name 文件名 * @return * @throws IOException */ public static Workbook creatExcel(List<List<String>> lists, String[] titles, String name) throws IOException { System.out.println(lists); //創建新的工作薄 Workbook wb = new HSSFWorkbook(); // 創建第一個sheet(頁),並命名 Sheet sheet = wb.createSheet(name); // 手動設置列寬。第一個參數表示要為第幾列設;,第二個參數表示列的寬度,n為列高的像素數。 for(int i=0;i<titles.length;i++){ sheet.setColumnWidth((short) i, (short) (35.7 * 150)); } // 創建第一行 Row row = sheet.createRow((short) 0); // 創建兩種單元格格式 CellStyle cs = wb.createCellStyle(); CellStyle cs2 = wb.createCellStyle(); // 創建兩種字體 Font f = wb.createFont(); Font f2 = wb.createFont(); // 創建第一種字體樣式(用於列名) f.setFontHeightInPoints((short) 10); f.setColor(IndexedColors.BLACK.getIndex()); f.setBoldweight(Font.BOLDWEIGHT_BOLD); // 創建第二種字體樣式(用於值) f2.setFontHeightInPoints((short) 10); f2.setColor(IndexedColors.BLACK.getIndex()); // 設置第一種單元格的樣式(用於列名) cs.setFont(f); cs.setBorderLeft(CellStyle.BORDER_THIN); cs.setBorderRight(CellStyle.BORDER_THIN); cs.setBorderTop(CellStyle.BORDER_THIN); cs.setBorderBottom(CellStyle.BORDER_THIN); cs.setAlignment(CellStyle.ALIGN_CENTER); // 設置第二種單元格的樣式(用於值) cs2.setFont(f2); cs2.setBorderLeft(CellStyle.BORDER_THIN); cs2.setBorderRight(CellStyle.BORDER_THIN); cs2.setBorderTop(CellStyle.BORDER_THIN); cs2.setBorderBottom(CellStyle.BORDER_THIN); cs2.setAlignment(CellStyle.ALIGN_CENTER); //設置列名 for(int i=0;i<titles.length;i++){ Cell cell = row.createCell(i); cell.setCellValue(titles[i]); cell.setCellStyle(cs); } if(lists == null || lists.size() == 0){ return wb; } //設置每行每列的值 for (short i = 1; i <= lists.size(); i++) { // Row 行,Cell 方格 , Row 和 Cell 都是從0開始計數的 // 創建一行,在頁sheet上 Row row1 = sheet.createRow((short)i); for(short j=0;j<titles.length;j++){ // 在row行上創建一個方格 Cell cell = row1.createCell(j); cell.setCellValue(lists.get(i-1).get(j)); cell.setCellStyle(cs2); } } return wb; } public static void main(String[] args) { String path = "d:/software/企發支付-員工信息表.xlsx"; List<List<String>> lists = readExcel(path); for (List<String> list : lists) { for (String strs : list) { System.out.println(strs); } } } }
需要導入的jar包:
<!-- POI EXCEL 文件讀寫 --> <dependency> <groupId>org.apache.poi</groupId> <artifactId>poi-excelant</artifactId> <version>3.14</version> </dependency>
准備需要讀寫的文件:
上述工具類中將每行放到一個list中,然后每行的每列放入到一個list中,這里再根據自己需求去對表中數據進行處理:
我現在要取得企業名稱(資和信***)和從第七行起開始的id、姓名、識別號存入數據庫中,這邊我只展示service層處理,mybatis進行批量插入:
public Map insEm(File file) throws FileNotFoundException { String companyName = ""; String epid = ""; List<Map> listMap = new ArrayList<>(); List<List<String>> lists = ExcelReaderUtil.readExcel(file.getPath()); for (int j = 0; j < lists.size(); j++) { List list = lists.get(j); for (int i = 0; i < list.size(); i++) { if (list.get(i).equals("企業名稱")) { companyName = list.get(i + 1).toString(); epid = employeeMapper.selEPID(companyName); break; } else if (list.get(i).equals("員工識別號")) { for (int m = j + 1; m < lists.size() - 1; m++) { Map map = new HashMap(); if (null != lists.get(m) && lists.get(m).size() > 0) { List dataList = lists.get(m); map.put("id", dataList.get(0)); map.put("epid", epid); map.put("name", dataList.get(1)); map.put("identify", dataList.get(2)); listMap.add(map); } } } } } Map dataMap = new HashMap(); dataMap.put("employees", listMap); employeeMapper.insEm(dataMap); return null; }
<insert id="insEm" parameterType="java.util.Map"> insert into qf_employee_info(epid,employee_id,user_name,phone,user_email,status,create_time,creater,create_type) value <foreach collection="employees" index="index" item="item" separator=","> (#{item.epid},"",#{item.name},#{item.identify},"",1,now(),"",2) </foreach> </insert>
最后數據庫