java 生成多个CSV文件

最新推荐文章于 2023-03-25 23:10:36 发布

idiot_qi

最新推荐文章于 2023-03-25 23:10:36 发布

阅读量759

点赞数

CC 4.0 BY-SA版权

分类专栏： # JAVA基础文章标签： java csv

本文链接：https://blog.youkuaiyun.com/idiot_qi/article/details/113700017

JAVA基础专栏收录该内容

3 篇文章

订阅专栏

这篇博客详细介绍了如何使用Java代码生成CSV文件，包括设置文件路径、列头、数据，并处理中文乱码问题。此外，还提供了一个批量删除文件的方法。对于需要处理CSV文件的开发者来说，这是一个实用的参考。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

存一波生成csv文件的代码，以便后续Ctrl C ,Ctrl V

参考博文：https://blog.youkuaiyun.com/wangchaox123/article/details/93646324

/**
     * <生成单个csv文件>
     *
     * @param filePath    文件路径
     * @param fileName    文件名
     * @param colNameList 列头集合
     * @param dataList    数据集合
     * @return File
     * @throws
     */
    public static File generateCsv(String filePath, String fileName,
                                   List<String> colNameList, List<Object[]> dataList)
            throws IOException {
        BufferedWriter csvWrite = null;
        String fileRealPath = filePath + fileName + ".csv" ;
        try {
            //定义文件类型
            File csvFile = new File(fileRealPath);
            //获取文件目录
            File parent = csvFile.getParentFile();
            if (!parent.exists()) {
                parent.mkdirs();
            }
            //创建文件
            csvFile.createNewFile();
            csvWrite = new BufferedWriter(
                    new OutputStreamWriter(new FileOutputStream(csvFile), "UTF-8"), 1024);

            //处理用Excel打开出现中文乱码
            csvWrite.write('\ufeff');
            //写入表头
            write(colNameList, csvWrite);
            //写入数据
            for (Object[] dataPerRow : dataList) {
                //转换格式
                List<?> dataPerRowList = new ArrayList<>(Arrays.asList(dataPerRow));
                write(dataPerRowList, csvWrite);
            }
            csvWrite.flush();
            return csvFile;
        } catch (IOException e) {
            logger.error("csv文件生成失败，原因：", e);
            throw new IOException("csv文件生成失败");
        } finally {
            try {
                if (null != csvWrite) {
                    csvWrite.close();
                }
            } catch (IOException e) {
                logger.error("关闭文件流失败，原因：", e);
                throw new IOException("关闭文件流失败");
            }
        }
    }

    /**
     * @param dataList
     * @param csvWrite
     * @return
     * @Title write
     * @Description: 将数据按行写入数据
     */
    private static void write(List<?> dataList, BufferedWriter csvWrite)
            throws IOException {
        for (Object data : dataList) {
            StringBuffer buffer = new StringBuffer();
            buffer.append("\"");
            if (null == data) {
                buffer.append(" ");
            } else {
                if (data instanceof Number) {
                    buffer.append(data);
                } else if (data instanceof Date) {
                    buffer.append(
                            new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format((Date) data));
                } else {
                    buffer.append(data);
                }
            }
            String rowStr = buffer.append("\",").toString();
            csvWrite.write(rowStr);
        }
        csvWrite.newLine();
    }


    /**
     * 批量删除文件
     *
     * @param fileRealPathList 文件绝对路径（带文件名）
     */
    public static void deleteFiles(List<String> fileRealPathList) {
        if (!CollectionUtils.isEmpty(fileRealPathList)) {
            File file = null;
            for (String fileRealPath : fileRealPathList) {
                file = new File(fileRealPath);
                if (file.exists() && file.isFile()) {
                    file.delete();
                }
            }
        }
    }