SSM整合开发实战-poi导入导出excel-开战

   上一篇博客介绍了如何完整的基于ssm搭建一个可用于实战开发的项目，本文开始将用此项目用来开发若干idea。本文就先小试牛刀开发基于ssm框架实现poi导入导出excel文件，凭良心讲，这一需求其实在目前企业级应用开发中几乎都用得到，那么本文我将基于自己在项目中遇到的需求来介绍如何实现poi导入导出。如有相关疑问可来此交流：java开源技术交流群-583522159-我是debug （视频教程地址：https://edu.csdn.net/course/detail/8894  欢迎支持！！）

    首先我们将基于这样的场景：某部门需要经常导出目前仓库具有产品列表；或者偶尔隔一段时间需要导入一些产品到系统数据库中用于其他业务使用！为了方便展示，首先我们建立一个表：产品表product，建表语句如下：

CREATE TABLE `product` (
  `id` int(11) NOT NULL AUTO_INCREMENT,
  `name` varchar(255) DEFAULT NULL COMMENT '名称',
  `unit` varchar(20) DEFAULT NULL COMMENT '单位',
  `price` double(15,2) DEFAULT NULL COMMENT '单价',
  `stock` double(11,0) DEFAULT NULL COMMENT '库存量',
  `remark` varchar(1000) DEFAULT NULL,
  `purchase_date` date DEFAULT NULL COMMENT '采购日期',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8 COMMENT='产品信息表';

    然后我们使用mybatis逆向工具生成实体类的相关文件，依次是实体类Product.java,ProductMapper.java以及ProductMapper.xml文件

package com.debug.steadyjack.model;

import java.util.Date;

public class Product {
    private Integer id;

    private String name;

    private String unit;

    private Double price;

    private Double stock;

    private String remark;

    private Date purchaseDate;
    
    public Integer getId() {
        return id;
    }

    public void setId(Integer id) {
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name == null ? null : name.trim();
    }

    public String getUnit() {
        return unit;
    }

    public void setUnit(String unit) {
        this.unit = unit == null ? null : unit.trim();
    }

    public Double getPrice() {
        return price;
    }

    public void setPrice(Double price) {
        this.price = price;
    }

    public String getRemark() {
        return remark;
    }

    public void setRemark(String remark) {
        this.remark = remark == null ? null : remark.trim();
    }

	public Date getPurchaseDate() {
		return purchaseDate;
	}

	public void setPurchaseDate(Date purchaseDate) {
		this.purchaseDate = purchaseDate;
	}

	public Double getStock() {
		return stock;
	}

	public void setStock(Double stock) {
		this.stock = stock;
	}

	@Override
	public String toString() {
		return "Product [id=" + id + ", name=" + name + ", unit=" + unit
				+ ", price=" + price + ", stock=" + stock + ", remark="
				+ remark + ", purchaseDate=" + purchaseDate + "]";
	}
	
}

package com.debug.steadyjack.mapper;

import java.util.List;

import org.apache.ibatis.annotations.Param;

import com.debug.steadyjack.model.Product;

public interface ProductMapper {
    int deleteByPrimaryKey(Integer id);

    int insert(Product record);

    int insertSelective(Product record);

    Product selectByPrimaryKey(Integer id);

    int updateByPrimaryKeySelective(Product record);

    int updateByPrimaryKey(Product record);
    
    List selectAll(@Param("name") String name);
    
    void insertBatch(@Param("dataList") List dataList);
}

  
    
    
    
    
    
    
    
  
  
    id, name, unit, price, stock, remark, purchase_date
  
  
    select 
    
    from product
    where id = #{id,jdbcType=INTEGER}
  
  
    delete from product
    where id = #{id,jdbcType=INTEGER}
  
  
    insert into product (id, name, unit, 
      price, stock, remark, 
      purchase_date)
    values (#{id,jdbcType=INTEGER}, #{name,jdbcType=VARCHAR}, #{unit,jdbcType=VARCHAR}, 
      #{price,jdbcType=DOUBLE}, #{stock,jdbcType=DOUBLE}, #{remark,jdbcType=VARCHAR}, 
      #{purchaseDate,jdbcType=DATE})
  
  
    insert into product
    
      
        id,
      
      
        name,
      
      
        unit,
      
      
        price,
      
      
        stock,
      
      
        remark,
      
      
        purchase_date,
      
    
    
      
        #{id,jdbcType=INTEGER},
      
      
        #{name,jdbcType=VARCHAR},
      
      
        #{unit,jdbcType=VARCHAR},
      
      
        #{price,jdbcType=DOUBLE},
      
      
        #{stock,jdbcType=DOUBLE},
      
      
        #{remark,jdbcType=VARCHAR},
      
      
        #{purchaseDate,jdbcType=DATE},
      
    
  
  
    update product
    
      
        name = #{name,jdbcType=VARCHAR},
      
      
        unit = #{unit,jdbcType=VARCHAR},
      
      
        price = #{price,jdbcType=DOUBLE},
      
      
        stock = #{stock,jdbcType=DOUBLE},
      
      
        remark = #{remark,jdbcType=VARCHAR},
      
      
        purchase_date = #{purchaseDate,jdbcType=DATE},
      
    
    where id = #{id,jdbcType=INTEGER}
  
  
    update product
    set name = #{name,jdbcType=VARCHAR},
      unit = #{unit,jdbcType=VARCHAR},
      price = #{price,jdbcType=DOUBLE},
      stock = #{stock,jdbcType=DOUBLE},
      remark = #{remark,jdbcType=VARCHAR},
      purchase_date = #{purchaseDate,jdbcType=DATE}
    where id = #{id,jdbcType=INTEGER}
  
  
  
  
  	SELECT  FROM product	WHERE 1=1 
  	
  		AND `name` LIKE CONCAT('%', #{name}, '%')
  	
  
  
  
  
	insert into product (name, unit,price, stock, remark,purchase_date)
	
	    values (#{data.name,jdbcType=VARCHAR}, #{data.unit,jdbcType=VARCHAR},#{data.price,jdbcType=DOUBLE}, 
	    #{data.stock,jdbcType=DOUBLE}, #{data.remark,jdbcType=VARCHAR}, #{data.purchaseDate,jdbcType=DATE})
	
  
  

    其中，我在上面以及实现了：查询产品列表（也可用于搜索）、批量插入产品数据  的sql-将分别用于导出以及导入

    然后，我们就得开发导出以及导入的controller功能了,PoiController.java：

package com.debug.steadyjack.controller;

import java.util.List;
import java.util.Map;

import javax.servlet.http.HttpServletResponse;

import org.apache.commons.lang.StringUtils;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Workbook;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.MediaType;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;

import com.debug.steadyjack.dto.BaseResponse;
import com.debug.steadyjack.enums.StatusCode;
import com.debug.steadyjack.mapper.ProductMapper;
import com.debug.steadyjack.model.Product;
import com.debug.steadyjack.service.PoiService;
import com.debug.steadyjack.utils.ExcelBeanUtil;
import com.debug.steadyjack.utils.ExcelUtil;
import com.debug.steadyjack.utils.WebUtil;

/**
 * 导入导出controller
 * @author zhonglinsen
 *
 */
@Controller
public class PoiController {
	
	private static final Logger log=LoggerFactory.getLogger(PoiController.class);

	private static final String prefix="poi";
	
	@Autowired
	private ProductMapper productMapper;
	
	@Autowired
	private PoiService poiService;
	
	@Value("${poi.excel.sheet.name}")
	private String sheetProductName;
	
	@Value("${poi.excel.file.name}")
	private String excelProductName;
	
	/**
	 * 获取产品列表-可用于搜索
	 * @param name
	 * @return
	 */
	@RequestMapping(value=prefix+"/list",method=RequestMethod.GET)
	@ResponseBody
	public BaseResponse> list(String name){
		BaseResponse> response=new BaseResponse>(StatusCode.Success);
		try {
			List products=productMapper.selectAll(name);
			response.setData(products);
		} catch (Exception e) {
			log.error("获取产品列表发生异常: ",e.fillInStackTrace());
		}
		
		return response;
	}
	
	/**
	 * 下载excel
	 * @param response
	 * @return
	 */
	@RequestMapping(value=prefix+"/excel/export",method=RequestMethod.GET)
	public @ResponseBody String exportExcel(HttpServletResponse response,String search){
		try {
			List products=productMapper.selectAll(search);
			String[] headers=new String[]{"id编号","名称","单位","单价","库存量","采购日期","备注信息"};
			List> dataList=ExcelBeanUtil.manageProductList(products);
			log.info("excel下载填充数据： {} ",dataList);
			
			Workbook wb=new HSSFWorkbook();
			ExcelUtil.fillExcelSheetData(dataList, wb, headers, sheetProductName);
			WebUtil.downloadExcel(response, wb, excelProductName);
			return excelProductName;
		} catch (Exception e) {
			log.error("下载excel 发生异常：",e.fillInStackTrace());
		}
		return null;
	}
	
	/**
	 * 上传excel导入数据
	 * @param request
	 * @return
	 * 1、不要忘了支持springmvc上传文件的配置
	 */
	@SuppressWarnings("rawtypes")
	@RequestMapping(value=prefix+"/excel/upload",method=RequestMethod.POST,consumes=MediaType.MULTIPART_FORM_DATA_VALUE)
	@ResponseBody
	public BaseResponse uploadExcel(MultipartHttpServletRequest request){
		BaseResponse response=new BaseResponse<>(StatusCode.Success);
		try {
			String version=request.getParameter("version");
			MultipartFile file=request.getFile("productFile");
			if (StringUtils.isEmpty(version) || file==null) {
				return new BaseResponse<>(StatusCode.Invalid_Param);
			}
			log.debug("版本号：{} file：{} ",version,file);
			
			HSSFWorkbook wb=new HSSFWorkbook(file.getInputStream());
			List products=poiService.readExcelData(wb);
			productMapper.insertBatch(products);
		} catch (Exception e) {
			log.error("上传excel导入数据 发生异常：",e.fillInStackTrace());
			return new BaseResponse<>(StatusCode.System_Error);
		}
		return response;
	}
}

    在上面涉及到了一些工具类，这些工具类在上篇博文已经贴出来了，自行去找即可！在这里重点再贴一个工具类ExcelBeanUtil.java：专门用于处理某些“业务实体”的excel工具

package com.debug.steadyjack.utils;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import com.debug.steadyjack.model.Product;

/**
 * 导入excel bean数据工具类
 * @author zhonglinsen
 *
 */
public class ExcelBeanUtil {

	/**
	 * 处理产品列表 塞入list-map 等待塞入excel的workbook进行处理
	 * @param products
	 * @return
	 */
	public static List> manageProductList(final List products){
		List> dataList=new ArrayList<>();
		if (products!=null && products.size()>0) {
			int length=products.size();
			
			Map dataMap;
			Product bean;
			for (int i = 0; i < length; i++) {
				bean=products.get(i);
				
				dataMap=new HashMap<>();
				dataMap.put(0, bean.getId());
				dataMap.put(1, bean.getName());
				dataMap.put(2, bean.getUnit());
				dataMap.put(3, bean.getPrice());
				dataMap.put(4, bean.getStock());
				dataMap.put(5, bean.getPurchaseDate());
				dataMap.put(6, bean.getRemark());
				dataList.add(dataMap);
			}
		}
		return dataList;
	}
}

 

    接下来介绍一下导入导出的思路：

 

1、导出：对于导出而已，首先我们得明确一点就是最终数据是导出到excel中，而观察excel每个sheet发现，是一个  “二维的世界”，即“矩阵”格式的数据： 头部行是你需要命名的那些字段名，之后的每一行的每一列都是那些字段对应的实际的值，如下图所示：

 

    所以，按照这个逻辑，首先我们需要定好头部行的字段名（当然啦，你也可以配置在配置文件），在这里我是直接建立一个字符串数组存储了，然后你还需要获取你需要导出到excel的productList。按照上面的截图，你会发现，其实每一行就是 一个Object，而多个Object构成了一个list，即List