迹忆客 专注技术分享

CSV 代表 Comma Separated Values；它是系统之间大量数据传输的常用格式。还有许多与 Java 一起使用的解析器库能够处理这种格式。

在 Java 中使用 PrintWriter 读取和写入 CSV 文件

Java 中的 PrinterWriter 函数是一个编写器类，用于将对象的格式化表示打印到文本输出流。我们创建了一个 writer 对象，传递一个名为 test.csv 的新文件作为 writer 的目的地。在这里，sb 对象将指定的字符串附加到字符序列。

writer 实例上的 write() 方法将文本内容写入流中。flush() 方法将内容刷新到文件中，close() 方法永久关闭流。我们还可以读取 test.csv 文件的内容。

在 TestCSV 类实例上调用 readCSVFile() 方法。在这里，我们使用指定文件中的值创建了 Scanner 类的实例。它使用分隔符模式将数据分解为标记。如果 Scanner 有另一行输入，则 hasNextLine() 返回 true。因此，我们使用 next() 方法读取每一行标记数据并将其存储在一个数组中，最后打印输出。

import java.io.File;
import java.io.FileNotFoundException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

public class TestCSV {
    public static void main(String args[]){
        try (PrintWriter writer = new PrintWriter(new File("test.csv"))) {

            StringBuilder sb = new StringBuilder();
            sb.append("id");
            sb.append(',');
            sb.append("Name");
            sb.append(',');
            sb.append("Address");
            sb.append('\n');

            sb.append("101");
            sb.append(',');
            sb.append("John Doe");
            sb.append(',');
            sb.append("Las Vegas");
            sb.append('\n');

            writer.write(sb.toString());
            writer.close();
            System.out.println("done!");

        } catch (FileNotFoundException e) {
            System.out.println(e.getMessage());
        }
        TestCSV testCSV =  new TestCSV();
        testCSV.readCSVFile();
    }

    public void readCSVFile(){
        List<List<String>> records = new ArrayList<>();
        try (Scanner scanner = new Scanner(new File("test.csv"));) {
            while (scanner.hasNextLine()) {
                records.add(getRecordFromLine(scanner.nextLine()));
            }
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        }
        System.out.println(records.toString());
    }
    private List<String> getRecordFromLine(String line) {
        List<String> values = new ArrayList<String>();
        try (Scanner rowScanner = new Scanner(line)) {
            rowScanner.useDelimiter(",");
            while (rowScanner.hasNext()) {
                values.add(rowScanner.next());
            }
        }
        return values;
    }

}

输出：

done!
[[id, Name, Address], [101, John Doe, Las Vegas]]

在 Java 中使用 OpenCSV 库读取和写入 CSV 文件

这个库的 maven 依赖如下：

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

OpenCsv 是一个简单的 java 解析器库；它有一组 OpenCsv 类，我们用来读取和写入 CSV 文件。在 main() 方法中，我们首先使用 CSVWriter 类调用该方法以写入 CSV 文件。此类用于将 CSV 数据写入编写器实现。

我们创建一个 CSVWriter 的 writer 实例，并在该对象上调用 writeNext() 函数以生成一个 CSV 文件，其中包含使用分隔符分隔的字符串数组中的数据。close() 方法关闭编写器流。

为了从我们以 CSV 格式创建的文件中读取数据，我们在主类对象上调用 readCSVFile 方法，在那里我们解析 CSV 文件。为了一次将所有记录读入列表，我们使用 readAll() 方法。我们遍历每条记录并将其打印出来。

import com.opencsv.CSVReader;
import com.opencsv.CSVWriter;
import com.opencsv.exceptions.CsvException;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.util.Arrays;
import java.util.List;

public class CSVLibrary {
    public static void main (String args[])  {
        CSVLibrary csvLibrary =  new CSVLibrary();
        csvLibrary.writeToCsv();
        csvLibrary.readCSVFile();
    }
    public void writeToCsv(){
        String csv = "data.csv";
        try{
            CSVWriter writer = new CSVWriter(new FileWriter(csv));
            String [] record = "2,Virat,Kohli,India,30".split(",");
            writer.writeNext(record);

            writer.close();
        }catch (IOException e) {
            e.printStackTrace();
        }

    }
    public void readCSVFile(){
        CSVReader reader = null;
        try {
            reader = new CSVReader(new FileReader("data.csv"));
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        }

        List<String[]> allRows = null;
        try {
            allRows = reader.readAll();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (CsvException e) {
            e.printStackTrace();
        }

        for(String[] row : allRows){
            System.out.println(Arrays.toString(row));
        }
    }
}

输出：

[2, Virat, Kohli, India, 30]