java将excel文件的数据导入到数据库-爱代码爱编程
一、应用场景
excel文件中的数据导入到数据库当中
二、代码
- 依赖
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi</artifactId>
<version>4.1.2</version>
</dependency>
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi-ooxml</artifactId>
<version>4.1.2</version>
</dependency>
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi-ooxml-schemas</artifactId>
<version>4.1.2</version>
</dependency>
这里尽量保持版本一致,否则可能会发生冲突。我使用5.0.0版本的poi、poi-ooxml与4.1.2版本的poi-ooxml-schemas产生了冲突问题。
2.实现方法
import cn.hutool.core.io.IoUtil;
import cn.hutool.poi.excel.ExcelUtil;
import cn.hutool.poi.excel.ExcelWriter;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.shandy.project.api.basic.vo.ExcelDataVo;
import lombok.extern.slf4j.Slf4j;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.DateUtil;
import org.apache.poi.ss.usermodel.Font;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFCellStyle;
import org.apache.poi.xssf.usermodel.XSSFColor;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.apache.poi.xssf.usermodel.extensions.XSSFCellBorder;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.awt.Color;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.math.BigDecimal;
import java.net.URLEncoder;
import java.text.SimpleDateFormat;
import java.util.List;
import java.util.*;
@Slf4j
public class ExcelExportUtil {
// 2003版本
public static final String EXCEL_2003L = ".xls";
// 2007版本
public static final String EXCEL_2007U = ".xlsx";
// 表头字段
public static final String HEADER = "header";
/**
* @param in excel文件输入流 ,fileName 文件名称
* @Description:获取IO流中的数据,组装成List<List<Map<String, Object>>>对象
*/
public static List<List<Map<String, Object>>> getListByExcel(InputStream in, String fileName) throws Exception {
//创建Excel工作薄
Workbook work = getWorkbook(in, fileName);
if (null == work) {
throw new Exception("创建Excel工作薄为空!");
}
Sheet sheet = null; //页数
Row row = null; //行数
Cell cell = null; //列数
List<List<Map<String, Object>>> sheets = new ArrayList<>(); //记录每一页的信息
//遍历Excel中所有的sheet
for (int i = 0; i < work.getNumberOfSheets(); i++) {
sheet = work.getSheetAt(i);
if (sheet == null) {
continue;
}
List<Map<String, Object>> page = new ArrayList(); // 存储一页(sheet)的信息,一页有多行,一个Map就是一行
//遍历当前sheet中的所有行
List<String> headers = new ArrayList<>(); // 存放表头信息
int start = -1;
for (int j = sheet.getFirstRowNum(); j <= sheet.getLastRowNum(); j++) {
row = sheet.getRow(j);
if (row == null) {
continue;
}
if (start == -1) {
//第一次遇到非空行,记录为开始位置
start = j;
}
Map<String, Object> rowMap = new HashMap<>(); // 存放excel每一行信息 key为表头名,value为单元格值
//遍历所有的列(或者说遍历一行的单元格)
for (int y = row.getFirstCellNum(); y < row.getLastCellNum(); y++) {
cell = row.getCell(y);
if (cell == null) {
continue;
}
if (j == start) {
//记录表头信息
headers.add(getValue(cell));
} else {
//记录非表头信息,key - val 形式
rowMap.put(headers.get(y), getValue(cell));
}
}
if (rowMap.size() > 0) {
page.add(rowMap);
}
}
sheets.add(page);
}
return sheets;
}
/**
* @param is 输入流,fileName 文件名
* @Description:根据文件后缀,选择excel版本
*/
public static Workbook getWorkbook(InputStream is, String fileName) throws Exception {
Workbook wb = null;
String fileType = fileName.substring(fileName.lastIndexOf("."));
if (Objects.equals(fileType, EXCEL_2003L)) {
wb = new HSSFWorkbook(is); //2003版本的workbook对象
} else if (Objects.equals(fileType, EXCEL_2007U)) {
wb = new XSSFWorkbook(is); //2007+版本的workbook对象
} else {
throw new Exception("解析的文件格式有误!");
}
return wb;
}
}
3.有图
现有如下1个包含2个sheet页的excel文件
sheet1表格
sheet2表格
生成的数据格式:1个大集合,包含2个小集合,每个小集合里包含2个map。
[
[{"网站":"www.bd.com","邮箱":"b@139.com","电话":"13607467788"},{"网站":"www.ac.com", "邮箱":"a@139.COM", "电话":"15607468899"}],
[{"名称":"bp","分类":0, "logo文字":"连花清瘟", "logo地址":"中国"},{"名称":"gf","分类":1, "logo文字":"辣椒炒肉", "logo地址":"湖南"}]
]
一般不同的sheet所装的内容主题也不同,这样我们只需要对大数组里的每个小数组做不同的处理,在每个小数组里封装了map,map就是一行的记录,转化为java类就可以存入到数据库当中