java 如何写入 csv文件

Java写入CSV文件的方法有多种，主要包括使用原生IO、Apache Commons CSV库、OpenCSV库等。本文将详细介绍这些方法，并提供实际代码示例和最佳实践。

一、使用原生IO写入CSV文件

使用Java原生IO写入CSV文件是最基本的方法。它不依赖于任何外部库，但需要手动处理CSV格式的细节。

1.1 创建CSV文件和写入数据

首先，我们需要创建一个新的CSV文件，并使用FileWriter或BufferedWriter将数据写入该文件。

import java.io.BufferedWriter;
import java.io.FileWriter;
import java.io.IOException;
public class CsvWriterExample {
    public static void main(String[] args) {
        String csvFile = "example.csv";
        try (BufferedWriter writer = new BufferedWriter(new FileWriter(csvFile))) {
            // 写入表头
            writer.write("Name,Age,Country");
            writer.newLine();
            // 写入数据
            writer.write("Alice,30,USA");
            writer.newLine();
            writer.write("Bob,25,UK");
            writer.newLine();
            writer.write("Charlie,35,Canada");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

1.2 处理CSV格式的细节

在上述代码中，我们手动处理了CSV格式，但在实际应用中可能需要考虑更多的细节，如处理逗号、换行符、引号等特殊字符。

public static String escapeSpecialCharacters(String data) {
    String escapedData = data.replaceAll("\R", " ");
    if (data.contains(",") || data.contains(""") || data.contains("'")) {
        data = data.replace(""", """");
        escapedData = """ + data + """;
    }
    return escapedData;
}

二、使用Apache Commons CSV库

Apache Commons CSV库是一个专门处理CSV文件的库，提供了简单易用的API。它可以简化CSV文件的读写操作，并处理各种CSV格式的细节。

2.1 添加依赖

首先，需要在项目中添加Apache Commons CSV库的依赖。以Maven为例：

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

2.2 使用Commons CSV库写入CSV文件

import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVPrinter;
import java.io.FileWriter;
import java.io.IOException;
public class ApacheCommonsCsvWriterExample {
    public static void main(String[] args) {
        String csvFile = "example.csv";
        try (CSVPrinter printer = new CSVPrinter(new FileWriter(csvFile), CSVFormat.DEFAULT.withHeader("Name", "Age", "Country"))) {
            printer.printRecord("Alice", 30, "USA");
            printer.printRecord("Bob", 25, "UK");
            printer.printRecord("Charlie", 35, "Canada");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

三、使用OpenCSV库

OpenCSV是另一个流行的CSV处理库，提供了丰富的功能和简单的API。

3.1 添加依赖

首先，需要在项目中添加OpenCSV库的依赖。以Maven为例：

<dependency>
    <groupId>com.opencsv</groupId>
    <artifactId>opencsv</artifactId>
    <version>5.3</version>
</dependency>

3.2 使用OpenCSV库写入CSV文件

import com.opencsv.CSVWriter;
import java.io.FileWriter;
import java.io.IOException;
public class OpenCsvWriterExample {
    public static void main(String[] args) {
        String csvFile = "example.csv";
        try (CSVWriter writer = new CSVWriter(new FileWriter(csvFile))) {
            // 写入表头
            String[] header = {"Name", "Age", "Country"};
            writer.writeNext(header);
            // 写入数据
            String[] record1 = {"Alice", "30", "USA"};
            writer.writeNext(record1);
            String[] record2 = {"Bob", "25", "UK"};
            writer.writeNext(record2);
            String[] record3 = {"Charlie", "35", "Canada"};
            writer.writeNext(record3);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

四、处理CSV文件中的特殊字符

无论使用哪种方法写入CSV文件，都需要处理特殊字符，如逗号、换行符、引号等。以下是一些处理特殊字符的最佳实践：

4.1 使用引号包裹包含特殊字符的字段

在CSV文件中，使用双引号包裹包含逗号或换行符的字段。例如：

String[] record = {"Alice", "30", "USA, New York"};
writer.writeNext(record);

4.2 转义双引号

如果字段中包含双引号，需要将双引号转义为两个双引号。例如：

String[] record = {"Alice", "30", ""USA""};
writer.writeNext(record);

五、Java写入大文件的优化

写入大文件时，需要注意性能优化，以避免内存溢出和提高写入速度。以下是一些优化建议：

5.1 使用缓冲输出流

使用BufferedWriter或BufferedOutputStream可以提高写入速度，因为它们减少了实际IO操作的次数。

try (BufferedWriter writer = new BufferedWriter(new FileWriter(csvFile))) {
    // 写入数据
}

5.2 批量写入数据

将数据分批写入文件可以减少IO操作的次数，并提高写入效率。

List<String[]> records = new ArrayList<>();
records.add(new String[]{"Alice", "30", "USA"});
records.add(new String[]{"Bob", "25", "UK"});
records.add(new String[]{"Charlie", "35", "Canada"});
try (CSVWriter writer = new CSVWriter(new FileWriter(csvFile))) {
    writer.writeAll(records);
}

六、总结

写入CSV文件是Java开发中常见的操作，本文介绍了使用原生IO、Apache Commons CSV库、OpenCSV库等多种方法，并提供了实际代码示例和最佳实践。无论选择哪种方法，都需要注意处理CSV格式的细节和性能优化。希望本文对你有所帮助！