java代码 EXCEL实战(一) 导入手机号

一、需求描述:

管理后台用户管理模块下某用户操作界面，需要一个批量导入手机号的功能。

二、前端代码:

2.1 html代码

<table>
  <tr>
     <td>
       <input type="radio" name="state" value="2">多用户
      </td>
      <td>
        <input type="file" placeholder="选择文件" name="file">
      </td>
    </tr>
    <tr>
      <td colspan="2">
        <button type="button" class="sub" onclick="addNewUser()">新增</button>
      </td>
     </tr>
</table>

2.2 js代码

 function addNewUser() {
        layer.confirm("确认新增用户？", {btn: ['确认', '取消']}, function () {
            var formData = new FormData($("#black")[0]);
            $.ajax({
                url: "/sys/batchAddMobiles",
                type: "POST",
                data: formData,
                contentType: false, //必须
                processData: false, //必须
                dataType: 'json',
                success: function (data) {
                    console.log(data);
                    var str = "";
                    if (data.mobiles.length > 0) {
                        str += data.mobiles + "格式不正确 ";
                    }
                    if (data.blackAlreay.length > 0) {
                        str += data.blackAlreay + "已在用户组中";

                    }
                    if ("" == str) {
                        window.location.reload();
                    } else {
                        layer.confirm(str, {btn: ["确认", "取消"]}, function () {
                            window.location.reload();
                        })
                    }
                }
            });
        });
    }

三、后台代码:

3.1 controller层

import org.apache.commons.io.FileUtils;

/**
     * 批量上传手机号
     * @param file
     * @param request
     * @return
     * @throws Exception
     */
    @RequestMapping(value = "/batchAddMobiles")
    @ResponseBody
    public String batchAddMobiles(@RequestParam(value = "file") MultipartFile file, HttpServletRequest request) throws Exception {
        Map<String, Object> result = new HashMap<String, Object>();
        List<String> mobiles = new ArrayList<String>();

        //excel文件上传
        String path = request.getSession().getServletContext().getRealPath("/");
        String fileName = System.currentTimeMillis() + file.getOriginalFilename();
        try {
            if (!file.isEmpty()) {
                // 这里将上传得到的文件保存指定目录下
                FileUtils.copyInputStreamToFile(file.getInputStream(), new File(path, fileName));
            }
            
            InputStream in = null;
            List<List<Object>> listob = null;
            in = file.getInputStream();
            listob = new ImportExcelUtil().getBankListByExcel(in, file.getOriginalFilename());
            for (int i = 0; i < listob.size(); i++) {
                List<Object> lo = listob.get(i);
                //TODO 根据excel格式获取指定的参数,还可优化:用正则表达式匹配过滤不符合的参数
                String mobile = String.valueOf(lo.get(0));
                mobiles.add(mobile);
            }
            
        } catch (Exception e) {
        	//TODO 记录完善的日志和异常处理
            logger.error("batchAddMobiles()失败,原因为:" + e.getMessage(), e);
        } finally {
            //处理完后需删除本地文件
            File file2 = new File(path + File.separator + fileName);
            if (file2.exists()) {
                file2.delete();
            }
        }

        //添加判断&业务处理
        if (null != mobiles && mobiles.size() > 0) {
            //TODO 业务逻辑处理
        }
        
        //返回结果
        result.put("mobiles", mobiles);
        return JSON.toJSONString(result);
    }

3.2 相关工具类

ImportExcelUtil.java

import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import org.apache.poi.hssf.usermodel.HSSFDataFormatter;
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 zengdq
 * 2019年2月27日 下午3:28:29
 */
public class ImportExcelUtil {
    private final static String excel2003L = ".xls"; // 2003- 版本的excel
    private final static String excel2007U = ".xlsx"; // 2007+ 版本的excel

    /**
     * 描述：获取IO流中的数据，组装成List<List<Object>>对象
     * @param in,fileName
     * @return
     * @throws Exception
     */
    public List<List<Object>> getBankListByExcel(InputStream in, String fileName) throws Exception {
        List<List<Object>> list = null;

        // 创建Excel工作薄
        Workbook work = this.getWorkbook(in, fileName);
        if (null == work) {
            throw new Exception("创建Excel工作薄为空！");
        }
        Sheet sheet = null;
        Row row = null;
        Cell cell = null;

        list = new ArrayList<List<Object>>();
        // 遍历Excel中所有的sheet
        for (int i = 0; i < work.getNumberOfSheets(); i++) {
            sheet = work.getSheetAt(i);
            if (sheet == null) {
                continue;
            }

            // 遍历当前sheet中的所有行
            for (int j = sheet.getFirstRowNum(); j < sheet.getLastRowNum() + 1; j++) { // 这里的加一是因为下面的循环跳过取第一行表头的数据内容了
                row = sheet.getRow(j);
                if (row == null || row.getFirstCellNum() == j) {
                    continue;
                }

                // 遍历所有的列
                List<Object> li = new ArrayList<Object>();
                for (int y = row.getFirstCellNum(); y < row.getLastCellNum(); y++) {
                    cell = row.getCell(y);
                    li.add(this.getCellValue(cell));
                }
                list.add(li);
            }
        }
        work.close();
        return list;
    }

    /**
     * 描述：根据文件后缀，自适应上传文件的版本
     * @param inStr,fileName
     * @return
     * @throws Exception
     */
    public Workbook getWorkbook(InputStream inStr, String fileName) throws Exception {
        Workbook wb = null;
        String fileType = fileName.substring(fileName.lastIndexOf("."));
        if (excel2003L.equals(fileType)) {
            wb = new HSSFWorkbook(inStr); // 2003-
        } else if (excel2007U.equals(fileType)) {
            wb = new XSSFWorkbook(inStr); // 2007+
        } else {
            throw new Exception("解析的文件格式有误！");
        }
        return wb;
    }

    /**
     * 描述：对表格中数值进行格式化
     * @param cell
     * @return
     */
    @SuppressWarnings("deprecation")
	public Object getCellValue(Cell cell) {
        Object value = null;
        switch (cell.getCellType()) {
            case Cell.CELL_TYPE_STRING:
                value = cell.getRichStringCellValue().getString();
                break;
            case Cell.CELL_TYPE_NUMERIC:
                HSSFDataFormatter dataFormatter = new HSSFDataFormatter();
                value= dataFormatter.formatCellValue(cell);
                break;
            case Cell.CELL_TYPE_BOOLEAN:
                value = cell.getBooleanCellValue();
                break;
            case Cell.CELL_TYPE_BLANK:
                value = "";
                break;
            default:
                break;
        }
        return value;
    }

}