当前位置:   article > 正文

【Springboot+mybatis】 解析Excel并批量导入到数据库_springboot excel表格导入多个对象表格

springboot excel表格导入多个对象表格

      更新时间2018-12-27 最近看了一个大神写的通过注解的形式来实现excel中表头名称 选项的设置,觉得很受启发,所以决定更新一版我的小文章。当然不会让大家失望的。这次更改我使用了注解工具类来侵入性更小的实现简单的excel导入导出。

demo源码地址:https://gitee.com/ytrlmy/importexcel/tree/importexcel

      再次更新一下,如何使用最简单的jar包更优雅的完成excel的导入和导出,包括多个sheet,多表头合并等等,我觉得再怎么复杂的excel都可以达到要求吧,先上图

  

下图来自csdn博客,我们来分析一下表头合并的问题。è¿éåå¾çæè¿°

第一种方案如下:

  1. /*
  2. * ExcelUtil工具类实现功能,借鉴了大神写法:
  3. * 导出时传入list<T>,即可实现导出为一个excel,其中每个对象T为Excel中的一条记录.
  4. * 导入时读取excel,得到的结果是一个list<T>.T是自己定义的对象.
  5. * 需要导出的实体对象只需简单配置注解就能实现灵活导出,通过注解您可以方便实现下面功能:
  6. * 1.实体属性配置了注解就能导出到excel中,每个属性都对应一列.
  7. * 2.列名称可以通过注解配置.
  8. * 3.导出到哪一列可以通过注解配置.
  9. * 4.鼠标移动到该列时提示信息可以通过注解配置.
  10. * 5.用注解设置只能下拉选择不能随意填写功能.
  11. * 6.用注解设置是否只导出标题而不导出内容,这在导出内容作为模板以供用户填写时比较实用.(导出一个空表,类似模板)
  12. */

详细代码如下:

  1. public class ExcelUtil<T> {
  2. private Class<T> clazz;
  3. public ExcelUtil(Class<T> clazz) {
  4. this.clazz = clazz;
  5. }
  6. /**
  7. * 导出excel文件
  8. *
  9. * @param sheetName 页名称
  10. * @param input 输入流程
  11. * @return 文件数据
  12. */
  13. public List<T> importExcel(String sheetName, InputStream input) {
  14. int maxCol = 0;
  15. List<T> list = new ArrayList<>();
  16. try {
  17. HSSFWorkbook workbook = new HSSFWorkbook(input);
  18. HSSFSheet sheet = workbook.getSheet(sheetName);
  19. if (!sheetName.trim().equals("")) {
  20. sheet = workbook.getSheet(sheetName);// 如果指定sheet名,则取指定sheet中的内容.
  21. }
  22. if (sheet == null) {
  23. sheet = workbook.getSheetAt(0); // 如果传入的sheet名不存在则默认指向第1个sheet.
  24. }
  25. int rows = sheet.getPhysicalNumberOfRows();
  26. if (rows > 0) {// 有数据时才处理
  27. // Field[] allFields = clazz.getDeclaredFields();// 得到类的所有field.
  28. List<Field> allFields = getMappedFiled(clazz, null);
  29. Map<Integer, Field> fieldsMap = new HashMap<>();// 定义一个map用于存放列的序号和field.
  30. for (Field field : allFields) { // 将有注解的field存放到map中.
  31. if (field.isAnnotationPresent(ExcelVOAttribute.class)) {
  32. ExcelVOAttribute attr = field.getAnnotation(ExcelVOAttribute.class);
  33. int col = getExcelCol(attr.column());// 获得列号
  34. maxCol = Math.max(col, maxCol);
  35. // System.out.println(col + "====" + field.getName());
  36. field.setAccessible(true);// 设置类的私有字段属性可访问.
  37. fieldsMap.put(col, field);
  38. }
  39. }
  40. for (int i = 1; i < rows; i++) {// 从第2行开始取数据,默认第一行是表头.
  41. HSSFRow row = sheet.getRow(i);
  42. // int cellNum = row.getPhysicalNumberOfCells();
  43. // int cellNum = row.getLastCellNum();
  44. T entity = null;
  45. for (int j = 0; j <= maxCol; j++) {
  46. HSSFCell cell = row.getCell(j);
  47. if (cell == null) {
  48. continue;
  49. }
  50. int cellType = cell.getCellType();
  51. String c;
  52. if (cellType == HSSFCell.CELL_TYPE_NUMERIC) {
  53. c = String.valueOf(cell.getNumericCellValue());
  54. } else if (cellType == HSSFCell.CELL_TYPE_BOOLEAN) {
  55. c = String.valueOf(cell.getBooleanCellValue());
  56. } else {
  57. c = cell.getStringCellValue();
  58. }
  59. if (c == null || c.equals("")) {
  60. continue;
  61. }
  62. entity = (entity == null ? clazz.newInstance() : entity);// 如果不存在实例则新建.
  63. // System.out.println(cells[j].getContents());
  64. Field field = fieldsMap.get(j);// 从map中得到对应列的field.
  65. if (field == null) {
  66. continue;
  67. }
  68. // 取得类型,并根据对象类型设置值.
  69. Class<?> fieldType = field.getType();
  70. if (String.class == fieldType) {
  71. field.set(entity, String.valueOf(c));
  72. } else if ((Integer.TYPE == fieldType) || (Integer.class == fieldType)) {
  73. field.set(entity, Integer.parseInt(c));
  74. } else if ((Long.TYPE == fieldType) || (Long.class == fieldType)) {
  75. field.set(entity, Long.valueOf(c));
  76. } else if ((Float.TYPE == fieldType) || (Float.class == fieldType)) {
  77. field.set(entity, Float.valueOf(c));
  78. } else if ((Short.TYPE == fieldType) || (Short.class == fieldType)) {
  79. field.set(entity, Short.valueOf(c));
  80. } else if ((Double.TYPE == fieldType) || (Double.class == fieldType)) {
  81. field.set(entity, Double.valueOf(c));
  82. } else if (Character.TYPE == fieldType) {
  83. if (c.length() > 0) {
  84. field.set(entity, c.charAt(0));
  85. }
  86. }
  87. }
  88. if (entity != null) {
  89. list.add(entity);
  90. }
  91. }
  92. }
  93. } catch (IOException | InstantiationException | IllegalAccessException | IllegalArgumentException e) {
  94. e.printStackTrace();
  95. }
  96. return list;
  97. }
  98. /**
  99. * 对list数据源将其里面的数据导入到excel表单
  100. *
  101. * @param lists 数据
  102. * @param sheetNames 工作表的名称
  103. * @param output java输出流
  104. * @return 状态
  105. */
  106. public boolean exportExcel(List<T> lists[], String sheetNames[], OutputStream output) {
  107. if (lists.length != sheetNames.length) {
  108. System.out.println("数组长度不一致");
  109. return false;
  110. }
  111. HSSFWorkbook workbook = new HSSFWorkbook();// 产生工作薄对象
  112. for (int ii = 0; ii < lists.length; ii++) {
  113. List<T> list = lists[ii];
  114. String sheetName = sheetNames[ii];
  115. List<Field> fields = getMappedFiled(clazz, null);
  116. HSSFSheet sheet = workbook.createSheet();// 产生工作表对象
  117. workbook.setSheetName(ii, sheetName);
  118. HSSFRow row;
  119. HSSFCell cell;// 产生单元格
  120. HSSFCellStyle style = workbook.createCellStyle();
  121. style.setFillForegroundColor(HSSFColor.SKY_BLUE.index);
  122. style.setFillBackgroundColor(HSSFColor.GREY_40_PERCENT.index);
  123. row = sheet.createRow(0);// 产生一行
  124. // 写入各个字段的列头名称
  125. for (Field field : fields) {
  126. ExcelVOAttribute attr = field.getAnnotation(ExcelVOAttribute.class);
  127. int col = getExcelCol(attr.column());// 获得列号
  128. cell = row.createCell(col);// 创建列
  129. cell.setCellType(HSSFCell.CELL_TYPE_STRING);// 设置列中写入内容为String类型
  130. cell.setCellValue(attr.name());// 写入列名
  131. // 如果设置了提示信息则鼠标放上去提示.
  132. if (!attr.prompt().trim().equals("")) {
  133. setHSSFPrompt(sheet, "", attr.prompt(), 1, lists[0].size(), col, col);// 这里默认设了2-101列提示.
  134. }
  135. // 如果设置了combo属性则本列只能选择不能输入
  136. if (attr.combo().length > 0) {
  137. setHSSFValidation(sheet, attr.combo(), 1, lists[0].size(), col, col);// 这里默认设了2-101列只能选择不能输入.
  138. }
  139. cell.setCellStyle(style);
  140. }
  141. int startNo = 0;
  142. int endNo = list.size(); // 写入各条记录,每条记录对应excel表中的一行
  143. for (int i = startNo; i < endNo; i++) {
  144. row = sheet.createRow(i + 1 - startNo);
  145. T vo = list.get(i); // 得到导出对象.
  146. for (Field field : fields) {
  147. field.setAccessible(true);// 设置实体类私有属性可访问
  148. ExcelVOAttribute attr = field.getAnnotation(ExcelVOAttribute.class);
  149. try {
  150. // 根据ExcelVOAttribute中设置情况决定是否导出,有些情况需要保持为空,希望用户填写这一列.
  151. if (attr.isExport()) {
  152. cell = row.createCell(getExcelCol(attr.column()));
  153. // 创建cell
  154. cell.setCellType(HSSFCell.CELL_TYPE_STRING);
  155. cell.setCellValue(field.get(vo) == null ? ""
  156. : String.valueOf(field.get(vo)));// 如果数据存在就填入,不存在填入空格.
  157. }
  158. } catch (IllegalArgumentException | IllegalAccessException e) {
  159. e.printStackTrace();
  160. }
  161. }
  162. }
  163. }
  164. try {
  165. output.flush();
  166. workbook.write(output);
  167. output.close();
  168. return true;
  169. } catch (IOException e) {
  170. e.printStackTrace();
  171. System.out.println("Output is closed ");
  172. return false;
  173. }
  174. }
  175. /**
  176. * 对list数据源将其里面的数据导入到excel表单
  177. *
  178. * @param list 数据
  179. * @param sheetName 工作表的名称
  180. * @param output java输出流
  181. * @return 状态
  182. */
  183. @SuppressWarnings("unchecked")
  184. public boolean exportExcel(List<T> list, String sheetName, OutputStream output) {
  185. //此处 对类型进行转换
  186. List<T> ilist = new ArrayList<>();
  187. ilist.addAll(list);
  188. List<T>[] lists = new ArrayList[1];
  189. lists[0] = ilist;
  190. String[] sheetNames = new String[1];
  191. sheetNames[0] = sheetName;
  192. return exportExcel(lists, sheetNames, output);
  193. }
  194. /**
  195. * 将EXCEL中A, B, C, D, E列映射成0, 1, 2, 3
  196. *
  197. * @param col 行
  198. */
  199. public static int getExcelCol(String col) {
  200. col = col.toUpperCase();
  201. // 从-1开始计算,字母重1开始运算。这种总数下来算数正好相同。
  202. int count = -1;
  203. char[] cs = col.toCharArray();
  204. count += IntStream.range(0, cs.length).map(i -> (int) ((cs[i] - 64) * Math.pow(26, cs.length - 1 - i))).sum();
  205. return count;
  206. }
  207. /**
  208. * 设置单元格上提示
  209. *
  210. * @param sheet 要设置的sheet.
  211. * @param promptTitle 标题
  212. * @param promptContent 内容
  213. * @param firstRow 开始行
  214. * @param endRow 结束行
  215. * @param firstCol 开始列
  216. * @param endCol 结束列
  217. * @return 设置好的sheet.
  218. */
  219. public static HSSFSheet setHSSFPrompt(HSSFSheet sheet, String promptTitle, String promptContent, int firstRow,
  220. int endRow, int firstCol, int endCol) {
  221. // 构造constraint对象
  222. DVConstraint constraint = DVConstraint.createCustomFormulaConstraint("DD1");
  223. // 四个参数分别是:起始行、终止行、起始列、终止列
  224. CellRangeAddressList regions = new CellRangeAddressList(firstRow, endRow, firstCol, endCol);
  225. // 数据有效性对象
  226. HSSFDataValidation data_validation_view = new HSSFDataValidation(regions, constraint);
  227. data_validation_view.createPromptBox(promptTitle, promptContent);
  228. sheet.addValidationData(data_validation_view);
  229. return sheet;
  230. }
  231. /**
  232. * 设置某些列的值只能输入预制的数据, 显示下拉框.
  233. *
  234. * @param sheet 要设置的sheet.
  235. * @param textlist 下拉框显示的内容
  236. * @param firstRow 开始行
  237. * @param endRow 结束行
  238. * @param firstCol 开始列
  239. * @param endCol 结束列
  240. * @return 设置好的sheet.
  241. */
  242. public static HSSFSheet setHSSFValidation(HSSFSheet sheet, String[] textlist, int firstRow, int endRow,
  243. int firstCol, int endCol) {
  244. // 加载下拉列表内容
  245. DVConstraint constraint = DVConstraint.createExplicitListConstraint(textlist);
  246. // 设置数据有效性加载在哪个单元格上,四个参数分别是:起始行、终止行、起始列、终止列
  247. CellRangeAddressList regions = new CellRangeAddressList(firstRow, endRow, firstCol, endCol);
  248. // 数据有效性对象
  249. HSSFDataValidation data_validation_list = new HSSFDataValidation(regions, constraint);
  250. sheet.addValidationData(data_validation_list);
  251. return sheet;
  252. }
  253. /**
  254. * 得到实体类所有通过注解映射了数据表的字段
  255. *
  256. * @param clazz 类型
  257. * @param fields 文件集合
  258. * @return 文件集合
  259. */
  260. @SuppressWarnings("rawtypes")
  261. private List<Field> getMappedFiled(Class clazz, List<Field> fields) {
  262. if (fields == null) {
  263. fields = new ArrayList<>();
  264. }
  265. Field[] allFields = clazz.getDeclaredFields();// 得到所有定义字段
  266. // 得到所有field并存放到一个list中.
  267. for (Field field : allFields) {
  268. if (field.isAnnotationPresent(ExcelVOAttribute.class)) {
  269. fields.add(field);
  270. }
  271. }
  272. if (clazz.getSuperclass() != null && !clazz.getSuperclass().equals(Object.class)) {
  273. getMappedFiled(clazz.getSuperclass(), fields);
  274. }
  275. return fields;
  276. }
  277. }

   工具类有了,我们需要一个自定义一个注解,方便在excel列对象上直接使用该注解对应出需要导出导入的内容,如下:

  1. @Retention(RetentionPolicy.RUNTIME)
  2. @Target({java.lang.annotation.ElementType.FIELD})
  3. public @interface ExcelVOAttribute {
  4. //导出到Excel中的名字.
  5. public abstract String name();
  6. //配置列的名称,对应A,B,C,D....
  7. public abstract String column();
  8. //提示信息.
  9. public abstract String prompt() default "";
  10. //设置只能选择不能输入的列内容
  11. public abstract String[] combo() default {};
  12. //是否导出数据,应对需求:有时我们需要导出一份模板,这是标题需要但内容需要用户手工填写.
  13. public abstract boolean isExport() default true;
  14. }

  接下来你就可以使用一个简单的例子来实现,我想导出一个excel,方便用户提现申请审核,然后再把审核结果批量导入到数据库,同时更新数据库内容,首先需要定义一个excel列对应的实体对象取名(可随意)为:BackstageWithdrawDepositInfo,可使用lombok插件,少写getter和setter,这里就不详细介绍了,如下:

  1. @Data
  2. @Builder
  3. @NoArgsConstructor
  4. @AllArgsConstructor
  5. public class BackstageWithdrawDepositInfo {
  6. @ExcelVOAttribute(name = "钱包详情id", column = "A")
  7. private Long walletDetailsId;
  8. @ExcelVOAttribute(name = "用户id", column = "B")
  9. private Long userId;
  10. @ExcelVOAttribute(name = "申请时间", column = "C")
  11. private String createDate;
  12. @ExcelVOAttribute(name = "用户昵称", column = "D")
  13. private String nickName;
  14. @ExcelVOAttribute(name = "用户头像", column = "E")
  15. private String headImageUrl;
  16. @ExcelVOAttribute(name = "用户电话", column = "F")
  17. private String phoneNumber;
  18. @ExcelVOAttribute(name = "用户截止当日总提现次数", column = "G")
  19. private Long withdrawCount;
  20. @ExcelVOAttribute(name = "用户成功提现总金额(分)", column = "H")
  21. private Long withdrawSum;
  22. @ExcelVOAttribute(name = "本次提现金额(分)", column = "I")
  23. private Long count;
  24. @ExcelVOAttribute(name = "批复状态", column = "J", combo = {"提现审核", "提现成功", "提现失败"})
  25. private String status;
  26. @ExcelVOAttribute(name = "拒绝原因", column = "K", combo = {"抱歉,活动已结束,更多精彩活动正在筹备中,敬请期待……",
  27. "抱歉,今日平台提现金额已达上线,明天早点来哦", "抱歉,平台显示账户异常,如有疑问,请联系小轻老师咨询(微信号:qingtiku)",
  28. "其他原因,详情可联系小轻老师咨询(微信号:qingtiku)"})
  29. private String note;
  30. }

   这样呢一个简单的对象就构建完成了,但是呢,如何使用咱们工具类中对应的导入和导出方法呢,回想一下里面方法参数可能就有思路了,回顾一下:

  1. /**
  2. * 对list数据源将其里面的数据导入到excel表单
  3. *
  4. * @param list 数据
  5. * @param sheetName 工作表的名称
  6. * @param output java输出流
  7. * @return 状态
  8. */
  9. @SuppressWarnings("unchecked")
  10. public boolean exportExcel(List<T> list, String sheetName, OutputStream output) {
  11. //此处 对类型进行转换
  12. List<T> ilist = new ArrayList<>();
  13. ilist.addAll(list);
  14. List<T>[] lists = new ArrayList[1];
  15. lists[0] = ilist;
  16. String[] sheetNames = new String[1];
  17. sheetNames[0] = sheetName;
  18. return exportExcel(lists, sheetNames, output);
  19. }

从上面方法中看出导出操作需要的参数是,一个list集合(对应的我们构建的实体类BackstageWithdrawDepositInfo,其次呢还要指定一个excel的Sheet名称和一个具体路径的输出流对象),具体的输出流就很简单了,就是把该excel指定一个输出的路径,FileOutputStream out = new FileOutputStream(filePath); 然后把out参数传入进去即可,最后调用exportExcel()方法即可,看着是不是很简单。但是下面这个需求就是在代码中一行行写每个excel的列,可做一点比较,我会把两个代码都放在码云上。

第二种方案(推荐)如下:

推荐使用Alibaba的开源二方包,简单的两行的两行代码,在配合上我第一种方案的Excel工具类,简直so easy,具体地址是:

https://github.com/alibaba/easyexcel

其maven依赖是:

  1. <dependency>
  2. <groupId>com.alibaba</groupId>
  3. <artifactId>easyexcel</artifactId>
  4. <version>1.0.1</version>
  5. </dependency>

我把我项目中用到的主要核心代码贴出来,让大家看一哈,

第一步,在值对象上来确定表头,这个需要继承BaseRowModel类,如下:

  1. @Data
  2. @Builder
  3. @AllArgsConstructor
  4. @NoArgsConstructor
  5. public class LegionUserDataDTO extends BaseRowModel {
  6. /**
  7. * 渠道来源
  8. */
  9. @ExcelProperty(value = "渠道来源", index = 0)
  10. private String branch;
  11. /**
  12. * 进入小程序次数
  13. */
  14. @ExcelProperty(value = "用户昵称", index = 1)
  15. private String nickName;
  16. /**
  17. * 授权手机号数量
  18. */
  19. @ExcelProperty(value = "授权手机号", index = 2)
  20. private String phoneNumber;
  21. }

第二步,需要构建一个文件的输出流,需要把数据写入到文件里,具体构建步骤方法有很多种,我这边使用了一个FileUtil工具类完成的,网上有很多,具体如下:

  1. // 先创建一个文件夹
  2. FileUtil.createTargetDir(BASE_PATH);
  3. // 指定该文件的路径
  4. String filePath = BASE_PATH + "芒果会计-军团数据统计-" + DateUtil.parseDateToStr(new Date(), "yyyyMMdd") + ".xls";
  5. // 通过文件路径构造出输出流
  6. FileOutputStream out = new FileOutputStream(filePath);
  7. // 接下来就是最主要的步骤,直接调用提供的Excelriter想excel写入数据
  8. ExcelWriter writer = new ExcelWriter(out, ExcelTypeEnum.XLS, true);
  9. // 指定写入的sheet 有多个sheet就new 出多个,也可使用工厂方法,传入的参数1 表示为第一个sheet
  10. 0 表示从第一个单元格开始写入,传入继承了BaseRowModel的类和sheet名称即可。
  11. Sheet sheetStatistics = new Sheet(1, 0, LegionDataDTO.class, "渠道数据", null);
  12. // 开始写入 传入需要写入的数据集合和sheet对象
  13. writer.write(legionDataDTOS, sheetStatistics);
  14. Sheet sheetLegionUsers = new Sheet(2, 0, LegionUserDataDTO.class, "用户手机号", null);
  15. writer.write(legionUserDataDTOS, sheetLegionUsers);
  16. // 写入完成
  17. writer.finish();
  18. try {
  19. out.close();
  20. } catch (IOException e) {
  21. log.error("芒果会计-军团数据统计, error:", e.getMessage());
  22. }

然后你就能看到excel里有你需要的数据了。更复杂的表头需要在值对象继承BaseRowModel时候在该对象的字段上进行设置。

第三种方案如下:

一个很简单的需求,从后台中选取一个的Excel文件,并直接导入到数据库,然后根据导入的表查询个人信息,由于表的数据内容还挺多,有好几万条,所幸是单表。

     一个很愚蠢的做法是每解析一个Row,然后构建一个新的对象,再把每个cell的值赋给该对象,直到循环结束,再通过mabatis的insert语句入库,这样很low,有没有???但是数据也能跑的过去,能完成入库,不过两万条数据,连解析带入库一共花了将近20分钟,天啦噜,这样客户会炸毛的。没办法,还得修改,看了很多办法,一种最直接的就是把解析和入库分开,然后重新构建sql批量插入,因为mybatis用的是generator插件自动生成的代码,想把自己构建的mapper接口内容放在mybatis自己的目录下,但是每次重新执行插件时候,自己新写的接口文件就会被清除掉,无奈之举,只能重新构建一个文件夹(excelmapper)来存储我的查询接口,这才逃过一劫,当我进行测试时候又报错了,无法扫描到excelmapper下面的接口内容,尝试在单元测试类上面配置@componentScan("com.net.excelmapper")注解,如果启动类在根包下面,则你可以在该类上添加@ComponentScan注解而不需要添加任何参数,Spring Boot会在根包下面搜索注有@Component@Service@Repository@Controller注解的所有类,并将他们注册为Spring Beans,否则,你需要在@ComponentScan注解上定义basePackages或者其他属性。最后的解决办法是在启动类上添加了对应excelmapper文件夹下的扫描路径,@MapperScan({"com.neo.mapper","com.neo.excelmapper"}),合法化测试终于通过了,可是还有一个问题,上传超过了限制的默认值,只好在配置文件添加了配置项:

spring.http.multipart.max-file-size = 2048KB //支持文件上传最大的限制

spring.http.multipart.max-request-size = 10240kB //最大支持请求

spring.http.multipart.enabled = true#默认支持文件上传。

spring.http.multipart.file-size-threshold = 0#支持文件写入磁盘。
spring.http.multipart.location =#上传文件的临时目录

因为用的是表单进行提交,<input type =“file”>这种情况下在form标签下需要特殊的指定属性文件为formdata文件,例如:<form role =“form”action =“/ importExcel”method =“post” enctype =“multipart / form-data”>,这下终于可以完成数据导入了,不料,两万条数据果然还是花了20分钟,修改意见也就上面提到的,分开操作,先把解析的数据封装到一个对象上,并且存储到列表集合中,在从几何中批量插入。

  1. excelmapper接口:
  2. 公共接口ExcelMapper {
  3. void batchInsert(List <Tbagent> tbagentList);
  4. }
公共接口ExcelMapper { void batchInsert(List <Tbagent> tbagentList); } excelmapper.xml中主要的SQL语句,也就是执行批量操作的SQL
  1. <insert id="batchInsert" parameterType="java.util.List">
  2. INSERT INTO agent(job_number,department,region,username,idcard,company_rankings,department_rankings,region_rankings,
  3. distance_first_company,distance_first_department,distance_first_region)
  4. VALUES
  5. <foreach collection="list" item="item" index="index" separator=",">
  6. (#{item.job_number},#{item.department},#{item.region},#{item.username},#{item.idcard},#{item.company_rankings},
  7. #{item.department_rankings},#{item.region_rankings},#{item.distance_first_company},#{item.distance_first_department},#{item.distance_first_region})
  8. </foreach>
  9. </insert>
  1. 主要的服务层:
  1. /**
  2. * Created by Administrator on 2018/1/11.
  3. */
  4. @Service
  5. public class IimportServiceImpl implements IimportService {
  6. private final static String XLS = "xls";
  7. public static final String XLSX = "xlsx";
  8. private final static Logger logger = LoggerFactory.getLogger(IimportServiceImpl.class);
  9. @Autowired private TbagentMapper tbagentMapper;
  10. @Autowired private SqlSessionTemplate sqlSessionTemplate;
  11. @Autowired private ExcelMapper excelMapper;
  12. @Override
  13. public Integer importExcel(MultipartFile myFile) {
  14. //1. 使用HSSFWorkbook 打开或者创建 “Excel对象”
  15. //2. 用HSSFWorkbook返回对象或者创建sheet对象
  16. //3. 用sheet返回行对象,用行对象得到Cell对象
  17. //4. 对Cell对象进行读写
  18. List<Tbagent> tbagents = new ArrayList<>();
  19. Workbook workbook = null;
  20. String fileName = myFile.getOriginalFilename();// 获取文件名
  21. logger.info("【fileName】{}",fileName);
  22. if (fileName.endsWith(XLS))
  23. {
  24. try {
  25. workbook = new HSSFWorkbook(myFile.getInputStream());// 2003版本
  26. } catch (IOException e) {
  27. e.printStackTrace();
  28. }
  29. }else if(fileName.endsWith(XLSX)){
  30. try {
  31. workbook = new XSSFWorkbook(myFile.getInputStream());// 2007版本
  32. } catch (IOException e) {
  33. e.printStackTrace();
  34. }
  35. }else{
  36. throw new LianjiaException(ResultEnum.FILE_IS_NOT_EXCEL); // 文件不是Excel文件
  37. }
  38. Sheet sheet = workbook.getSheet("sheet1");
  39. int rows = sheet.getLastRowNum();
  40. logger.info("【rows】{}",rows);
  41. if(rows == 0){
  42. throw new LianjiaException(ResultEnum.DATA_IS_NULL);// 数据为空 请填写数据
  43. }
  44. long startTime = System.currentTimeMillis();
  45. for(int i = 1;i<= rows+1;i++){
  46. Row row = sheet.getRow(i);
  47. if(row !=null){
  48. Tbagent tbagent = new Tbagent();
  49. // 部门
  50. String department = getCellValue(row.getCell(0));
  51. tbagent.setDepartment(department);
  52. // 用户姓名
  53. String username = getCellValue(row.getCell(1));
  54. tbagent.setUsername(username);
  55. // 工号
  56. String jobNumer = getCellValue(row.getCell(2));
  57. if(!StringUtils.isEmpty(jobNumer)){
  58. Integer job_number = Integer.parseInt(jobNumer);
  59. tbagent.setJob_number(Integer.valueOf(jobNumer));
  60. }
  61. // 身份证后六位
  62. String idcard = getCellValue(row.getCell(3));
  63. tbagent.setIdcard(idcard);
  64. // 公司排名
  65. String companyRankings = getCellValue(row.getCell(4));
  66. if(!StringUtils.isEmpty(companyRankings)){
  67. Integer new_companyRankings = Integer.parseInt(companyRankings);
  68. tbagent.setCompany_rankings(new_companyRankings);
  69. }
  70. // 事业部排名
  71. String departmenRanks = getCellValue(row.getCell(5));
  72. if(!StringUtils.isEmpty(departmenRanks)){
  73. Integer new_departmentRanks = Integer.parseInt(departmenRanks);
  74. tbagent.setDepartment_rankings(new_departmentRanks);
  75. }
  76. // 大区排名
  77. String region_Ranks = getCellValue(row.getCell(6));
  78. if(!StringUtils.isEmpty(region_Ranks)){
  79. Integer new_region_Rankings = Integer.parseInt(region_Ranks);
  80. //logger.info("【大区排名】{}",new_region_Rankings);
  81. tbagent.setRegion_rankings(new_region_Rankings);
  82. }// 距离公司第一名差距
  83. String distance_first_company = getCellValue(row.getCell(7));
  84. if(!StringUtils.isEmpty(distance_first_company)){
  85. Long new_distance_first_company = Long.parseLong(distance_first_company);
  86. tbagent.setDistance_first_company(new_distance_first_company);
  87. }
  88. // 距离事业部第一名差距
  89. String distance_first_deparment = getCellValue(row.getCell(8));
  90. if(!StringUtils.isEmpty(distance_first_deparment)){
  91. Long new_distance_first_deparment = Long.parseLong(distance_first_deparment);
  92. tbagent.setDistance_first_department(new_distance_first_deparment);
  93. }
  94. // 距离大区第一名差距
  95. String distance_first_region = getCellValue(row.getCell(9));
  96. if(!StringUtils.isEmpty(distance_first_region)){
  97. Long new_distance_first_region = Long.parseLong(distance_first_region);
  98. tbagent.setDistance_first_region(new_distance_first_region);
  99. }
  100. //System.out.println(JSON.toJSON(tbagent));
  101. //tbagentMapper.insert(tbagent);
  102. tbagents.add(tbagent);
  103. //logger.info("插入数据完成");
  104. }
  105. }
  106. excelMapper.batchInsert(tbagents); // 批量插入 五秒完成
  107. long endTime = System.currentTimeMillis();
  108. long totaltime = endTime - startTime;
  109. logger.info("【消耗时间为】{}",totaltime); // 将近两万条数据 3秒解析完成
  110. logger.info("【第一条数据为】{}",JSON.toJSON(tbagents.get(0)));
  111. return rows;
  112. }
  113. public String getCellValue(Cell cell) {
  114. String value = "";
  115. if (cell != null) {
  116. switch(cell.getCellType()){
  117. case HSSFCell.CELL_TYPE_NUMERIC:// 数字
  118. value = cell.getNumericCellValue()+ " ";
  119. if(HSSFDateUtil.isCellDateFormatted(cell)){
  120. Date date = cell.getDateCellValue();
  121. if(date != null){
  122. value = new SimpleDateFormat("yyyy-MM-dd").format(date); // 日期格式化
  123. }else{
  124. value = "";
  125. }
  126. }else {
  127. // 解析cell时候 数字类型默认是double类型的 但是想要获取整数类型 需要格式化
  128. value = new DecimalFormat("0").format(cell.getNumericCellValue());
  129. }
  130. break;
  131. case HSSFCell.CELL_TYPE_STRING: // 字符串
  132. value = cell.getStringCellValue();
  133. break;
  134. case HSSFCell.CELL_TYPE_BOOLEAN: // Boolean类型
  135. value = cell.getBooleanCellValue()+"";
  136. break;
  137. case HSSFCell.CELL_TYPE_BLANK: // 空值
  138. value = "";
  139. break;
  140. case HSSFCell.CELL_TYPE_ERROR: // 错误类型
  141. value ="非法字符";
  142. break;
  143. default:
  144. value = "未知类型";
  145. break;
  146. }
  147. }
  148. return value.trim();
  149. }
  150. }

部分效果图:

 

登录效果图如上。

不太完美的地方就是没有一个进度条给客户看,后期需要做处理。清空数据用的是MySQL的的截断语句,直接清空所有的数据。然后让用户在导入到数据库。

源码地址 importexcel 分支

 

 

本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号