1. 程式人生 > >Java SSM註解實現POI匯入匯出Excel

Java SSM註解實現POI匯入匯出Excel

POI生成Excel設定單元格格式:

HSSFWorkbook demoWorkBook = new HSSFWorkbook();   
HSSFSheet demoSheet = demoWorkBook.createSheet("The World's 500 Enterprises");   
HSSFCell cell = demoSheet.createRow(0).createCell(0);
 
設定單元格為文字格式
HSSFCellStyle cellStyle2 = demoWorkBook.createCellStyle();
HSSFDataFormat format = demoWorkBook.createDataFormat();
cellStyle2.setDataFormat(format.getFormat("@"));
cell.setCellStyle(cellStyle2);


第一種:日期格式
cell.setCellValue(new Date(2008,5,5));
//set date format
HSSFCellStyle cellStyle = demoWorkBook.createCellStyle();
HSSFDataFormat format= demoWorkBook.createDataFormat();
cellStyle.setDataFormat(format.getFormat("yyyy年m月d日"));
cell.setCellStyle(cellStyle);


第二種:保留兩位小數格式
cell.setCellValue(1.2);
HSSFCellStyle cellStyle = demoWorkBook.createCellStyle();
cellStyle.setDataFormat(HSSFDataFormat.getBuiltinFormat("0.00"));
cell.setCellStyle(cellStyle);


第三種:貨幣格式
cell.setCellValue(20000);
HSSFCellStyle cellStyle = demoWorkBook.createCellStyle();
HSSFDataFormat format= demoWorkBook.createDataFormat();
cellStyle.setDataFormat(format.getFormat("¥#,##0"));
cell.setCellStyle(cellStyle);


第四種:百分比格式
cell.setCellValue(20);
HSSFCellStyle cellStyle = demoWorkBook.createCellStyle();
cellStyle.setDataFormat(HSSFDataFormat.getBuiltinFormat("0.00%"));
cell.setCellStyle(cellStyle);


第五種:中文大寫格式
cell.setCellValue(20000);
HSSFCellStyle cellStyle = demoWorkBook.createCellStyle();
HSSFDataFormat format= demoWorkBook.createDataFormat();
cellStyle.setDataFormat(format.getFormat("[DbNum2][$-804]0"));
cell.setCellStyle(cellStyle);
 
第六種:科學計數法格式
cell.setCellValue(20000);
HSSFCellStyle cellStyle = demoWorkBook.createCellStyle();
cellStyle.setDataFormat( HSSFDataFormat.getBuiltinFormat("0.00E+00"));
cell.setCellStyle(cellStyle);

POI註解匯入匯出工具類:


/**
 * 匯出匯入excel相關工具
 */
public class ExcelUtil<T> {
    /**
     * datas為匯出的資料
     * fileName為excel檔名稱
     * sheetName為sheet名稱
     * 實體類註解通用匯出方法
     *
     * @param datas
     * @param fileName
     * @param sheetName
     */
    public static <T> void exportExcel(List<T> datas, String fileName, String sheetName, HttpServletResponse response, boolean displayAll) throws IOException, IllegalAccessException {
        // String fileExcelName = new SimpleDateFormat("yyyyMMddhh").format(new Date()).toString() + fileName;
        HSSFWorkbook workbook = new HSSFWorkbook();
        HSSFSheet sheet = workbook.createSheet(sheetName);
        HSSFRow row = sheet.createRow(0);//建立第一行
        HSSFCellStyle cellStyle = workbook.createCellStyle();
        HSSFDataFormat format = workbook.createDataFormat();
        cellStyle.setDataFormat(format.getFormat("@"));
        Boolean isHasTitle = false;
        for (int i = 0; i < datas.size(); i++) {
            HSSFRow rowBatch = sheet.createRow(i + 1);
            Object o = datas.get(i);
            // Class c = o.getClass();
            if (!isHasTitle) {
                for (Field field : o.getClass().getDeclaredFields()) {
                    field.setAccessible(true);
                    if (field.isAnnotationPresent(ExcelDesc.class)) {
                        //獲取欄位名
                        //String fieldNames = o.getClass().getSimpleName() + "." + field.getName();
                        //獲取欄位註解
                        ExcelDesc columnConfig = field.getAnnotation(ExcelDesc.class);
                        if (displayAll) {
                            //判斷是否已經獲取過該code的字典資料 避免重複獲取
                            HSSFCell cell = row.createCell(Integer.valueOf(columnConfig.orderBy().toString()));
                            cell.setCellStyle(cellStyle);
                            cell.setCellValue(columnConfig.value().toString());
                        }
                        if (!displayAll && columnConfig.display()) {
                            HSSFCell cell = row.createCell(Integer.valueOf(columnConfig.orderBy().toString()) - 1);
                            cell.setCellStyle(cellStyle);
                            cell.setCellValue(columnConfig.value().toString());
                        }
                    }
                }
                isHasTitle = true;
            }
            for (Field field : o.getClass().getDeclaredFields()) {
                field.setAccessible(true);
                if (field.isAnnotationPresent(ExcelDesc.class)) {
                    //獲取欄位名
                    //  String fieldNames = o.getClass().getSimpleName() + "." + field.getName();
                    //獲取欄位註解
                    ExcelDesc columnConfig = field.getAnnotation(ExcelDesc.class);
                    if (displayAll) {
                        HSSFCell cell = rowBatch.createCell(Integer.valueOf(columnConfig.orderBy().toString()));
                        cell.setCellStyle(cellStyle);
                        //判斷是否已經獲取過該code的字典資料 避免重複獲取
                        cell.setCellValue(field.get(o) == null ? "" : field.get(o).toString());
                        cell.setCellType(Cell.CELL_TYPE_STRING);
                    }
                    if (!displayAll && columnConfig.display()) {
                        HSSFCell cell = rowBatch.createCell(Integer.valueOf(columnConfig.orderBy().toString()) - 1);
                        cell.setCellStyle(cellStyle);
                        cell.setCellValue(field.get(o) == null ? "" : field.get(o).toString());
                        cell.setCellType(Cell.CELL_TYPE_STRING);
                    }
                }
            }

        }
        // 彈出儲存框方式
        ByteArrayOutputStream os = new ByteArrayOutputStream();
        try {
            workbook.write(os);

        } catch (IOException e) {
            e.printStackTrace();
        }
        byte[] content = os.toByteArray();
        InputStream is = new ByteArrayInputStream(content);
        // 設定response引數,可以開啟下載頁面
        response.reset();
        response.setContentType("application/vnd.ms-excel;charset=utf-8");
        response.setHeader("Content-Disposition", "attachment;filename=" + new String(fileName.getBytes("gb2312"), "ISO8859-1") + ".xls");

        ServletOutputStream out = response.getOutputStream();
        BufferedInputStream bis = null;
        BufferedOutputStream bos = null;
        try {
            bis = new BufferedInputStream(is);
            bos = new BufferedOutputStream(out);
            byte[] buff = new byte[2048];
            int bytesRead;
            // Simple read/write loop.
            while (-1 != (bytesRead = bis.read(buff, 0, buff.length))) {
                bos.write(buff, 0, bytesRead);
            }
        } catch (Exception e) {
            // TODO: handle exception
            e.printStackTrace();
        } finally {
            if (bis != null)
                bis.close();
            if (bos != null)
                bos.close();
        }

        // sheet.autoSizeColumn(i);
        // sheet.SetColumnWidth(i, 100 * 256);
    }


    Class<T> clazz;

    public ExcelUtil(Class<T> clazz) {
        this.clazz = clazz;
    }

    public List<T> importExcel(String fileName, String sheetName, InputStream input) {
        int maxCol = 0;
        List<T> list = new ArrayList<T>();
        try {
            Workbook workbook = null;
            String fileType = fileName.substring(fileName.lastIndexOf("."), fileName.length());
            if (".xls".equals(fileType.trim().toLowerCase())) {
                workbook = new HSSFWorkbook(input);// 建立 Excel 2003 工作簿物件
            } else if (".xlsx".equals(fileType.trim().toLowerCase())) {
                workbook = new XSSFWorkbook(input);//建立 Excel 2007 工作簿物件
            }//解決Excel匯入的相容2003和2007版本的問題
            Sheet sheet = workbook.getSheet(sheetName);
            if (!sheetName.trim().equals("")) {
                sheet = workbook.getSheet(sheetName);// 如果指定sheet名,則取指定sheet中的內容.
            }
            if (sheet == null) {
                sheet = workbook.getSheetAt(0); // 如果傳入的sheet名不存在則預設指向第1個sheet.
            }
            int rows = sheet.getPhysicalNumberOfRows();

            if (rows > 0) {// 有資料時才處理
                // Field[] allFields = clazz.getDeclaredFields();// 得到類的所有field.
                List<Field> allFields = getMappedFiled(clazz, null);

                Map<Integer, Field> fieldsMap = new HashMap<Integer, Field>();// 定義一個map用於存放列的序號和field.
                for (Field field : allFields) {
                    // 將有註解的field存放到map中.
                    if (field.isAnnotationPresent(ExcelDesc.class)) {
                        ExcelDesc attr = field.getAnnotation(ExcelDesc.class);
                        int col = Integer.valueOf(attr.orderBy().toString());
                        maxCol = Math.max(col, maxCol);
                        System.out.println(col + "====" + field.getName());
                        field.setAccessible(true);// 設定類的私有欄位屬性可訪問.
                        fieldsMap.put(col, field);
                    }
                }
                for (int i = 1; i < rows; i++) {// 從第2行開始取資料,預設第一行是表頭.
                    Row row = sheet.getRow(i);
                    // int cellNum = row.getPhysicalNumberOfCells();
                    // int cellNum = row.getLastCellNum();
                    int cellNum = maxCol;
                    T entity = null;
                    for (int j = 0; j < cellNum; j++) {
                        Cell cell = row.getCell(j);
                        if (cell == null) {
                            continue;
                        }
                        int cellType = cell.getCellType();
                        String c = "";
                        if (cellType == HSSFCell.CELL_TYPE_NUMERIC) {
                            c = String.valueOf(cell.getNumericCellValue());
                        } else if (cellType == HSSFCell.CELL_TYPE_BOOLEAN) {
                            c = String.valueOf(cell.getBooleanCellValue());
                        } else {
                            c = cell.getStringCellValue();
                        }
                        if (c == null || c.equals("")) {
                            continue;
                        }
                        entity = (entity == null ? clazz.newInstance() : entity);// 如果不存在例項則新建.
                        // System.out.println(cells[j].getContents());
                        Field field = fieldsMap.get(j + 1);// 從map中得到對應列的field.
                        if (field == null) {
                            continue;
                        }
                        // 取得型別,並根據物件型別設定值.
                        Class<?> fieldType = field.getType();
                        if (String.class == fieldType) {
                            field.set(entity, String.valueOf(c));
                        } else if ((Integer.TYPE == fieldType) || (Integer.class == fieldType)) {
                            field.set(entity, Integer.parseInt(c));
                        } else if ((Long.TYPE == fieldType) || (Long.class == fieldType)) {
                            field.set(entity, Long.valueOf(c));
                        } else if ((Float.TYPE == fieldType) || (Float.class == fieldType)) {
                            field.set(entity, Float.valueOf(c));
                        } else if ((Short.TYPE == fieldType) || (Short.class == fieldType)) {
                            field.set(entity, Short.valueOf(c));
                        } else if ((Double.TYPE == fieldType) || (Double.class == fieldType)) {
                            field.set(entity, Double.valueOf(c));
                        } else if (Character.TYPE == fieldType) {
                            if ((c != null) && (c.length() > 0)) {
                                field.set(entity, Character.valueOf(c.charAt(0)));
                            }
                        }

                    }
                    if (entity != null) {
                        list.add(entity);
                    }
                }
            }

        } catch (IOException e) {
            e.printStackTrace();
        } catch (InstantiationException e) {
            e.printStackTrace();
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        } catch (IllegalArgumentException e) {
            e.printStackTrace();
        }
        return list;
    }

    /**
     * 得到實體類所有通過註解映射了資料表的欄位
     *
     * @return
     */
    @SuppressWarnings("rawtypes")
    private List<Field> getMappedFiled(Class clazz, List<Field> fields) {
        if (fields == null) {
            fields = new ArrayList<Field>();
        }

        Field[] allFields = clazz.getDeclaredFields();// 得到所有定義欄位
        // 得到所有field並存放到一個list中.
        for (Field field : allFields) {
            if (field.isAnnotationPresent(ExcelDesc.class)) {
                fields.add(field);
            }
        }
        if (clazz.getSuperclass() != null
                && !clazz.getSuperclass().equals(Object.class)) {
            getMappedFiled(clazz.getSuperclass(), fields);
        }

        return fields;
    }
}


註解類:

@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
public @interface ExcelDesc {
    String value();
    String orderBy();
    boolean display() default true;
}

實體類:
public class ProductExcel {

    @ExcelDesc(value = "商品編碼" , orderBy = "1")
    public String  SKU;
    @ExcelDesc(value = "商品名稱" , orderBy = "2")
    public String productName;
    @ExcelDesc(value = "價格" , orderBy = "3")
    public String productPrice;
}

maven,pom.xml:
<dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi-ooxml</artifactId>
            <version>3.14</version>
        </dependency>
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi-ooxml-schemas</artifactId>
            <version>3.14</version>
        </dependency>
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi</artifactId>
            <version>3.14</version>
        </dependency>

SSM,controller層:
@Controller
@RequestMapping("product")
public class ProductController{

    /**
     * 匯出商品價格報表
     * @param response
     */
    @RequestMapping(value = "export_excel/product_list",method = RequestMethod.GET)
    public void excelProductList(HttpServletResponse response) {
        try {
            List<ProductExcel> productExcels = productService.getProductExcel();
            ExcelUtil.exportExcel(productExcels, "商品價格統計", "商品價格列表", response, false);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 匯入商品價格報表
     */
    @RequestMapping(value = "import_excel/product_list",method = RequestMethod.POST)
    @ResponseBody
    public String importProductExcel(@RequestParam(value = "excel") MultipartFile excel) {
        try {
            ExcelUtil<ProductExcel> util = new ExcelUtil<ProductExcel>(ProductExcel.class);// 建立excel工具類
            List<ProductExcel> list = util.importExcel(excel.getOriginalFilename(),"商品價格列表", excel.getInputStream());// 匯出
            if(list.size() > 0){
                if(productService.batchUpdateImportProduct(list,user) > 0){
                    return "success";
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return "error";
    }
}