Java程序中持久化ArrayList数据：文件序列化实践

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

public class DataPersistenceUtil {

    /**
     * 将ArrayList对象保存到指定文件
     * @param arr 要保存的ArrayList实例
     * @param filename 保存数据的文件名
     */
    public static <T> void saveArrayList(ArrayList<T> arr, String filename) {
        ObjectOutputStream oos = null;
        try {
            // 创建文件输出流
            FileOutputStream fos = new FileOutputStream(filename);
            // 创建对象输出流
            oos = new ObjectOutputStream(fos);
            // 将ArrayList对象写入流
            oos.writeObject(arr);
            System.out.println("数据已成功保存到文件：" + filename);
        } catch (IOException e) {
            System.err.println("保存数据时发生IO错误：" + e.getMessage());
            e.printStackTrace();
        } finally {
            // 确保流被关闭
            if (oos != null) {
                try {
                    oos.close();
                } catch (IOException e) {
                    System.err.println("关闭ObjectOutputStream时发生错误：" + e.getMessage());
                    e.printStackTrace();
                }
            }
        }
    }

    /**
     * 从指定文件加载ArrayList对象
     * @param filename 包含ArrayList数据的文件名
     * @return 加载的ArrayList对象，如果加载失败则返回null
     */
    @SuppressWarnings("unchecked") // 抑制未经检查的类型转换警告
    public static <T> ArrayList<T> loadArrayList(String filename) {
        ObjectInputStream ois = null;
        try {
            // 创建文件输入流
            FileInputStream fis = new FileInputStream(filename);
            // 创建对象输入流
            ois = new ObjectInputStream(fis);
            // 从流中读取对象并进行类型转换
            ArrayList<T> arr = (ArrayList<T>) ois.readObject();
            System.out.println("数据已成功从文件加载：" + filename);
            return arr;
        } catch (FileNotFoundException e) {
            System.out.println("数据文件未找到，将创建新的ArrayList。");
            return new ArrayList<>(); // 文件不存在时，返回一个新的空列表
        } catch (IOException e) {
            System.err.println("加载数据时发生IO错误：" + e.getMessage());
            e.printStackTrace();
            return null;
        } catch (ClassNotFoundException e) {
            System.err.println("加载数据时类未找到：" + e.getMessage());
            e.printStackTrace();
            return null;
        } finally {
            // 确保流被关闭
            if (ois != null) {
                try {
                    ois.close();
                } catch (IOException e) {
                    System.err.println("关闭ObjectInputStream时发生错误：" + e.getMessage());
                    e.printStackTrace();
                }
            }
        }
    }
}

import java.io.Serializable;

// User类必须实现Serializable接口才能被序列化
public class User implements Serializable {
    private static final long serialVersionUID = 1L; // 建议添加serialVersionUID

    public int ID;
    public String name;

    public User(int ID, String name) {
        this.ID = ID;
        this.name = name;
    }

    @Override
    public String toString() {
        return "User [ID=" + ID + ", name=" + name + "]";
    }
}

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

public class UserManagementApp {

    // 定义用于存储用户数据的列表，使用ArrayList<User>
    private static List<User> userList = new ArrayList<>();
    // 定义保存数据的文件名
    private static final String DATA_FILE = "users.dat";

    public static void main(String[] args) {
        // 1. 程序启动时，尝试从文件加载之前保存的用户数据
        ArrayList<User> loadedUsers = DataPersistenceUtil.loadArrayList(DATA_FILE);
        if (loadedUsers != null) {
            userList = loadedUsers; // 如果成功加载，则使用加载的数据
        } else {
            System.out.println("未能加载现有用户数据，将从空列表开始。");
        }

        System.out.println("当前用户列表：" + userList);

        int tempID = 5000;
        if (args.length > 0) {
            try {
                tempID = Integer.parseInt(args[0]);
            } catch (NumberFormatException e) {
                System.err.println("无效的程序ID参数，使用默认值5000。");
            }
        }
        System.out.println("登录 ID: " + tempID);

        Scanner scanner = new Scanner(System.in);
        System.out.print("请输入您的姓名：");
        String tempName = scanner.nextLine();

        // 创建新的User对象
        User newUser = new User(tempID, tempName);

        // 将新用户添加到列表中
        userList.add(newUser);
        System.out.println("新用户已添加：" + newUser);
        System.out.println("更新后的用户列表：" + userList);

        // 2. 程序退出前（或在适当的时机），将当前的用户列表保存到文件
        DataPersistenceUtil.saveArrayList((ArrayList<User>) userList, DATA_FILE);

        scanner.close();
        System.out.println("程序结束。");
    }
}