직렬화 연습

커피메뉴를 파일에 쓰기

SerialMenuWrite

package 직렬화메뉴쓰기;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.ObjectOutputStream;
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;

public class SerialMenuWrite {
    // 키와 값으로 구성되는 Map 컬렉션
    static Map<String, CoffeeMenuInfo> map = new HashMap<>();
    public static void main(String[] args) throws IOException {
        menuWrite();
        selectMenu();
    }
    static void menuWrite() {
        map.put("Americano",
                new CoffeeMenuInfo("Americano", 2500, "Coffee", "그냥 커피"));
        map.put("Espresso",
                new CoffeeMenuInfo("Espresso", 2500, "Coffee", "진한 커피"));
        map.put("Latte",
                new CoffeeMenuInfo("Latte", 2500, "Coffee", "진한 커피"));
    }
    static void selectMenu() throws IOException {
        Scanner sc = new Scanner(System.in);
        while(true) {
            System.out.println("메뉴를 선택 하세요 : ");
            System.out.print("[1]메뉴 보기, [2]메뉴 추가, [3]종료 : ");
            int inputMenu = sc.nextInt();
            switch(inputMenu) {
                case 1 :
                    System.out.println("========= 메뉴 보기 =========");
                    // map에서 key 가져와서 향상된 for문 수행
                    for(String e : map.keySet()) {
                        System.out.println("메뉴 : " + map.get(e).getName());
                        System.out.println("메뉴 : " + map.get(e).getPrice());
                        System.out.println("분류 : " + map.get(e).getGroup());
                        System.out.println("설명 : " + map.get(e).getDesc());
                        System.out.println("---------------------------------");
                    }
                    break;
                case 2 :
                    System.out.print("추가 할 메뉴를 입력 하세요 : ");
                    String key = sc.next(); // key값이 Map의 Key이면서 메뉴의 이름
                    if(map.containsKey(key)) { // 해당 키가 맵에 존재하는 확인 (존재하면 true)
                        System.out.println("해당 메뉴가 이미 존재 합니다.");
                    } else {
                        System.out.print("가격 입력 : ");
                        int price = sc.nextInt();
                        System.out.print("분류 입력 : ");
                        String grp = sc.next();
                        System.out.print("설명 입력 : ");
                        String desc = sc.next();
                        map.put(key, new CoffeeMenuInfo(key, price, grp, desc));
                    }
                    break;
                case 3 :
                    System.out.println("메뉴를 종료 합니다. 내용을 파일에 저장 합니다.");
                    FileOutputStream fos = new FileOutputStream("C:/파일입출력tmp/커피메뉴만들기.txt");
                    ObjectOutputStream oos = new ObjectOutputStream(fos);
                    oos.writeObject(map); // 객체를 직렬화해서 파일에 저장
                    oos.flush(); // 출력했으면 버퍼를 지워야 함.
                    oos.close(); // 출력을 위한 스트림의 자원을 반납하고 닫기
                    return;
            }
        }
    }
}

CoffeeMenuInfo

package 직렬화메뉴쓰기;
import java.io.Serializable;
// 직렬화할 클래스
// transient 키워드를 사용하면 직렬화에서 제외 된다.
public class CoffeeMenuInfo implements Serializable {
    private String name; // 메뉴 이름
    private int price;   // 가격
    private String group; // 분류
    private String desc;  // 설명

    public CoffeeMenuInfo(String name, int price, String group, String desc) {
        this.name = name;
        this.price = price;
        this.group = group;
        this.desc = desc;
    }

    public String getName() {
        return name;
    }

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

    public int getPrice() {
        return price;
    }

    public void setPrice(int price) {
        this.price = price;
    }

    public String getGroup() {
        return group;
    }

    public void setGroup(String group) {
        this.group = group;
    }

    public String getDesc() {
        return desc;
    }

    public void setDesc(String desc) {
        this.desc = desc;
    }
}

커피 메뉴를 파일에서 읽기

SerialMenuRead

package 직렬화메뉴읽기;
import 직렬화메뉴쓰기.CoffeeMenuInfo;
import java.io.*;
import java.util.HashMap;
import java.util.Map;
// 역직렬화 : 바이트로 직렬화된 파일을 읽어 객체로 변환하기
public class SerialMenuRead {
    public static void main(String[] args) throws IOException, ClassNotFoundException {
        getMenuList();
    }
    static void getMenuList() throws IOException, ClassNotFoundException {
        FileInputStream fis = new FileInputStream("C:/파일입출력tmp/커피메뉴만들기.txt");
        ObjectInputStream ois = new ObjectInputStream(fis);
        Map<String, CoffeeMenuInfo> map = new HashMap<>();
        map = (Map<String, CoffeeMenuInfo>) ois.readObject();
        System.out.println("<< 역직렬화해서 메뉴를 가져 옴 >>");
        for(String e : map.keySet()) {
            // get(key) : 키로 값으로얻어 옴
            System.out.println("메뉴 : " + map.get(e).getName());
            System.out.println("가격 : " + map.get(e).getPrice());
            System.out.println("분류 : " + map.get(e).getGroup());
            System.out.println("설명 : " + map.get(e).getDesc());
            System.out.println("-------------------------------------");
        }
        fis.close();
        ois.close();
    }
}