package onlyfun.caterpillar;
 
import java.io.*;
import java.util.*;
 
public class ObjectStreamDemo {
    public static void main(String[] args) {
        User[] users = {new User("cater", 101),
                        new User("justin", 102)}; 
        // 寫入新文件
        writeObjectsToFile(users, args[0]);

        try {
            // 讀取文件數據
            users = readObjectsFromFile(args[0]);
            // 顯示讀回的對象

            for(User user : users) {
                System.out.printf("%s\t%d%n", user.getName(), user.getNumber());
            }
            System.out.println();
            
            users = new User[2];
            users[0] = new User("momor", 103);
            users[1] = new User("becky", 104);
            
           // 附加新對象至文件
            appendObjectsToFile(users, args[0]);
            
            // 讀取文件數據
            users = readObjectsFromFile(args[0]);
            // 顯示讀回的對象

            for(User user : users) {
                System.out.printf("%s\t%d%n", user.getName(), user.getNumber());
            }
        }
        catch(ArrayIndexOutOfBoundsException e) {
            System.out.println("沒有指定文件名");
        }
        catch(FileNotFoundException e) {
            e.printStackTrace();
        }
    }

    //將指定的對象寫入至指定的文件
    public static void writeObjectsToFile(
                         Object[] objs, String filename) { 
        File file = new File(filename);
 
        try { 
            ObjectOutputStream objOutputStream = 
                new ObjectOutputStream(
                      new FileOutputStream(file)); 
            for(Object obj : objs) {
                // 將對象寫入文件
                objOutputStream.writeObject(obj); 
            }
            // 關閉流

            objOutputStream.close(); 
        } 
        catch(IOException e) { 
            e.printStackTrace(); 
        }
    }
    
     // 將指定文件中的對象數據讀回
    public static User[] readObjectsFromFile(
                             String filename) 
                               throws FileNotFoundException {
        File file = new File(filename); 
 
        // 如果文件不存在就丟出異常
        if(!file.exists()) 
            throw new FileNotFoundException(); 
 
        // 使用List先存儲讀回的對象

        List<User> list = new ArrayList<User>();
        
        try {
            FileInputStream fileInputStream = 
                new FileInputStream(file);
            ObjectInputStream objInputStream = 
                new ObjectInputStream(fileInputStream); 
            
            while(fileInputStream.available() > 0) {
                list.add((User) objInputStream.readObject());
            }
            objInputStream.close(); 
        } 
        catch(ClassNotFoundException e) { 
            e.printStackTrace(); 
        } 
        catch(IOException e) { 
            e.printStackTrace(); 
        }
 
        User[] users = new User[list.size()];
        return list.toArray(users);
    }
 
    // 將對象附加至指定的文件之后
    public static void appendObjectsToFile(
                           Object[] objs, String filename) 
                               throws FileNotFoundException {
  
        File file = new File(filename); 
 
        // 如果文件不存在則丟出異常
        if(!file.exists()) 
             throw new FileNotFoundException(); 

        try {
             // 附加模式
            ObjectOutputStream objOutputStream = 
               new ObjectOutputStream(
                  new FileOutputStream(file, true)) { 
                    // 如果要附加對象至文件后
                    // 必須重新定義這個方法
                    protected void writeStreamHeader() 
                                     throws IOException {} 
               };  
 
            for(Object obj : objs) {
                // 將對象寫入文件

                objOutputStream.writeObject(obj); 
            }
            objOutputStream.close(); 
        } 
        catch(IOException e) { 
            e.printStackTrace(); 
        } 
    }
}