Java读写CSV文件

Maven依赖:

        <!-- CSV 依赖-->
        <!-- https://mvnrepository.com/artifact/net.sourceforge.javacsv/javacsv -->
        <dependency>
            <groupId>net.sourceforge.javacsv</groupId>
            <artifactId>javacsv</artifactId>
            <version>2.0</version>
        </dependency>

导包:

import com.csvreader.CsvReader;
import com.csvreader.CsvWriter;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;

import java.io.File;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.List;

代码:

@Slf4j
public class CsvUtils {
    /**
     * 从SN文件中读取内容(可读取某一列也可读取整行,根据需要调整逻辑)
     *
     * @param filePath 文件路径
     * @param encoding 编码方式
     * @return 返回SN集合
     */
    public static List<String> read(String filePath, String encoding) {
        List<String> snList = new ArrayList<>();
        try {
            // 创建CSV读对象
            CsvReader csvReader = new CsvReader(filePath, ',', Charset.forName(encoding));
            // 读表头
            csvReader.readHeaders();
            while (csvReader.readRecord()) {
                // 读一整行
//                System.out.println(csvReader.getRawRecord());
                // 读这行的某一列
                String sn = csvReader.get("SN号");
                if (!StringUtils.isEmpty(sn) && !snList.contains(sn)) {
                    //过滤:去空、去重
                    snList.add(sn);
                }
            }
        } catch (Exception e) {
            log.error("CSV文件读取失败,异常:", e);
        }
        return snList;
    }

    /**
     * 生成结果文件
     *
     * @param filePath 结果文件目录
     * @param encoding 编码方式
     * @param headers  表头内容
     * @param dataList 数据集合
     */
    public static void write(String filePath, String encoding, String[] headers, List<String[]> dataList) {
        try {
            File file = new File(filePath).getParentFile();
            if (!file.exists()) {
                file.mkdirs();
            }
            // 创建CSV写对象
            CsvWriter csvWriter = new CsvWriter(filePath, ',', Charset.forName(encoding));
            // 写表头
            if (headers != null && headers.length > 0) {
                csvWriter.writeRecord(headers);
            }
            if (dataList != null && dataList.size() > 0) {
                for (String[] content :
                        dataList) {
                    csvWriter.writeRecord(content);
                }
            }
            csvWriter.close();
        } catch (Exception e) {
            log.error("CSV文件写入失败,异常:", e);
        }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值