數據庫導入導出表頭
之前的工具類GenerateExcelToFile新增兩個導出這種二級表頭方法
package com.njry.utils;import cn.hutool.core.util.IdUtil;
import com.njry.config.FileProperties;
import com.njry.exception.BadRequestException;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.util.IOUtils;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;public class GenerateExcelToFile {public static final String SYS_TEM_DIR = System.getProperty("java.io.tmpdir") + File.separator;/*** 根據導出表在后面添加每個原子的基本值和目標值* @param list 數據庫查詢出來的信息* @param atomMbListAllString 后面動態添加的原子(要動態生成二級表頭) 原子id,模版名稱,維度名稱三個用下劃線分割,在導入的時候取數據用 例如:AT10681_0627-2_測試,AT10600_水電費debug_按開戶廳* @param tableHead 基本表頭 例如list 三個寫死的表頭:組織ID,組織名稱,組織等級* @param type 導出的excel類型 1 xlsx 2 xls* @param response* @throws IOException*/public static void dataToExcel(List<Map<String, Object>> list, List<String> atomMbListAllString, List<String> tableHead, int type, HttpServletResponse response) throws IOException{String tempPath = SYS_TEM_DIR + IdUtil.fastSimpleUUID() + ".xlsx";File file = new File(tempPath);Workbook wb = null;if(type == 1){wb = new XSSFWorkbook();}if(type == 2){wb = new HSSFWorkbook();}Workbook exportWorkbook = exportMultiExcel(wb, list, atomMbListAllString, tableHead, type);
// 臨時文件 寫出流FileOutputStream outStream = new FileOutputStream(file);// 寫入Workbook到文件exportWorkbook.write(outStream);
// 也可以通過流獲取大小
// long size = outStream.getChannel().size();// 強制刷新文件流,確保所有數據都被寫入到文件中outStream.flush();outStream.close();//response為HttpServletResponse對象response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8");//test.xls是彈出下載對話框的文件名,不能為中文,中文請自行編碼response.setHeader("Content-Disposition", "attachment;filename=file.xlsx");ServletOutputStream out = response.getOutputStream();FileInputStream fileInputStream = new FileInputStream(file);IOUtils.copy(fileInputStream,out);file.deleteOnExit();fileInputStream.close();}/**** @param wb 操作的Workbook* @param list 數據庫查詢出來的信息* @param atomMbListAllString 后面動態添加的原子(要動態生成二級表頭)* @param tableHead 基本表頭* @param type 導出的excel類型 1 xlsx 2 xls* @return*/public static Workbook exportMultiExcel(Workbook wb,List<Map<String, Object>> list, List<String> atomMbListAllString, List<String> tableHead, int type){HSSFWorkbook HSSwb = null;XSSFWorkbook XSSwb = null;if(type == 1){XSSwb = (XSSFWorkbook)wb;}if(type == 2){HSSwb = (HSSFWorkbook)wb;}// 創建樣式CellStyle style = wb.createCellStyle();style.setAlignment(HorizontalAlignment.CENTER); // 水平居中style.setVerticalAlignment(VerticalAlignment.CENTER); // 垂直居中if(type == 1){XSSFSheet sheet = XSSwb.createSheet();
// 處理sheet的表頭
// XSSFRow row = sheet.createRow((short) 0);(表頭兩行)
// 合并第一行和第二行的第一個單元格
// int startrow = 0;
// int endrow = 1;
// int startcell = 1;
// int endcell = 1;
// sheet.addMergedRegion(new CellRangeAddress(startrow, endrow, startcell, endcell));// 處理表頭(合并第一行和第二行)XSSFRow row = sheet.createRow((short) 0);XSSFRow row1 = sheet.createRow((short) 1);int size = tableHead.size();//固定表頭長度for (int i = 0; i < size; i++) {int startrow = 0;int endrow = 1;int startcell = i;int endcell = i;
// 在合并單元格第一個寫內容XSSFCell cell = row.createCell(i);cell.setCellValue(tableHead.get(i));sheet.addMergedRegion(new CellRangeAddress(startrow, endrow, startcell, endcell));cell.setCellStyle(style);}int asyncSize = atomMbListAllString.size() * 2;//動態表頭長度for (int j = size; j < asyncSize+size; j += 2) {int startrow = 0;int endrow = 0;int startcell = j;int endcell = j + 1;
// 在合并單元格第一個寫內容XSSFCell cell = row.createCell(j);int circulationNum = (j - size) / 2;//循環次數cell.setCellValue(atomMbListAllString.get(circulationNum));
// 合并第二行寫固定的基本值和目標值XSSFCell cellMerge1 = row1.createCell(j);cellMerge1.setCellValue("基本值");XSSFCell cellMerge2 = row1.createCell(j+1);cellMerge2.setCellValue("目標值");sheet.addMergedRegion(new CellRangeAddress(startrow, endrow, startcell, endcell));cellMerge1.setCellStyle(style);cellMerge2.setCellStyle(style);cell.setCellStyle(style);}
// 寫入數據(固定表頭下面內容)for (int m = 0; m < list.size(); m++) {
// 已經添加的兩個行XSSFRow rowTemp = sheet.createRow(m + 2);
// 每一行數據的map,map對應的key就是固定表頭Map<String, Object> stringObjectMap = list.get(m);Cell dataCell = null;for (int n = 0; n < tableHead.size(); n++) {dataCell = rowTemp.createCell(n);dataCell.setCellValue(StringUtils.notEmpty(stringObjectMap.get(tableHead.get(n))));}}}if(type == 2){HSSFSheet sheet = HSSwb.createSheet();
// 處理sheet的表頭
// XSSFRow row = sheet.createRow((short) 0);(表頭兩行)
// 合并第一行和第二行的第一個單元格
// int startrow = 0;
// int endrow = 1;
// int startcell = 1;
// int endcell = 1;
// sheet.addMergedRegion(new CellRangeAddress(startrow, endrow, startcell, endcell));// 處理表頭(合并第一行和第二行)HSSFRow row = sheet.createRow((short) 0);HSSFRow row1 = sheet.createRow((short) 1);int size = tableHead.size();//固定表頭長度for (int i = 0; i < size; i++) {int startrow = 0;int endrow = 1;int startcell = i;int endcell = i;
// 在合并單元格第一個寫內容HSSFCell cell = row.createCell(i);cell.setCellValue(tableHead.get(i));sheet.addMergedRegion(new CellRangeAddress(startrow, endrow, startcell, endcell));cell.setCellStyle(style);}int asyncSize = atomMbListAllString.size() * 2;//動態表頭長度for (int j = size; j < asyncSize+size; j += 2) {int startrow = 0;int endrow = 1;int startcell = j;int endcell = j + 1;
// 在合并單元格第一個寫內容HSSFCell cell = row.createCell(j);int circulationNum = (j - size) / 2;//循環次數cell.setCellValue(atomMbListAllString.get(circulationNum));
// 合并第二行寫固定的基本值和目標值HSSFCell cellMerge1 = row1.createCell(j);cellMerge1.setCellValue("基本值");HSSFCell cellMerge2 = row1.createCell(j+1);cellMerge2.setCellValue("目標值");sheet.addMergedRegion(new CellRangeAddress(startrow, endrow, startcell, endcell));cellMerge1.setCellStyle(style);cellMerge2.setCellStyle(style);cell.setCellStyle(style);}
// 寫入數據(固定表頭下面內容)for (int m = 0; m < list.size(); m++) {
// 已經添加的兩個行HSSFRow rowTemp = sheet.createRow(m + 2);
// 每一行數據的map,map對應的key就是固定表頭Map<String, Object> stringObjectMap = list.get(m);Cell dataCell = null;for (int n = 0; n < tableHead.size(); n++) {dataCell = rowTemp.createCell(n);dataCell.setCellValue(StringUtils.notEmpty(stringObjectMap.get(tableHead.get(n))));}}}return wb;}/**** @param list 數據庫查詢出來的信息* @param tableHead 數據庫的表頭(簡單表頭)* @param dirPath 保存的文件路徑* @param fileSuffix 生成的excel格式后綴* @param separator 生成的excel帶水印格要多加一層文件路徑* @throws IOException*/public static Map<String, Object> dataToExcel(List<Map<String, Object>> list,List<String> tableHead,String dirPath, String fileSuffix, String separator) throws IOException {Map<String, Object> resultMap = new HashMap<>();Workbook wb = null;FileInputStream tempInput = null;String watermarkPath = dirPath + "watermarkPath" + separator;int type = 3;//默認用type區分一個sheet有多少行if("xlsx".equals(fileSuffix)){wb = new XSSFWorkbook();type = 1;}if("xls".equals(fileSuffix)){wb = new HSSFWorkbook();type = 2;}Workbook exportWorkbook = export(wb, list, tableHead, type);String attachmentId = IdUtil.simpleUUID();String fileSize = "";String to_file_name = attachmentId + "." + fileSuffix; // 結果文件名稱
// 判斷保存的文件路徑是否存在,不存在就創建File outFileExist = new File(dirPath);if(!outFileExist.exists()){outFileExist.mkdirs();}File outFile = new File(dirPath, to_file_name);try {FileOutputStream outStream = new FileOutputStream(outFile);// 寫入Workbook到文件exportWorkbook.write(outStream);
// 也可以通過流獲取大小
// long size = outStream.getChannel().size();// 強制刷新文件流,確保所有數據都被寫入到文件中outStream.flush();// 獲取文件對象File outputFile = new File(dirPath, to_file_name);long length = outputFile.length();fileSize = length + " bytes";outStream.close();} catch (Exception e) {throw new BadRequestException("導出結果文件異常:" + e);}// 將有數據的excel的文件再加水印File tempFile = new File(dirPath, to_file_name);tempInput = new FileInputStream(tempFile);XSSFWorkbook xSSFWorkbook = new XSSFWorkbook(tempInput);
// 將導出的數據加水印放到另一個文件watermarkPath里面File watermarkFileExist = new File(watermarkPath, to_file_name);PoiSecurity.addWatermarkToXlsx(new String[]{"test"},xSSFWorkbook,watermarkFileExist);xSSFWorkbook.close();tempInput.close();// 終止后刪除臨時文件tempFile.deleteOnExit();
// 處理文件信息返回resultMap.put("attachmentId",attachmentId);resultMap.put("fileName",to_file_name);resultMap.put("fileType","excel");resultMap.put("fileExtension",fileSuffix);resultMap.put("fileSize",fileSize);resultMap.put("attachmentPath",dirPath);return resultMap;}/**** @param wb 操作的Workbook* @param list 數據庫查詢出來的信息* @param tableHead 數據庫的表頭(簡單表頭)* @param type excel類型 xlsx 1 xls 2 -------區分sheet最大告訴* @return*/public static Workbook export(Workbook wb,List<Map<String, Object>> list, List<String> tableHead,int type) {HSSFWorkbook HSSwb = null;XSSFWorkbook XSSwb = null;
// 不定義sheet名字,自生成
// Excel 2003及更早的版本中,行數上限是65,536行
// 2007開始,行數上限增加到了1,048,576行int maxRow = 49999;//去除一個表頭行if(type == 1){maxRow = 1048575;//去除一個表頭行XSSwb = (XSSFWorkbook)wb;}if(type == 2){maxRow = 65535;//去除一個表頭行HSSwb = (HSSFWorkbook)wb;}maxRow = 49999;//去除一個表頭行(無論啥格式默認都是50000一個sheet)
// 處理數據需要多少個sheetint size = list.size();int result = size / maxRow + 1;if(result == 0){result = 1;}
// 循環sheetfor (int i = 0; i < result; i++) {if(type == 1){XSSFSheet sheet = XSSwb.createSheet();// 處理每個sheet的表頭XSSFRow row = sheet.createRow((short) 0);Cell cell = null;for (int j = 0; j < tableHead.size(); j++) {cell = row.createCell(j);
// cell.setCellStyle(headStyle);cell.setCellValue(tableHead.get(j));}
// 寫入數據for (int n = 0 + maxRow * i; n < maxRow * (i + 1); n++) {
// 判斷數據list的大小是否大于要創建的行if(size - 1 >= n ){//下面list.get(n)就取不到數據,不應該繼續創建行 size 14 get(n)時候 n只能到13row = sheet.createRow(n % maxRow + 1);Cell dataCell = null;for (int m = 0; m < tableHead.size(); m++) {dataCell = row.createCell(m);dataCell.setCellValue(StringUtils.notEmpty(list.get(n).get(tableHead.get(m))));}}}}if(type == 2){HSSFSheet sheet = HSSwb.createSheet();// 處理每個sheet的表頭HSSFRow row = sheet.createRow((short) 0);Cell cell = null;for (int j = 0; j < tableHead.size(); j++) {cell = row.createCell(j);
// cell.setCellStyle(headStyle);cell.setCellValue(tableHead.get(j));}
// 寫入數據for (int n = 0 + maxRow * i; n < maxRow * (i + 1); n++) {
// 判斷數據list的大小是否大于要創建的行if(size - 1 >= n ){//下面list.get(n)就取不到數據,不應該繼續創建行 size 14 get(n)時候 n只能到13row = sheet.createRow(n % maxRow + 1);Cell dataCell = null;for (int m = 0; m < tableHead.size(); m++) {dataCell = row.createCell(m);dataCell.setCellValue(StringUtils.notEmpty(list.get(n).get(tableHead.get(m))));}}}}}return wb;}
}
處理之前導出的模板,把用戶輸入的基本值和目標值再放回數據庫里
解析excel保存到表里(貼一個實現層代碼)
@Override@Transactional(rollbackFor = Exception.class)public void excelImportBatch(MultipartFile file,Long sceneId,String month) throws Exception {
// 保存在 T_ATOM_MB_CFG_ITEM 但是需要目標值,
// 前端傳了場景id,以及excel里面原子id_模式名稱_維度名稱 有atomId,模式名稱,維度名稱
// 因此要根據atomId,模式名稱,維度名稱以及場景id 獲取目標idString extension = file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf(".") + 1).toLowerCase(); // 獲取輸入流try (java.io.InputStream fis = file.getInputStream();Workbook workbook = "xlsx".equals(extension) ? new XSSFWorkbook(fis) : new HSSFWorkbook(fis);){Sheet sheet = workbook.getSheetAt(0);//獲取Sheet中的合并單元格信息(為下文HeaderCell下屬性判斷使用)HeaderRegion[] headerRegions = new HeaderRegion[sheet.getNumMergedRegions()];for(int k = 0; k < sheet.getNumMergedRegions(); k++){HeaderRegion headerRegion = null;CellRangeAddress region = sheet.getMergedRegion(k);headerRegion = new HeaderRegion();int firstRow = region.getFirstRow();int lastRow = region.getLastRow();int firstColumn = region.getFirstColumn();int lastColumn = region.getLastColumn();headerRegion.setTargetRowFrom(firstRow);headerRegion.setTargetRowTo(lastRow);headerRegion.setTargetColumnFrom(firstColumn);headerRegion.setTargetColumnTo(lastColumn);Cell cell = sheet.getRow(firstRow).getCell(firstColumn);headerRegion.setText(cell.getStringCellValue());headerRegion.setColLength(1 + (lastColumn - firstColumn));headerRegion.setRowLength(1 + (lastRow - firstRow));headerRegions[k] = headerRegion;}
// 找到不是基礎表頭的合并單元格(前三列是基礎數據表頭合并單元格,要獲取后面動態生成的合并單元格)int size = 0;//后面動態生成的個數int sizeLength = 0;//動態占用列數int baseSize = 0;//基礎表頭Map<String,List<AtomMbCfgItem>> map = new LinkedHashMap<>();//key放動態生成的目標id(一個映射表)Map<Integer,String> cellMap = new LinkedHashMap<>();//key列對應的目標id(一個映射表,方便下面循環excel的列時候找到所屬的目標id)for(HeaderRegion item : headerRegions){
// 同一列,基礎表頭if(item.getTargetColumnFrom() == item.getTargetColumnTo()){baseSize++;}else{size++;sizeLength += item.getColLength();int targetColumnFrom = item.getTargetColumnFrom();int targetColumnTo = item.getTargetColumnTo();String text = item.getText();String[] nameText = text.split("_");String atomId = nameText[0];//atomIdString modeName = nameText[1];//模式名稱String dimenName = nameText[2];//維度名稱
// T_ATOM_MB_CFG表里的mb_id 是numberList<AtomMbCfg> atomMbCfgs = atomMbCfgMapper.queryMBId(sceneId, atomId, modeName, dimenName);if(atomMbCfgs == null || atomMbCfgs.size() == 0){throw new BadRequestException("導入的目標不對,可能不是當前場景的模版");}AtomMbCfg atomMbCfg = atomMbCfgs.get(0);String mbId = atomMbCfg.getMbId();cellMap.put(targetColumnFrom,mbId);cellMap.put(targetColumnTo,mbId);List<AtomMbCfgItem> temp = new ArrayList<>();map.put(mbId,temp);}}String orgid=null;String orgName="";int orglevel= -1;String orglevelName="";int baseval= 0;int chalval= 0;// 必輸驗證提示信息StringBuffer error = new StringBuffer();// 遍歷數據行(這里知道是從第三行是實際數據行)for (int i = 2; i <= sheet.getLastRowNum(); i++){int n = i + 1;//實際行數(顯示報錯用的)StringBuffer err = new StringBuffer();Row row = sheet.getRow(i);if (row == null)continue;orgid = row.getCell(0).toString().trim();if(orgid.length() == 0){err.append("\r\n第[" + n + "]行"+ 1 +"列沒有值!");}orgName = row.getCell(1).toString().trim();//獲取也存入表里if(orgName.length() == 0){err.append("\r\n第[" + n + "]行"+ 2 +"列沒有值!");}orglevelName = row.getCell(2).toString().trim();if(orglevelName.length() == 0){err.append("\r\n第[" + n + "]行"+ 3 +"列沒有值!");}
// 不循環數據固定表頭,直接上面寫死for (int j = 3; j < baseSize+sizeLength; j += 2) {String mbId = cellMap.get(j);//獲取動態表頭的目標idList<AtomMbCfgItem> atomMbCfgItems = map.get(mbId);//當前目標id的列表AtomMbCfgItem atomMbCfgitem = new AtomMbCfgItem();String basevalStr = row.getCell(j).toString().trim();if(basevalStr.endsWith(".0")){basevalStr = basevalStr.substring(0, basevalStr.length() - 2);baseval = Integer.parseInt(basevalStr);}String chalvalStr = row.getCell(j+1).toString().trim();if(chalvalStr.endsWith(".0")){chalvalStr = chalvalStr.substring(0, chalvalStr.length() - 2);chalval = Integer.parseInt(chalvalStr);}
// baseval = Integer.parseInt(basevalStr);
// chalval = Integer.parseInt(chalvalStr);if(baseval < 0){err.append("\r\n第[" + n + "]行"+ j+1 +"列數據為負數!");}if(chalval < 0){err.append("\r\n第[" + n + "]行"+ j+2 +"列為負數!");}// 為啥還要獲取一遍,不用excel里面的值 根據組織名稱和組織等級獲取組織id并不唯一
// select * from t_report_org_template_1
// where org_name='黨政軍組' and
// org_level= 3
// orgid= atomMbCfgMapper.selectOrgIdByName(orgName,orglevelName,atomMbCfgcx.getAtomId());atomMbCfgitem.setMbId(mbId);atomMbCfgitem.setMonth(month);if(orgid!=null && !orgid.isEmpty()){atomMbCfgitem.setOrgId(orgid);}// 在t_atom_template_level只有原子是tempalte_id 是1的才能獲取到組織層級 先寫死1 查詢組織層級
// List<Integer> integers = atomMbCfgMapper.selectOrgLevel(orglevelName, atomMbCfgcx.getAtomId());
//
// if(integers.size()>0){
// orglevel = integers.get(0);
// atomMbCfgitem.setOrgLevel(orglevel);
// }else{
// err.append("\r\n第[" + n + "]無對應組織level,請檢查!");
// }// 根據組織名稱和 先寫死的模版id 是1 查找組織等級Integer integer = atomMbCfgMapper.selectOrgLevelByorglevelNameTemplateId(orglevelName, "1");atomMbCfgitem.setOrgLevel(integer);atomMbCfgitem.setBaseVal(baseval);atomMbCfgitem.setChalVal(chalval);atomMbCfgitem.setCreateBy(SecurityUtils.getCurrentUsername());atomMbCfgitem.setCreateTime(new Date());if ("".equals(err.toString())){atomMbCfgItems.add(atomMbCfgitem);}else{error.append(err);error.append("\r\n請修改!");throw new BadRequestException(error.toString());}}}for (String key : map.keySet()) {List<AtomMbCfgItem> atomMbCfgItems = map.get(key);System.out.println(key + " :" + atomMbCfgItems);atomMbCfgMapper.saveItam(atomMbCfgItems);}// 文件流關閉fis.close();}}