ObjectInputStream （2）

package Aug14.IO;

import java.io.*;
import java.util.*;
import java.util.Map.Entry;

public class TestCurrentAccount {

    private void objectWriter(Map<Integer, CurrentAccount> m, File f) {
        try {
            // File f = new File("D:\\Account.txt");
            ObjectOutputStream oos = new ObjectOutputStream(
                    new FileOutputStream(f));
            oos.writeObject(m);
            oos.flush();
            oos.close();
            System.out.println("File Writing Success");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    @SuppressWarnings("unchecked")
    private void objectReader(File f) {
        try {

            ObjectInputStream ois = new ObjectInputStream(
                    new FileInputStream(f));

            Map<Integer, CurrentAccount> m = (Map<Integer, CurrentAccount>) ois
                    .readObject();

            Set view = m.entrySet();
            Iterator it = view.iterator();
            while (it.hasNext()) {
                Entry<Integer, CurrentAccount> ca = (Entry<Integer, CurrentAccount>) it
                        .next();
                CurrentAccount cur = ca.getValue();
                System.out.println("Key:- " + ca.getKey() + "\tValues:- "
                        + cur.getAccountId() + "  " + cur.getName() + "  "
                        + cur.getBalanceAmount());

            }
            ois.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
        System.out.println("File Reading Success");
    }

    @SuppressWarnings("unchecked")
    public static void main(String[] args) {
        Scanner reader = new Scanner(System.in);
        CurrentAccount cu;
        // Map map=new HashMap();
        Map<Integer, CurrentAccount> map = new HashMap<Integer, CurrentAccount>();
        for (int i = 0; i < 1; i++) {
            cu = new CurrentAccount();

            System.out.println("Enter AccountId");
            cu.setAccountId(reader.nextInt());
            System.out.println("Enter name");
            cu.setName(reader.next());
            System.out.println("Enter balanceAmount");
            cu.setBalanceAmount(reader.nextDouble());
            map.put(10000 + i, cu);
        }

        String answer = "No";

        while (answer.equals("No") || !answer.equals("Yes")) {
            System.out.println("Enter the path to store the data:");
            // File f = new File("D:\\Account.txt");
            String path = reader.next();
            File file = new File(path);

            if (file.exists()) {
                System.out
                        .println("The file already exists ,do you want it being overwritten?(Yes/No)");
                answer = reader.next();

                if (answer.equals("Yes")) {
                    TestCurrentAccount ta = new TestCurrentAccount();
                    ta.objectWriter(map, file);
                    ta.objectReader(file);
                    break;
                } else if (answer.equals("No")) {
                    System.out
                            .println("Enter the new  path to store the data:");

                } else {
                    System.out
                            .println("Please enter the right answer,try again:");
                }
            } else {
                TestCurrentAccount ta = new TestCurrentAccount();
                ta.objectWriter(map, file);
                ta.objectReader(file);
                break;

            }

        }
    }
}

class CurrentAccount implements Serializable {
    int AccountId;
    String name;
    double balanceAmount;

    public int getAccountId() {
        return AccountId;
    }

    public void setAccountId(int accountId) {
        AccountId = accountId;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public double getBalanceAmount() {
        return balanceAmount;
    }

    public void setBalanceAmount(double balanceAmount) {
        this.balanceAmount = balanceAmount;
    }

}