java 生产csv文件转成流

原创

mob649e816880fe 2023-10-15 09:48:51 ©著作权

文章标签 CSV java Java 文章分类 Java 后端开发

©著作权归作者所有：来自51CTO博客作者mob649e816880fe的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java生产CSV文件转成流

CSV（Comma Separated Values）是一种常见的文件格式，用于存储和交换表格数据。在Java中，我们可以使用各种库和技术来生成CSV文件并将其转换为流。

本文将介绍如何使用Java生成CSV文件，并将其转换为流供其他程序使用。我们将使用Apache Commons CSV库来简化CSV文件的创建和处理过程。

Apache Commons CSV库

Apache Commons CSV是一个开源的Java库，用于处理CSV文件。它提供了一组简单而强大的API，可用于读取、写入和操作CSV文件。

要使用Apache Commons CSV库，请将以下Maven依赖项添加到您的项目中：

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

生成CSV文件

首先，我们需要创建CSV文件并将数据写入其中。以下是一个示例方法，它接受一个List作为输入，将其转换为CSV格式并将其写入文件：

import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVPrinter;

import java.io.FileWriter;
import java.io.IOException;
import java.io.Writer;
import java.util.List;

public class CSVGenerator {

    public static void generateCSV(List<String[]> data, String filePath) {
        try (Writer writer = new FileWriter(filePath);
             CSVPrinter csvPrinter = new CSVPrinter(writer, CSVFormat.DEFAULT)) {
            for (String[] record : data) {
                csvPrinter.printRecord((Object[]) record);
            }
            csvPrinter.flush();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

在上面的示例中，我们使用CSVPrinter来将数据写入CSV文件。CSVFormat.DEFAULT表示使用默认的CSV格式。

转换为流

要将生成的CSV文件转换为流，我们可以使用Java的InputStream或ByteArrayInputStream类。以下是一个示例方法，将CSV文件转换为InputStream：

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;

public class CSVConverter {

    public static InputStream convertToStream(String filePath) {
        try {
            Path path = Paths.get(filePath);
            byte[] data = Files.readAllBytes(path);
            return new ByteArrayInputStream(data);
        } catch (IOException e) {
            e.printStackTrace();
            return null;
        }
    }
}

上面的示例中，我们使用Files.readAllBytes方法将CSV文件读取为字节数组，然后使用ByteArrayInputStream将其转换为InputStream。

示例应用

让我们使用一个示例应用程序来演示如何使用上述代码生成CSV文件并将其转换为流。

import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;

public class Main {

    public static void main(String[] args) {
        List<String[]> data = new ArrayList<>();
        data.add(new String[]{"Name", "Age", "City"});
        data.add(new String[]{"John Doe", "30", "New York"});
        data.add(new String[]{"Jane Smith", "25", "London"});

        String filePath = "data.csv";
        CSVGenerator.generateCSV(data, filePath);

        InputStream inputStream = CSVConverter.convertToStream(filePath);

        // 在这里使用InputStream进行其他操作
    }
}

在上面的示例中，我们首先创建一个List对象，其中包含要写入CSV文件的数据。然后，我们使用CSVGenerator类将数据写入名为data.csv的文件。最后，我们使用CSVConverter类将生成的CSV文件转换为流。