Skip to content

polysoft-nafmii / polysoft-parent

Go to a project
Commit a5f04db9 by 孙长庆
Options

导入EXCEL添加读取合并单元格功能

1 parent 5c4d0db5
Inline Side-by-side
Showing 5 changed files with 70 additions and 175 deletions
nafmii-admin/src/main/java/org/nafmii/admin/user/controller/ImportExcelController.java
package org.nafmii.admin.user.controller;
import com.alibaba.excel.EasyExcel;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.nafmii.admin.user.dto.ImportExcelDto;
import org.nafmii.admin.user.service.IImportExcelService;
import org.nafmii.common.response.ApiResponse;
import org.springframework.web.bind.annotation.PostMapping;
......@@ -13,7 +11,7 @@ import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;
import javax.annotation.Resource;
import java.util.List;
import java.io.IOException;
@RestController
@RequestMapping("/nafmii/admin/import")
......@@ -25,7 +23,7 @@ public class ImportExcelController {
@ApiOperation( value = "中心端批量导入Excel",notes = "中心端批量导入Excel")
@PostMapping("/batchImportExcel")
@ResponseBody
public ApiResponse batchImportExcel(MultipartFile file){
public ApiResponse batchImportExcel(MultipartFile file) throws IOException {
ApiResponse result = iImportExcelService.batchImportExcel(file);
return result;
}
......
nafmii-admin/src/main/java/org/nafmii/admin/user/dao/ImportExcelMapper.java
......@@ -2,10 +2,8 @@ package org.nafmii.admin.user.dao;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import org.nafmii.admin.user.dto.ImportExcelDto;
import org.nafmii.admin.user.entity.CenterUserPO;
import java.util.List;
/**
* <p>
......@@ -16,14 +14,14 @@ import java.util.List;
* @since 2021-04-27
*/
@Mapper
public interface ImportExcelMapper extends BaseMapper<CenterUserPO> {
public interface ImportExcelMapper extends BaseMapper<ImportExcelDto> {
/**
* @author:suncq
* @Function:中心端用户登陆
* @date 2021/4/27
* @Function:批量导入EXCEL数据
* @date 2021/5/6
* @ClassName:
* @version:
* @remark:
*/
int insertExcelData(ImportExcelDto data);
int insertExcelData(@Param("data") ImportExcelDto data);
}
nafmii-admin/src/main/java/org/nafmii/admin/user/service/IImportExcelService.java
package org.nafmii.admin.user.service;
import org.nafmii.admin.user.dto.ImportExcelDto;
import org.nafmii.common.response.ApiResponse;
import org.springframework.web.multipart.MultipartFile;
import java.util.List;
import java.io.IOException;
/**
* <p>
* 中心端-用户表 服务类
* 中心端批量导入Excel
* </p>
*
* @author suncq
......@@ -17,12 +16,12 @@ import java.util.List;
public interface IImportExcelService {
/**
* @author:suncq
* @Function:中心端用户登陆
* @Function:中心端批量导入Excel
* @date 2021/4/27
* @ClassName:
* @version:
* @remark:
*/
ApiResponse batchImportExcel(MultipartFile file);
ApiResponse batchImportExcel(MultipartFile file) throws IOException;
}
nafmii-admin/src/main/java/org/nafmii/admin/user/service/impl/ImportExcelServiceImpl.java
package org.nafmii.admin.user.service.impl;
import com.alibaba.excel.EasyExcel;
import lombok.extern.slf4j.Slf4j;
import org.nafmii.admin.user.dao.ImportExcelMapper;
import org.nafmii.admin.user.dto.ImportExcelDto;
import org.nafmii.admin.user.service.IImportExcelService;
import org.nafmii.common.response.ApiResponse;
import org.springframework.stereotype.Service;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
......@@ -15,23 +8,25 @@ import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.nafmii.admin.user.dao.ImportExcelMapper;
import org.nafmii.admin.user.dto.ImportExcelDto;
import org.nafmii.admin.user.service.IImportExcelService;
import org.springframework.beans.factory.annotation.Autowired;
import org.nafmii.common.response.ApiResponse;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.multipart.MultipartFile;
import javax.annotation.Resource;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import javax.annotation.Resource;
import java.util.List;
/**
* <p>
* 中心端-用户表 服务实现类
* 中心端批量导入Excel
* </p>
*
* @author suncq
......@@ -45,181 +40,91 @@ public class ImportExcelServiceImpl implements IImportExcelService {
private ImportExcelMapper importExcelMapper;
@Override
public ApiResponse batchImportExcel(MultipartFile file) {
List<ImportExcelDto> list = null;
try {
list = EasyExcel.read(file.getInputStream()).sheet(0).head(ImportExcelDto.class).doReadSync();
} catch (IOException e) {
e.printStackTrace();
}
log.info("导入后生成的数据源=============>"+list);
if (!list.isEmpty()) {
for (ImportExcelDto data : list) {
int i = importExcelMapper.insertExcelData(data);
if (i < 0) {
return new ApiResponse(ApiResponse.FAIL, ApiResponse.FAIL_TEXT);
} else {
return new ApiResponse(ApiResponse.SUCCESS, ApiResponse.SUCCESS_TEXT);
}
}
}
return new ApiResponse(ApiResponse.FAIL, "导入时数据为空，导入失败");
}
public String importExcel(InputStream inputStream, String fileName) throws Exception {
String message = "Import success";
@Transactional(rollbackFor = Exception.class)
public ApiResponse batchImportExcel(MultipartFile file) throws IOException {
List<ImportExcelDto> list = new ArrayList<ImportExcelDto>();
String fileName = file.getOriginalFilename();
byte [] byteArr=file.getBytes();
InputStream inputStream = new ByteArrayInputStream(byteArr);
boolean isE2007 = false;
//判断是否是excel2007格式
if (fileName.endsWith("xlsx")) {
isE2007 = true;
}
int rowIndex = 0;
try {
InputStream input = inputStream; //建立输入流
Workbook wb;
//根据文件格式(2003或者2007)来初始化
if (isE2007) {
wb = new XSSFWorkbook(input);
} else {
wb = new HSSFWorkbook(input);
}
Sheet sheet = wb.getSheetAt(0); //获得第一个表单
int rowCount = sheet.getLastRowNum() + 1;
for (int i = 1; i < rowCount; i++) {
rowIndex = i;
Row row;
for (int j = 0; j < 26; j++) {
if (isMergedRegion(sheet, i, j)) {
System.out.print(getMergedRegionValue(sheet, i, j) + "\t");
} else {
row = sheet.getRow(i);
System.out.print(row.getCell(j) + "\t");
InputStream input = inputStream; //建立输入流
Workbook wb;
//根据文件格式(2003或者2007)来初始化
if (isE2007) {
wb = new XSSFWorkbook(input);
} else {
wb = new HSSFWorkbook(input);
}
Sheet sheet = wb.getSheetAt(0); //获得第一个表单
int rowCount = sheet.getLastRowNum() + 1;//总行数
for (int i = 1; i < rowCount; i++) {
ImportExcelDto dto = new ImportExcelDto();
Row row;
int columns = sheet.getRow(i).getLastCellNum();//单行列数
for (int j = 0; j < columns; j++) {
if (isMergedRegion(sheet, i, j)) {
if(j == 0){
dto.setColumn1(getMergedRegionValue(sheet, i, j));
}else if(j == 1){
dto.setColumn2(getMergedRegionValue(sheet, i, j));
}else if(j == 2){
dto.setColumn3(getMergedRegionValue(sheet, i, j));
}
} else {
row = sheet.getRow(i);
if(j == 0){
dto.setColumn1(String.valueOf(row.getCell(j)));
}else if(j == 1){
dto.setColumn2(String.valueOf(row.getCell(j)));
}else if(j == 2){
dto.setColumn3(String.valueOf(toString()));
}
}
System.out.print("\n");
}
} catch (Exception ex) {
message = "Import failed, please check the data in " + rowIndex + " rows ";
list.add(dto);
}
return message;
if (!list.isEmpty()) {
for (ImportExcelDto data : list) {
int i = importExcelMapper.insertExcelData(data);
if (i < 0) {
return new ApiResponse(ApiResponse.FAIL, ApiResponse.FAIL_TEXT);
}
}
}else{
return new ApiResponse(ApiResponse.FAIL, "导入时数据为空，导入失败");
}
return new ApiResponse(ApiResponse.SUCCESS, ApiResponse.SUCCESS_TEXT);
}
/**
* 获取单元格的值
*
* @param cell
* @return
*/
public String getCellValue(Cell cell) {
if (cell == null) return "";
return cell.getStringCellValue();
}
/**
* 合并单元格处理,获取合并行
*
* @param sheet
* @return List<CellRangeAddress>
*/
public List<CellRangeAddress> getCombineCell(Sheet sheet) {
List<CellRangeAddress> list = new ArrayList<>();
//获得一个 sheet 中合并单元格的数量
int sheetmergerCount = sheet.getNumMergedRegions();
//遍历所有的合并单元格
for (int i = 0; i < sheetmergerCount; i++) {
//获得合并单元格保存进list中
CellRangeAddress ca = sheet.getMergedRegion(i);
list.add(ca);
}
return list;
}
private int getRowNum(List<CellRangeAddress> listCombineCell, Cell cell, Sheet sheet) {
int xr = 0;
int firstC = 0;
int lastC = 0;
int firstR = 0;
int lastR = 0;
for (CellRangeAddress ca : listCombineCell) {
//获得合并单元格的起始行, 结束行, 起始列, 结束列
firstC = ca.getFirstColumn();
lastC = ca.getLastColumn();
firstR = ca.getFirstRow();
lastR = ca.getLastRow();
if (cell.getRowIndex() >= firstR && cell.getRowIndex() <= lastR) {
if (cell.getColumnIndex() >= firstC && cell.getColumnIndex() <= lastC) {
xr = lastR;
}
}
}
return xr;
}
/**
* 判断单元格是否为合并单元格，是的话则将单元格的值返回
*
* @param listCombineCell 存放合并单元格的list
* @param cell 需要判断的单元格
* @param sheet sheet
* @return
*/
public String isCombineCell(List<CellRangeAddress> listCombineCell, Cell cell, Sheet sheet)
throws Exception {
int firstC = 0;
int lastC = 0;
int firstR = 0;
int lastR = 0;
String cellValue = null;
for (CellRangeAddress ca : listCombineCell) {
//获得合并单元格的起始行, 结束行, 起始列, 结束列
firstC = ca.getFirstColumn();
lastC = ca.getLastColumn();
firstR = ca.getFirstRow();
lastR = ca.getLastRow();
if (cell.getRowIndex() >= firstR && cell.getRowIndex() <= lastR) {
if (cell.getColumnIndex() >= firstC && cell.getColumnIndex() <= lastC) {
Row fRow = sheet.getRow(firstR);
Cell fCell = fRow.getCell(firstC);
cellValue = getCellValue(fCell);
break;
}
} else {
cellValue = "";
}
}
return cellValue;
}
/**
* 获取合并单元格的值
*
* @param sheet
* @param row
* @param column
* @return
*/
public String getMergedRegionValue(Sheet sheet, int row, int column) {
int sheetMergeCount = sheet.getNumMergedRegions();
for (int i = 0; i < sheetMergeCount; i++) {
CellRangeAddress ca = sheet.getMergedRegion(i);
int firstColumn = ca.getFirstColumn();
int lastColumn = ca.getLastColumn();
int firstRow = ca.getFirstRow();
int lastRow = ca.getLastRow();
if (row >= firstRow && row <= lastRow) {
if (column >= firstColumn && column <= lastColumn) {
Row fRow = sheet.getRow(firstRow);
......@@ -228,19 +133,16 @@ public class ImportExcelServiceImpl implements IImportExcelService {
}
}
}
return null;
}
/**
* 判断指定的单元格是否是合并单元格
*
* @param sheet
* @param row 行下标
* @param column 列下标
* @return
*/
private boolean isMergedRegion(Sheet sheet, int row, int column) {
int sheetMergeCount = sheet.getNumMergedRegions();
for (int i = 0; i < sheetMergeCount; i++) {
......@@ -257,5 +159,4 @@ public class ImportExcelServiceImpl implements IImportExcelService {
}
return false;
}
}
\ No newline at end of file
nafmii-admin/src/main/resources/mapper/ImportExcelMapper.xml
......@@ -3,8 +3,7 @@
<mapper namespace="org.nafmii.admin.user.dao.ImportExcelMapper">
<!-- 通用查询映射结果 -->
<resultMap id="BaseResultMap" type="org.nafmii.admin.user.entity.ImportExcelPO">
<id column="ID" property="id" />
<resultMap id="BaseResultMap" type="org.nafmii.admin.user.dto.ImportExcelDto">
<result column="COLUMN1" property="column1" />
<result column="COLUMN2" property="column2" />
<result column="COLUMN3" property="column3" />
......@@ -13,9 +12,9 @@
<insert id ="insertExcelData" parameterType="org.nafmii.admin.user.dto.ImportExcelDto" >
insert into IMPORT_EXCEL_TEST
(COLUMN1,COLUMN2,COLUMN3)
values(#{ImportExcelPO.column1},
#{ImportExcelPO.column2},
#{ImportExcelPO.column3})
values(#{data.column1},
#{data.column2},
#{data.column3})
</insert >
</mapper>
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!