java上传excel文件导入数据到数据库

快来打我* 2021-11-23 05:28 430阅读 0赞

# pom.xml #

<dependency>
                <groupId>org.apache.poi</groupId>
                <artifactId>poi</artifactId>
                <version>3.11</version>
            </dependency>
            <dependency>
                <groupId>org.apache.poi</groupId>
                <artifactId>poi-ooxml</artifactId>
                <version>3.11</version>
            </dependency>
            <dependency>
                <groupId>org.apache.poi</groupId>
                <artifactId>poi-ooxml-schemas</artifactId>
                <version>3.11</version>
            </dependency>
            <dependency>
                <groupId>org.apache.xmlbeans</groupId>
                <artifactId>xmlbeans</artifactId>
                <version>2.6.0</version>
            </dependency>
            <dependency>
                <groupId>org.apache.poi</groupId>
                <artifactId>poi-examples</artifactId>
                <version>3.11</version>
            </dependency>

# js-------使用了layui的上传组件 #

var loading;
            //上传导入校友excel表格
            upload.render({
                elem: '#uploadExcel'
                ,url: '/alumni/importAlumnis'
                ,accept: 'file' //普通文件
                ,exts: 'xls|xlsx' //允许上传的文件后缀
                ,field:'alumniData'     //文件域字段名
                ,before: function(obj){         //loading.....
                    loading=layer.msg("正在导入数据，请稍后......", {icon: 16}); //上传loading
                }
                ,done: function(result){//返回值接收
                    if(result.success){
                        layer.close(loading);
                        layer.msg(result.msg);
                      //上传成功后刷新数据列表
                        tableIns.reload();
                    }else{
                        layer.close(loading);
                        layer.msg(result.msg);
                    }
                },error:function (index,upload) {
                    layer.close(loading);
                    layer.msg("导入失败，请检查网络是否正常！");
                }
            });

# controller #

@PostMapping("/importAlumnis")
        @ResponseBody
        public Result importAlumnis(@RequestParam(value = "alumniData") MultipartFile file) throws IOException {
            InputStream inputStream=null;
            try {
                //输入流
                inputStream = file.getInputStream();
                //原始文件名
                String originalFilename = file.getOriginalFilename();
                //文件后缀
                String suffix = originalFilename.substring(originalFilename.lastIndexOf(".") + 1);
                ParseExcel parser = new ParseExcel();
                //第三行开始读取
                int startRow = 2;
                List<String[]> result = parser.parseExcel(inputStream, suffix, startRow);
                Alumni alumni=null;
                int count=0;
                for(String[] ss : result){
                    // System.out.println(Arrays.toString(ss));
                    alumni=new Alumni();
                    alumni.setName(ss[1]);
                    alumni.setCreatetime(new Date());
                    alumni.setSex(ss[2]);
                    alumni.setCompany(ss[3]);
                    alumni.setDuty(ss[4]);
                    alumni.setPhone(ss[5]);
                    //alumni.setDesc(ss[ss.length-1]);
                    //导入校友信息进入数据库
                    int i = alumniService.addAlumni(alumni);
                    if(i==1){
                        count++;
                    }
                }
                if(count==result.size()){
                    //全部数据导入成功
                    return Result.ok("全部数据导入成功");
                }
                return Result.ok("部分数据导入成功");
    
            } catch (IOException e) {
                e.printStackTrace();
                return Result.error("导入数据失败");
            } finally {
                //关闭流
                inputStream.close();
            }
        }

# 工具类 #

package com.iot.donation.util;
    
    import java.io.IOException;
    import java.io.InputStream;
    import java.text.DecimalFormat;
    import java.text.SimpleDateFormat;
    import java.util.ArrayList;
    import java.util.Date;
    import java.util.List;
    
    import org.apache.poi.hssf.usermodel.HSSFCell;
    import org.apache.poi.hssf.usermodel.HSSFDataFormat;
    import org.apache.poi.hssf.usermodel.HSSFDateUtil;
    import org.apache.poi.hssf.usermodel.HSSFWorkbook;
    import org.apache.poi.ss.usermodel.Cell;
    import org.apache.poi.ss.usermodel.Row;
    import org.apache.poi.ss.usermodel.Sheet;
    import org.apache.poi.ss.usermodel.Workbook;
    import org.apache.poi.xssf.usermodel.XSSFWorkbook;
    
    /**
     * 解析Excel文件
     * 
     * @author Administrator
     *
     */
    public class ParseExcel {
    	
    	/**
    	 * 解析文件的方法. 
    	 * @param inputStream 文件输入流, 要解析的Excel文件输入流
    	 * @param suffix 文件后缀名, xls或xlsx. 代码决定使用什么方式解析Excel.
    	 * @param startRow 从第几行开始读取数据.
    	 * @return List<String[]> 集合中的一个元素对应一行解析的数据. 
    	 * 		元素为字符串数组类型. 数组中的每个元素对应一列数据.
    	 * @throws IOException 
    	 */
    	public List<String[]> parseExcel(InputStream inputStream, String suffix, int startRow)
    			throws IOException{
    		
    		// 1. 定义excel对象变量
    		Workbook workbook = null;
    		
    		// 2. 判断后缀.决定使用的解析方式. 决定如何创建具体的对象
    		if("xls".equals(suffix)){
    			// 2003
    			workbook = new HSSFWorkbook(inputStream);
    		}else if("xlsx".equals(suffix)){
    			// 2007
    			workbook = new XSSFWorkbook(inputStream);
    		}else{
    			// 未知内容
    			return null;
    		}
    		
    		// 获取工作表  excel分为若干个表. sheet
    		Sheet sheet = workbook.getSheetAt(0);
    		
    		if(sheet == null){
    			return null;
    		}
    		
    		// 获取表格中最后一行的行号
    		int lastRowNum = sheet.getLastRowNum();
    		
    		// 最后一行的行号大于startRow
    		if(lastRowNum <= startRow){
    			return null;
    		}
    		
    		
    		List<String[]> result = new ArrayList<String[]>();
    		
    		// 定义行变量和单元格变量
    		Row row = null;
    		Cell cell = null;
    		// 循环读取
    		for(int rowNum = startRow; rowNum <= lastRowNum; rowNum++){
    			row = sheet.getRow(rowNum);
    			// 获取当前行的第一列和最后一列的标记
    			short firstCellNum = row.getFirstCellNum();
    			short lastCellNum = row.getLastCellNum();
    			if(lastCellNum != 0){
    				String[] rowArray = new String[lastCellNum];
    				for(int cellNum = firstCellNum; cellNum < lastCellNum; cellNum++){
    					cell = row.getCell(cellNum);
    					// 判断单元格是否有数据
    					if(cell == null){
    						rowArray[cellNum] = null;
    					}else{
    						rowArray[cellNum] = parseCell(cell);
    					}
    				}
    				result.add(rowArray);
    			}
    		}
    		
    		return result;
    	}
    
    	/**
    	 * 解析单元格
    	 * @return String 单元格数据
    	 */
    	private String parseCell(Cell cell){
    		String cellStr = null;
    		
    		// 判断单元格的类型
    		switch(cell.getCellType()){
    			case HSSFCell.CELL_TYPE_STRING :
    				// 字符串类型单元格
    				cellStr = cell.getRichStringCellValue().toString();
    				break;
    			case HSSFCell.CELL_TYPE_BLANK : 
    				// 空数据, 标准数据
    				cellStr = "";
    				break;
    			case HSSFCell.CELL_TYPE_NUMERIC :
    				// 数学类型. 数学类型包含日期,时间,数字
    				// 判断日期[年月日2016-11-17 | 时分10:00]类型
    				if(HSSFDateUtil.isCellDateFormatted(cell)){
    					// 判断具体类型, 是日期还是时间
    					SimpleDateFormat sdf = null;
    					if(cell.getCellType() == HSSFDataFormat.getBuiltinFormat("h:mm")){
    						// 时间
    						sdf = new SimpleDateFormat("HH:mm");
    					}else{
    						// 日期
    						sdf = new SimpleDateFormat("yyyy-MM-dd");
    					}
    					Date temp = cell.getDateCellValue();
    					cellStr = sdf.format(temp);
    				}else{
    					// 数字
    					double temp = cell.getNumericCellValue();
    					// 数学格式化工具
    					DecimalFormat format = new DecimalFormat();
    					// 查看单元格中的具体样式类型
    					String formatStr = cell.getCellStyle().getDataFormatString();
    					if(formatStr.equals("General")){
    						// 定义格式化正则. 使用具体有效数据进行个时候.且只保留有效数据.
    						format.applyPattern("#");
    					}
    					cellStr = format.format(temp);
    				}
    				break;
    			default : 
    				cellStr = "";
    		}
    		
    		return cellStr;
    	}
    }