java實現接口導出csv文件

Tomxin7

Simple, Interesting | 簡單，有趣

業務介紹

項目要求從數據庫中查詢出相關數據后，通過表格展示給用戶，如果用戶需要，可以點擊導出按鈕，導出數據為csv格式。

開發環境：JDK7、Tomcat7、SpringBoot

網頁中的表格

下載后的表格

業務流程

1、數據庫查詢數據
2、創建臨時csv文件
3、輸出csv流文件，提供給瀏覽器下載
4、刪除臨時文件

完整代碼

測試地址：127.0.0.1:8080/export

pom文件添加以下依賴
    <dependency>
      <groupId>net.sourceforge.javacsv</groupId>
      <artifactId>javacsv</artifactId>
      <version>2.0</version>
    </dependency>

       /**
     * 導出行為執行明細記錄
     * @return
     */
    @RequestMapping(value = "/export", method = RequestMethod.GET)
    public void exportRecord(HttpServletResponse response){

            // 創建對象，如果是實際業務請從數據庫取出數據
            ArrayList<ActionDetailsDto> datas = new ArrayList();
            ActionDetailsDto actionDetailsDto = new ActionDetailsDto();
            actionDetailsDto.setContent("您好，請確認你7月份的績效系統");
            actionDetailsDto.setObjectName("tomxin");
            actionDetailsDto.setObjectId(123456L);
            datas.add(actionDetailsDto);
            
        try {
            //創建臨時csv文件
            File tempFile = createTempFile(datas);
            //輸出csv流文件，提供給瀏覽器下載
            outCsvStream(response, tempFile);
            //刪除臨時文件
            deleteFile(tempFile);

        } catch (IOException e) {
            System.out.println("導出失敗");
        }
    }

    /**
     * 創建臨時的csv文件
     * @return
     * @throws IOException
     */
    public File createTempFile(List<ActionDetails> datas) throws IOException {
        File tempFile = File.createTempFile("vehicle", ".csv");
        CsvWriter csvWriter = new CsvWriter(tempFile.getCanonicalPath(), ',', Charset.forName("UTF-8"));
        // 寫表頭
        String[] headers = {"行為明細", "執行對象","觸發時間","狀態"};
        csvWriter.writeRecord(headers);
        for (ActionDetails data : datas) {
            //這里如果數據不是String類型，請進行轉換
            csvWriter.write(data.getContent());
            csvWriter.write(data.getObjectName());
            csvWriter.write(data.getCreateTime());
            csvWriter.write(data.getStatus());
            csvWriter.endRecord();
        }
        csvWriter.close();
        return tempFile;
    }

    /**
     * 寫入csv結束，寫出流
     */
    public void outCsvStream(HttpServletResponse response,File tempFile) throws IOException {
        java.io.OutputStream out = response.getOutputStream();
        byte[] b = new byte[10240];
        java.io.File fileLoad = new java.io.File(tempFile.getCanonicalPath());
        response.reset();
        response.setContentType("application/csv");
        response.setHeader("content-disposition", "attachment; filename=export.csv");
        java.io.FileInputStream in = new java.io.FileInputStream(fileLoad);
        int n;
        //為了保證excel打開csv不出現中文亂碼
        out.write(new   byte []{( byte ) 0xEF ,( byte ) 0xBB ,( byte ) 0xBF });
        while ((n = in.read(b)) != -1) {
            //每次寫入out1024字節
            out.write(b, 0, n);
        }
        in.close();
        out.close();
    }

    /**
     * 刪除單個文件
     *
     * @return 單個文件刪除成功返回true，否則返回false
     */
    public static boolean deleteFile( File file) {
        // 如果文件路徑所對應的文件存在，並且是一個文件，則直接刪除
        if (file.exists() && file.isFile()) {
            if (file.delete()) {
                return true;
            } else {
                return false;
            }
        } else {
            return false;
        }
    }

部分參考：https://blog.csdn.net/qq_32447321/article/details/75025824