需求描述

將從資料庫查出的資料以生成csv檔案的方式下載下來（匯出）。

實現方案

最開始的方法是：

將從資料庫中查出的資料拼成符合csv檔案格式的字串，然後將字串轉為輸入流，最後返回InputStreamResource，達到下載的目的。下邊是實現的程式碼：

    @GetMapping("/test/csv")
    public ResponseEntity<InputStreamResource> createCSV(@RequestParam(required = false) String key1, @RequestParam(required = false
 
) String key2) throws IOException{

        String fileName = UuidUtils.base58Uuid()+".csv"; //檔名是生成的base58的uuid，這個方法是本人自己寫的

        HttpHeaders headers = new HttpHeaders();
        headers.add("Cache-Control", "no-cache, no-store, must-revalidate");
        headers.add("Content-Disposition","attachment; fileName="
 
+fileName);
        headers.add("Pragma", "no-cache");
        headers.add("Expires", "0");

        //查詢員工列表
        List<User> users = findUsers(key1,key2);
        //資料準備
        StringBuilder str = new StringBuilder();
        str.append("員工id,姓名,手機號\r\n");
        for(User e : users){
                str.append(e.getId()+","
 
+e.getName()+","+e.getMobilePhone()+"\r\n");
        }

        //生成輸入流
        InputStream in = new ByteArrayInputStream(str.toString().getBytes("gbk"));
        return ResponseEntity
                .ok()
                .headers(headers)
                .contentType(MediaType.parseMediaType("application/x-msdownload"))
                .body(new InputStreamResource(in));
    }

上邊程式碼的問題在於：先將資料庫中查出的資料放入記憶體中，最後轉成輸入流，但是當資料庫中資料龐大的時候，會給記憶體造成很大的負擔，甚至溢位，說俗點就是你的記憶體炸了 哦呵呵呵。

思考解決方法： 能不能分頁查詢，每查出一部分都放入流中，然後清除記憶體中的資料，然後再進行查詢，再放入流中，再清除記憶體中的資料，如此迴圈直到查完為止。

解決方案：

    @GetMapping("/test/csv")
    public void createCSV(HttpServletResponse response, @RequestParam(required = false) String key1, @RequestParam(required = false) String key2)  throws IOException {
        //生成檔名
        String fileName = UuidUtils.base58Uuid()+".csv";

        //設定返回資料名字、型別和編碼格式
        response.addHeader("Content-Disposition","attachment; fileName="+fileName);
        response.setContentType("application/octet-stream");

        //編碼格式設為gbk，因為window底層編碼格式是gbk，如果這裡不設為gbk的話 office開啟會亂碼，wps不會。
        //另外 如果根本就不設定編碼的話會亂碼的，所以一定要設定，至於設定成gbk還是utf-8視情況而定吧。個人推薦是gbk
        response.setCharacterEncoding("gbk"); 

        PrintWriter writer = response.getWriter();

        int page = 0; //查詢第幾頁數
        int size = 500; //每頁資料條數
        Pageable pageable = new PageRequest(page,size);
        Page<User> userPage = findUsers(key1,key2,pageable);

        StringBuilder str = new StringBuilder();
        str.append("員工id,姓名,手機號\r\n");

        boolean flag = false;
        for(;true;){
            List<User> users = userPage.getContent();
            for(User e : users){
                str.append(csvString(e.getId()));
                str.append(",");
                str.append(csvString(e.getName()));
                str.append(",");
                str.append(csvString(e.getMobilePhone()));
                str.append("\r\n");
            }


            //  這裡是重點！！！ 
            //  先把拼好的字串write出去，然後flush下 ：Response物件之Flush方法，立即傳送緩衝區中的輸出
            //  如果未將 Response.Buffer 設定為 TRUE，則該方法將導致執行時錯誤。
            //  語法：Response.Flush；註釋：如果在 ASP 頁上呼叫 Flush 方法，則伺服器將響應該頁上保持活動的請求。應用於Response物件。
            writer.write(str.toString());
            writer.flush();

            //清空str中的資料
            str.delete(0,str.length());

            //判斷是否是最後一頁
            flag = userPage.isLast();
            if(flag){
                break;
            }

            //不是最後一頁 繼續查資料
            page++;
            pageable = new PageRequest(page,size);
            userPage = findUsers(key1,key2,pageable);
        }

        writer.close();
    }



    /**
     * 生成csv檔案之前特殊字元轉義
     * @param str
     * @return
     */
    public String csvString(String str){

        if(str.contains(",")){
            str = str.replace("\"","\"\"");
            str ="\"" + str + "\"";
        }

        return str;
    }

說明：第二個方法中Pageable是spring boot封裝的一個實體類，在jpa中查詢資料的時候傳入pageable會自動進行分頁查詢，具體使用方法百度

最後推薦一個 Java檔案下載方法總結 的比較全的文章 點選檢視