Java 匯出 CSV 檔案操作詳情

2022-08-17 18:01:33

首先第一步匯入座標：

<dependency>
<groupId>org.apache.commons</groupId>
<artifactId>commons-csv</artifactId>
<version>1.6</version>
</dependency>

第二步

引入工具類說明下因為這個工具類用到是Listj集合
我就順帶吧實體類和map 之間的轉換也說了

import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVParser;
import org.apache.commons.csv.CSVPrinter;
import org.apache.commons.csv.CSVRecord;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;

import java.io.*;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

/**
* 匯出至csv檔案
* */
public class CsvUtil {
//CSV檔案分隔符
private final static String NEW_LINE_SEPARATOR="n";
/** CSV檔案列分隔符 */
private static final String CSV_COLUMN_SEPARATOR = ",";
/** CSV檔案列分隔符 */
private static final String CSV_RN = "rn";

/**寫入csv檔案
* @param headers 列頭
* @param data 資料內容
* @param filePath 建立的csv檔案路徑
* @throws IOException **/
public static void writeCsvWithHeader(String[] headers, List<Object[]> data, String filePath) {
//初始化csvformat
CSVFormat format = CSVFormat.DEFAULT.withHeader(headers);
try {
//根據路徑建立檔案，並設定編碼格式
FileOutputStream fos = new FileOutputStream(filePath);
OutputStreamWriter osw = new OutputStreamWriter(fos, "GBK");
//建立CSVPrinter物件
CSVPrinter printer = new CSVPrinter(osw, format);

if(null!=data){
//迴圈寫入資料
for(Object[] lineData:data){
printer.printRecord(lineData);
}
}
printer.flush();
printer.close();
} catch (IOException e) {
e.printStackTrace();
}
}
/**寫入csv檔案
* @param headers 列頭
* @param data 資料內容
* @param filePath 建立的csv檔案路徑
* @throws IOException **/
public static void writeCsvWithRecordSeparator(Object[] headers, List<Object[]> data, String filePath){
//初始化csvformat
CSVFormat format = CSVFormat.DEFAULT.withRecordSeparator(NEW_LINE_SEPARATOR);
try {
//根據路徑建立檔案，並設定編碼格式
FileOutputStream fos = new FileOutputStream(filePath);
OutputStreamWriter osw = new OutputStreamWriter(fos, "GBK");
//建立CSVPrinter物件
CSVPrinter printer = new CSVPrinter(osw,format);
//寫入列頭資料
printer.printRecord(headers);

if(null!=data){
//迴圈寫入資料
for(Object[] lineData:data){
printer.printRecord(lineData);
}
}
printer.flush();
printer.close();
System.out.println("CSV檔案建立成功,檔案路徑:"+filePath);
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* @filePath 檔案路徑
*/
public static List<CSVRecord> readCsvParse(String filePath){
List<CSVRecord> records = new ArrayList<>();
try {
FileInputStream in = new FileInputStream(filePath);
BufferedReader reader = new BufferedReader (new InputStreamReader(in,"GBK"));
CSVParser parser = CSVFormat.EXCEL.parse(reader);
records = parser.getRecords();
parser.close();
} catch (IOException e) {
e.printStackTrace();
}finally {
return records;
}
}

/**
* 自定義欄位
* @filePath 檔案路徑
*/
public static List<CSVRecord> readCsvParseWithHeader(String filePath,String[] headers){
List<CSVRecord> records = new ArrayList<>();
try {
FileInputStream in = new FileInputStream(filePath);
BufferedReader reader = new BufferedReader (new InputStreamReader(in,"GBK"));
CSVParser parser = CSVFormat.EXCEL.withHeader(headers).parse(reader);
records = parser.getRecords();
/*for (CSVRecord record : parser.getRecords()) {
System.out.println(record.get("id") + ","
+ record.get("name") + ","
+ record.get("code"));
}*/
parser.close();
}catch (IOException e){
e.printStackTrace();
}finally {
return records;
}
}
/**
* 匯出至多個csv檔案
* */
public void writeMuti() throws InterruptedException {
ExecutorService executorService = Executors.newFixedThreadPool(3);
CountDownLatch doneSignal = new CountDownLatch(2);

List<Map<String, String>> recordList = new ArrayList<>();

executorService.submit(new CsvExportThread("E:/0.csv", recordList, doneSignal));
executorService.submit(new CsvExportThread("E:/1.csv", recordList, doneSignal));

doneSignal.await();
System.out.println("Finish!!!");
}
/**
* @param colNames 表頭部資料
* @param dataList 集合資料
* @param mapKeys 查詢的對應資料
*/
public static ByteArrayOutputStream doExport(String[] colNames, String[] mapKeys, List<Map> dataList) {
try {
StringBuffer buf = new StringBuffer();

// 完成資料csv檔案的封裝
// 輸出列頭
for (int i = 0; i < colNames.length; i++) {
buf.append(colNames[i]).append(CSV_COLUMN_SEPARATOR);
}
buf.append(CSV_RN);

if (null != dataList) { // 輸出資料
for (int i = 0; i < dataList.size(); i++) {
for (int j = 0; j < mapKeys.length; j++) {
buf.append(dataList.get(i).get(mapKeys[j])).append(CSV_COLUMN_SEPARATOR);
}
buf.append(CSV_RN);
}
}
// 寫出響應
ByteArrayOutputStream os = new ByteArrayOutputStream();
//OutputStream os = new ByteArrayOutputStream();
os.write(buf.toString().getBytes("GBK"));
os.flush();
os.close();
return os;
} catch (Exception e) {
LogUtils.error("doExport錯誤...", e);
e.printStackTrace();
}
return null;
}
public static HttpHeaders setCsvHeader(String fileName) {
HttpHeaders headers = new HttpHeaders();
try {
// 設定檔案字尾
SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMddHHmmss");
String filename = new String(fileName.getBytes("gbk"), "iso8859-1") + sdf.format(new Date()) + ".csv";
headers.add("Pragma", "public");
headers.add("Cache-Control", "max-age=30");
headers.add("Content-Disposition", "attachment;filename="+filename);
headers.setContentType(MediaType.valueOf("application/vnd.ms-excel;charset=UTF-8"));
}catch (Exception e){
e.printStackTrace();
}
return headers;
}
}

第三步

controller 層：

package com.example.demo.controller;

import com.example.demo.service.DemoService;
import com.example.demo.util.CsvUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;

@Controller
@RequestMapping("/demo")
public class DemoController {

@Autowired
private DemoService demoService;

@RequestMapping("/exportCsv")
public ResponseEntity<byte[]> exportCsv(){
//設定excel檔名
String fileName="使用者表";
//設定HttpHeaders，設定fileName編碼，排除匯出檔名稱亂碼問題
HttpHeaders headers = CsvUtil.setCsvHeader(fileName);
byte[] value = null;
try {
//獲取要匯出的資料
value = this.demoService.exportCsv();
}catch (Exception e){
e.printStackTrace();
}
return new ResponseEntity<byte[]>(value,headers, HttpStatus.OK);
}
}

第四步：

service 介面

package com.example.demo.service;

public interface DemoService {
/*匯出csv檔案*/
byte[] exportCsv();
}

第五步：

service 實現類

package com.example.demo.service.impl;
import com.example.demo.pojo.User;
import com.example.demo.service.DemoService;
import com.example.demo.util.CsvUtil;
import org.apache.commons.beanutils.BeanUtils;
import org.springframework.stereotype.Service;

import java.io.ByteArrayOutputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

@Service
public class DemoServiceImpl implements DemoService {

@Override
public byte[] exportCsv() {
byte[] content = null;
try {
String[] sTitles = new String[]{"名稱","年齡","性別"};
String[] mapKeys = new String[]{"name","age","sex"};
List<Map> dataList = new ArrayList<>();
//資料
for (int i = 0; i < 10; i++) {
User user = new User("小明" + i, i, "男" + i);
Map map = BeanUtils.describe(user);
dataList.add(map);
}

ByteArrayOutputStream os = CsvUtil.doExport(sTitles,mapKeys,dataList);
content = os.toByteArray();
}catch (Exception e){
e.printStackTrace();
}
return content;
}
}

補充 :

象和map 互相轉換
座標

<dependency>
<groupId>commons-beanutils</groupId>
<artifactId>commons-beanutils</artifactId>
<version>1.9.3</version>
</dependency>

用到的工具類：

BeanUtils

1. bean( 實體類)轉Map

例子:

Person person=new Person();
person1.setName("張三");
person1.setSex("不男不女");
Map<String, Object> map=null;

map = BeanUtils.describe(person1);

2. map轉bean(實體類)

例子: 估計誰沒事也不會用map 轉bean 可能是我見識短了/**

* Map轉換層Bean，使用泛型免去了型別轉換的麻煩。
* @param <T>
* @param map
* @param class1
* @return
*/
public static <T> T map2Bean(Map<String, String> map, Class<T> class1) {
T bean = null;
try {
bean = class1.newInstance();
BeanUtils.populate(bean, map);
} catch (Exception e) {
e.printStackTrace();
}
return bean;
}

到此這篇關於Java 匯出 CSV 檔案操作詳情的文章就介紹到這了,更多相關Java 匯出 CSV 檔案內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！

Java 匯出 CSV 檔案操作詳情

熱門文章