Java POI將數據庫表查詢結果導出到Excel實現及亂碼問題

使用java 的PO實現將從數據庫查詢的結果，導入到excel文件，並返回到頁面下載鏈接。

依賴：

        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi</artifactId>
            <version>3.15</version>
        </dependency>

代碼：

　　ResultSet rs為數據庫查詢結果

    public void write2excel(HttpServletResponse response,HttpServletRequest request, ResultSet rs, String fileName)
        throws SQLException, IOException {
        //創建HSSFWorkbook，一個HSSFWorkbook對應一個Excel文件
        HSSFWorkbook wb = new HSSFWorkbook();
        //在workbook中添加一個sheet,對應Excel文件中的sheet
        HSSFSheet sheet = wb.createSheet("sheet");
        //創建第一行
        HSSFRow row = sheet.createRow(0);
        HSSFCell cell;
        // 將列名寫入
        for (int j = 0; j < rs.getMetaData().getColumnCount(); j++) {
            String columnName = rs.getMetaData().getColumnLabel(j + 1);
            // 給列寫入數據,創建單元格，寫入數據
            cell = row.createCell(j);
            cell.setCellValue(columnName);

        }

        // 寫入正式數據
        int i = 0;
        while (rs.next()) {
            row = sheet.createRow(i + 1);
            for (int j = 0; j < rs.getMetaData().getColumnCount(); j++) {
                String columnValue = rs.getString(j + 1);
                row.createCell(j).setCellValue(columnValue);
            }
            ++i;
        }
        //上面的操作已經是生成一個完整的文件了，只需要將生成的流轉換成文件即可
        ServletOutputStream outputStream = null;


        //配置header
        // attachment這個代表要下載的，如果去掉就直接打開了(attachment-作為附件下載,inline-在線打開)
        // filename是文件名，另存為或者下載時，為默認的文件名
        String agent = request.getHeader("USER-AGENT").toLowerCase();
        response.reset();

        //下面開始到try之前主要為解決excel亂碼問題
        response.setContentType("multipart/form-data");
        response.setCharacterEncoding("utf-8");

        //火狐瀏覽器
        if (agent.contains("firefox")) {
            fileName = new String(fileName.getBytes("utf-8"), "ISO8859-1");
        } else {
            fileName = java.net.URLEncoder.encode(fileName, "UTF-8");
        }

        response.setHeader("Content-Disposition", "attachment;filename=" + fileName + ".xlsx");

        try{
            //生成response傳輸流，若是傳輸到本地則需創建FileOutputStream
            outputStream = response.getOutputStream();
            wb.write(outputStream);
        }catch (Exception e){
            e.printStackTrace();
        }finally {
            outputStream.flush();
            outputStream.close();

        }

    }

 

tips:

　　此外在開發中踩到的坑還有：在使用swagger調試接口的時候，下載導出文件excel到頁面時，會亂碼，但使用url加上參數（swagger接口下面有寫好的了）則導出到頁面的鏈接下載文件后能正常顯示，不亂碼。