程序重启后覆盖写入的CSV数据丢失问题排查与修复请求

问题原因 #

• 文件流未被可靠关闭：虽然调用了flush()，但它仅将JVM缓冲区的数据刷入操作系统缓冲区，操作系统可能不会立即写入磁盘。只有调用close()才能强制操作系统把数据持久化到磁盘并释放资源。如果程序退出时没调用shutDown()，FileWriter缓冲区的未写入数据会直接丢失。
• 构造函数异常处理缺陷：创建FileWriter时若抛出IO异常，fw会保持null状态，后续调用writeToCSV()或shutDown()会触发空指针异常，导致数据根本没写入文件。
• 异常处理过于简陋：仅打印异常信息而不中断流程或通知调用方，可能让调用者误以为数据已成功写入。

修复方案 #

1. 用try-with-resources自动管理资源 #

Java的try-with-resources语法会自动关闭实现了AutoCloseable接口的资源，无论是否发生异常，都能确保文件流被正确关闭，彻底避免因未关闭流导致的数据丢失。

2. 优化异常处理逻辑 #

在构造函数或方法中抛出受检异常，让调用方明确处理异常，避免因null指针或静默异常导致的数据写入失败。

3. 优化字符串拼接性能 #

使用StringBuilder替代字符串累加，尤其是处理大量数据时，能显著提升性能并减少内存开销。

修复后的代码示例：

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

public class WriteFile { // 类名遵循Java大驼峰命名规范

    private final Writer fw;

    // 构造函数抛出异常，将异常处理交给调用方
    public WriteFile(String fileName) throws IOException {
        fw = new FileWriter(fileName, false);
    }

    public void writeToCSV(String[] org) throws IOException {
        if (org == null || org.length == 0) {
            return;
        }
        StringBuilder text = new StringBuilder();
        for (int i = 0; i < org.length; i++) {
            text.append(org[i]);
            if (i != org.length - 1) {
                text.append(",");
            } else {
                text.append(System.lineSeparator()); // 使用系统原生换行符，保证跨平台兼容
            }
        }
        fw.write(text.toString());
        fw.flush(); // 可选，实时刷入可降低异常时的数据丢失风险
    }

    // 若不使用try-with-resources，需确保程序退出时调用此方法
    public void shutDown() throws IOException {
        if (fw != null) {
            fw.close();
        }
    }

    // 示例调用：用try-with-resources自动管理资源
    public static void main(String[] args) {
        try (WriteFile writer = new WriteFile("data.csv")) {
            writer.writeToCSV(new String[]{"张三", "25", "工程师"});
            // 无需手动调用shutDown，try-with-resources会自动关闭流
        } catch (IOException e) {
            e.printStackTrace(); // 可根据业务需求替换为日志记录等逻辑
        }
    }
}

额外建议 #

如果是长期运行的后台程序，可添加JVM关闭钩子，确保程序退出时强制关闭文件流：

WriteFile writer = new WriteFile("data.csv");
Runtime.getRuntime().addShutdownHook(new Thread(() -> {
    try {
        writer.shutDown();
    } catch (IOException e) {
        e.printStackTrace();
    }
}));

内容的提问来源于stack exchange，提问作者allmight