java TXT写入,每个文件允许1W条数据。超过1W条数据分割成多个文件

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

public class DataWriter {
    public void writeData(List<String> dataList, String fileName) {

        try {
            FileWriter writer = new FileWriter(fileName, true); // 打开文件，参数"true"表示追加写入
            int count = 0;

            for (String data : dataList) {
                count++;

                // 如果超过1万条数据，关闭当前文件，创建新的文件
                if (count > 10000) {
                    writer.close();
                    // 根据当前文件名，生成下一个文件名
                    fileName = getNextFileName(fileName);
                    writer = new FileWriter(fileName, true);
                    count = 1;
                }

                writer.write(data + "\n"); // 写入数据，并换行
            }

            writer.close(); // 关闭文件

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

    private String getNextFileName(String currentFileName) {
        // 解析当前文件名，获取文件名的数字部分
        String number = currentFileName.substring(0, currentFileName.indexOf('.'));
        int nextNumber = Integer.parseInt(number) + 1;

        // 生成下一个文件名
        return nextNumber + ".txt";
    }
}

import java.util.ArrayList;
import java.util.List;

public class Main {
    public static void main(String[] args) {
        DataWriter writer = new DataWriter();
        List<String> dataList = new ArrayList<>();
        dataList.add("data1");
        dataList.add("data2");
        dataList.add("data3");

        writer.writeData(dataList, "1.txt");
    }
}