用POI实现 报表读取

package com.bjnlmf.nerc.operation.web.controller.picture;

import com.bjnlmf.nerc.common.api.enumeration.ImageType;
import com.bjnlmf.nerc.common.api.util.ResponseJson;
import com.bjnlmf.nerc.configration.api.enumeration.ServiceType;
import com.bjnlmf.nerc.configration.api.pojo.Area;
import com.bjnlmf.nerc.configration.api.pojo.AreaQuery;
import com.bjnlmf.nerc.configration.api.pojo.News;
import com.bjnlmf.nerc.configration.api.service.AreaService;
import com.bjnlmf.nerc.configration.api.service.NewsService;
import com.bjnlmf.nerc.configration.api.service.UploadService;
import com.bjnlmf.nerc.operation.web.shiro.ShiroUtils;
import org.apache.commons.jexl2.UnifiedJEXL;
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;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
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 java.io.*;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;
import java.util.Map;

@Controller
@RequestMapping("/upload")
public class UploadController {
    @Autowired
    UploadService uploadService;
    @Autowired
    NewsService newsService;

    @Autowired
    AreaService areaService;
    private Logger logger = LoggerFactory.getLogger(UploadController.class);

    /**
     * 上传图片和缩略图
     */
    @RequestMapping(value = "image", method = RequestMethod.POST)
    @ResponseBody
    public ResponseJson<Map<String,String>> imageUpload(ImageType imageType, MultipartFile image) throws IOException{
        String prefix=getPrefix(image);//获得文件后缀
        return ResponseJson.ok(uploadService.upload(image.getBytes(),prefix,imageType));
    }

    /**
     * 上传文件
     */
    @RequestMapping(value = "file", method = RequestMethod.POST)
    @ResponseBody
    public ResponseJson<String> fileUpload(MultipartFile  file) throws IOException {
        String prefix=getPrefix(file);//获得文件后缀
        return ResponseJson.ok(uploadService.upload(file.getBytes(),prefix));
    }

    //获得文件后缀
    private String getPrefix(MultipartFile file){
        String fileName = file.getOriginalFilename();
        return fileName.substring(fileName.lastIndexOf(".")+1);
    }

    /**
     * 读取excel文件
     */
    @RequestMapping(value = "excel", method = RequestMethod.POST)
    @ResponseBody
    public ResponseJson<String> excelUpload(MultipartFile  file) throws IOException {
            // 检查
            // this.preReadCheck(filePath);
            // 获取workbook对象
            Workbook workbook = null;
            try {
                ByteArrayInputStream  is = new ByteArrayInputStream(file.getBytes());
                String prefix=getPrefix(file);
                if (prefix.equals("xls")) {
                workbook = new HSSFWorkbook(is);
                } else if (prefix.equals("xlsx")) {
                    workbook = new XSSFWorkbook(is);
                }
            // 读文件 一个sheet一个sheet地读取
            for (int numSheet = 0; numSheet < 1; numSheet++) {
                Sheet sheet = workbook.getSheetAt(numSheet);
                if (sheet == null) {
                    continue;
                }

                int firstRowIndex = sheet.getFirstRowNum();
                int lastRowIndex = sheet.getLastRowNum();

                // 读取首行 即,表头
                Row firstRow = sheet.getRow(firstRowIndex);

                if(!this.getCellValue(firstRow.getCell(0), true).equals("标题")){
                    return  ResponseJson.ok("excel文件字段不对应");
                }
                if(!this.getCellValue(firstRow.getCell(1), true).equals("来源")){
                    return  ResponseJson.ok("excel文件字段不对应");
                }
                if(!this.getCellValue(firstRow.getCell(2), true).equals("政策发布时间")){
                    return  ResponseJson.ok("excel文件字段不对应");
                }
                if(!this.getCellValue(firstRow.getCell(3), true).equals("内容")){
                    return  ResponseJson.ok("excel文件字段不对应");
                }
                if(!this.getCellValue(firstRow.getCell(4), true).equals("所属省份")){
                    return  ResponseJson.ok("excel文件字段不对应");
                }
                if(!this.getCellValue(firstRow.getCell(5), true).equals("市名")){
                    return  ResponseJson.ok("excel文件字段不对应");
                }


                // 读取数据行
                for (int rowIndex = firstRowIndex + 1; rowIndex <= lastRowIndex; rowIndex++) {
                    Row currentRow = sheet.getRow(rowIndex);// 当前行
                    News news = new News();
                    String title = this.getCellValue(currentRow.getCell(0), true);
                    if (title.equals("") || title == null) {
                        break;
                    }
                    news.setTitle(title);
                    String resource = this.getCellValue(currentRow.getCell(1), true);
                    if(resource!="" && resource !=null){
                        news.setSource(resource);
                    }
                    String time = this.getCellValue(currentRow.getCell(2), true);
                    if(time!=""&&time!=null){
                        SimpleDateFormat sf = new SimpleDateFormat("yyyy-MM-dd");
                        Date date = sf.parse(time);
                        news.setReleaseTime(date);
                    }else{
                        news.setReleaseTime(new Date());
                    }
                    news.setContent(this.getCellValue(currentRow.getCell(3), true));
                    //news.setServiceType(this.getCellValue(currentRow.getCell(4), true));
                    String province = this.getCellValue(currentRow.getCell(4), true);
                    String city = this.getCellValue(currentRow.getCell(5), true);
                    List<Area> province1s= areaService.queryListProvince();
                    for (Area province1:province1s) {
                            if(province1.getAreaName().equals(province)){
                                if(city == null || city == ""){
                                    news.setTerritory(province1.getAreaCode());
                                    break;
                                }else{
                                    AreaQuery query = new AreaQuery();
                                    query.setParentId(province1.getId());
                                    List<Area> citys = areaService.queryList(query);
                                    for (Area ct:citys) {
                                        if(ct.getAreaName().equals((city))){
                                            news.setTerritory(ct.getAreaCode());
                                            break;
                                        }
                                    }
                                    break;
                                }
                            }
                    }
                    String userToken = ShiroUtils.getCurrentUserToken();
                    news.setUserToken(userToken);
                    news.setServiceState("1");
                    newsService.save(news);
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (workbook != null) {
                try {
                    workbook.cloneSheet(1);
                } catch (UnifiedJEXL.Exception e) {
                    e.printStackTrace();
                }
            }
        }
        return  ResponseJson.ok("导入成功");
    }

        /**
         * 取单元格的值
         * @param cell 单元格对象
         * @param treatAsStr 为true时，当做文本来取值 (取到的是文本，不会把“1”取成“1.0”)
         * @return
         */
        private String getCellValue(Cell cell ,boolean treatAsStr) {
            if (cell == null) {
                return "";
            }

            if (treatAsStr) {
                // 虽然excel中设置的都是文本，但是数字文本还被读错，如“1”取成“1.0”
                // 加上下面这句，临时把它当做文本来读取
                cell.setCellType(Cell.CELL_TYPE_STRING);
            }

            if (cell.getCellType() == Cell.CELL_TYPE_BOOLEAN) {
                return String.valueOf(cell.getBooleanCellValue());
            } else if (cell.getCellType() == Cell.CELL_TYPE_NUMERIC) {
                return String.valueOf(cell.getNumericCellValue());
            } else {
                return String.valueOf(cell.getStringCellValue());
            }
        }
}