本文旨在解决Java程序中，如何将控制台输入添加到ArrayList，并在程序重启后保持数据持久化的问题。通过使用`java.io.ObjectInputStream`和`java.io.ObjectOutputStream`，我们可以将ArrayList对象序列化到本地文件，并在程序启动时反序列化，从而实现数据的持久化存储。
在Java程序中，如果需要将控制台输入的数据存储到ArrayList中，并且希望在程序下次运行时能够恢复这些数据，就需要实现数据的持久化。每次程序启动时都重新初始化ArrayList会导致之前输入的数据丢失。本文将介绍如何使用Java的序列化机制来实现这一目标。
序列化与反序列化
Java提供了对象序列化的机制，可以将对象转换为字节流，并存储到文件中。反序列化则是将字节流转换回对象的过程。通过这种方式，我们可以将ArrayList对象保存到磁盘，并在程序启动时重新加载。
立即学习“Java免费学习笔记（深入）”；
实现步骤
- 创建保存和加载ArrayList的方法
首先，我们需要创建两个方法，分别用于保存ArrayList到文件和从文件加载ArrayList。
import java.io.*;
import java.util.ArrayList;
public class ArrayListPersistence {
public static ArrayList<Object> loadArrayList(String filename) {
ObjectInputStream ois = null;
try {
ois = new ObjectInputStream(new FileInputStream(filename));
ArrayList<Object> arr = (ArrayList<Object>) ois.readObject();
return arr;
} catch (IOException e) {
System.err.println("Error loading ArrayList: " + e.getMessage());
return new ArrayList<>(); // Return a new empty ArrayList in case of error
} catch (ClassNotFoundException e) {
System.err.println("Class not found: " + e.getMessage());
return new ArrayList<>(); // Return a new empty ArrayList in case of error
} finally {
if (ois != null) {
try {
ois.close();
} catch (IOException e) {
System.err.println("Error closing ObjectInputStream: " + e.getMessage());
}
}
}
}
public static void saveArrayList(ArrayList<Object> arr, String filename) {
ObjectOutputStream oos = null;
try {
oos = new ObjectOutputStream(new FileOutputStream(filename));
oos.writeObject(arr);
} catch (IOException e) {
System.err.println("Error saving ArrayList: " + e.getMessage());
} finally {
if (oos != null) {
try {
oos.close();
} catch (IOException e) {
System.err.println("Error closing ObjectOutputStream: " + e.getMessage());
}
}
}
}
}这段代码定义了两个静态方法：loadArrayList和saveArrayList。loadArrayList方法尝试从指定文件中读取ArrayList对象，如果文件不存在或发生其他异常，则返回一个新的空ArrayList。saveArrayList方法将ArrayList对象写入到指定文件中。注意，这里使用了try-catch-finally块来确保资源得到正确关闭，即使发生异常。
- 修改User类
接下来，我们需要修改User类，使其能够使用上述方法来加载和保存数据。同时需要确保User类实现了Serializable接口。
import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
public class User implements Serializable {
private static final long serialVersionUID = 1L; // Recommended for Serializable classes
private static List<String> listNames = new ArrayList<>();
private static List<Integer> listIds = new ArrayList<>();
private int ID;
private String name;
private static final String DATA_FILE = "user_data.ser"; // File to store the data
public static void main(String[] args) {
// Load existing data from file
List<?> loadedNames = (List<?>) ArrayListPersistence.loadArrayList(DATA_FILE + "_names");
List<?> loadedIds = (List<?>) ArrayListPersistence.loadArrayList(DATA_FILE + "_ids");
if (loadedNames != null && !loadedNames.isEmpty()) {
listNames = new ArrayList<>(loadedNames.stream().map(Object::toString).toList());
}
if (loadedIds != null && !loadedIds.isEmpty()) {
listIds = new ArrayList<>(loadedIds.stream().map(o -> (Integer) o).toList());
}
int tempID = 5000;
if (args.length > 0) tempID = Integer.parseInt(args[0]);
System.out.println("Login " + tempID);
Scanner scanner = new Scanner(System.in);
System.out.print("Enter your Name : ");
String tempName = scanner.nextLine();
User n = new User();
n.ID = tempID;
n.name = tempName;
listIds.add(n.ID);
listNames.add(n.name);
// Save the updated lists to file
ArrayListPersistence.saveArrayList(new ArrayList<>(listNames), DATA_FILE + "_names");
ArrayListPersistence.saveArrayList(new ArrayList<>(listIds), DATA_FILE + "_ids");
System.out.println("Name List: " + listNames);
System.out.println("ID List: " + listIds);
scanner.close();
}
}在这个修改后的User类中，我们首先定义了一个DATA_FILE常量，用于指定保存数据的文件名。在main方法中，我们首先尝试从文件中加载已有的数据，如果文件存在，则将数据加载到listNames和listIds中。然后，我们获取用户的输入，并将新的数据添加到列表中。最后，我们将更新后的列表保存到文件中。serialVersionUID是用于序列化版本控制的，建议为每个可序列化的类都定义一个。
注意事项
- 确保要序列化的类及其所有成员变量都实现了Serializable接口。
- 如果某些成员变量不希望被序列化，可以使用transient关键字修饰。
- 在反序列化时，需要确保类的定义与序列化时的定义一致，否则可能会出现InvalidClassException。
- 异常处理非常重要，确保在文件操作过程中能够捕获并处理可能出现的异常，例如FileNotFoundException、IOException和ClassNotFoundException。
总结
通过使用Java的序列化机制，我们可以很方便地将ArrayList对象保存到文件，并在程序启动时重新加载。这为实现数据的持久化提供了一种简单而有效的方法。在实际应用中，可以根据需要选择不同的存储方式，例如数据库、XML文件等。序列化适用于简单的数据存储需求，对于复杂的数据关系和高性能要求，数据库可能更适合。
