Design Pattern | Memento Pattern (메멘토 패턴)

package com.devkuma.designpattern.behavioral.memento.game;

import java.util.ArrayList;

public class Memento {

    int money;
    ArrayList<String> fruits;

    public int getMoney() {
        return money;
    }

    Memento(int money) {
        this.money = money;
        this.fruits = new ArrayList();
    }

    void addFruit(String fruit) {
        fruits.add(fruit);
    }

    ArrayList<String> getFruits() {
        return (ArrayList<String>) fruits.clone();
    }
}

package com.devkuma.designpattern.behavioral.memento.game;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.Random;

public class Gamer {

    private int money;
    private ArrayList<String> fruits = new ArrayList();
    private Random random = new Random();
    private static String[] fruitNames = {
            "사과", "포도", "바나나", "귤",
    };

    public Gamer(int money) {
        this.money = money;
    }

    public int getMoney() {
        return money;
    }

    // 베팅...게임 진행한다.
    public void bet() {
        // 주사위를 던진다.
        int dice = random.nextInt(6) + 1;

        if (dice == 1) {
            // 1인 경우, 소지금 증가한다.
            money += 100;
            System.out.println("소지금이 증가하였습니다.");
        } else if (dice == 2) {
            // 2인 경우, 소지금이 절반이 된다.
            money /= 2;
            System.out.println("소지금이 절반이 되었습니다.");
        } else if (dice == 6) {
            // 6인 경우, 과일을 받는다.
            String fruit = getFruit();
            System.out.println("과일(" + fruit + ")을 받았습니다.");
            fruits.add(fruit);
        } else {
            // 그외인 경우, 아무 일도 일어나지 않는다.
            System.out.println("아무 일도 일어나지 않았습니다.");
        }
    }

    // 스냅샷 생성: 현재 상황을 저장한 객체를 생성하고 반환한다.
    public Memento createMemento() {
        Memento memento = new Memento(money);
        Iterator it = fruits.iterator();
        while (it.hasNext()) {
            String fruit = (String) it.next();
            if (fruit.startsWith("맛있는")) {
                // 맛있는 과일만 저장한다.
                memento.addFruit(fruit);
            }
        }
        return memento;
    }

    // 실행 취소(Undo)을 실행: 저장했던 객체를 전달받아 이전 상태로 되돌린다.
    public void restoreMemento(Memento memento) {
        this.money = memento.money;
        this.fruits = memento.getFruits();
    }

    public String toString() {
        return "[money = " + money + ", fruits = " + fruits + "]";
    }

    private String getFruit() {
        String prefix = "";
        if (random.nextBoolean()) {
            prefix = "맛있는 ";
        }
        return prefix + fruitNames[random.nextInt(fruitNames.length)];
    }
}

package com.devkuma.designpattern.behavioral.memento;

import com.devkuma.designpattern.behavioral.memento.game.Gamer;
import com.devkuma.designpattern.behavioral.memento.game.Memento;

public class Main {

    public static void main(String[] args) {
        // 첫 소지금은 100이다.
        Gamer gamer = new Gamer(100);
        // 첫번째 상태를 저장한다.
        Memento memento = gamer.createMemento();

        for (int i = 0; i < 10; i++) {
            System.out.println("==== " + i);
            System.out.println("현재 상태:" + gamer);

            // 게임을 진행한다.
            gamer.bet();

            System.out.println("소지금은 " + gamer.getMoney() + "원이 되었습니다.");

            if (gamer.getMoney() > memento.getMoney()) {
                System.out.println("    (많이 증가했으므로 현재 상태를 저장하자)");
                memento = gamer.createMemento();
            } else if (gamer.getMoney() < memento.getMoney() / 2) {
                System.out.println("    (많이 줄어들었으므로 이전 상태로 복원하자)");
                gamer.restoreMemento(memento);
            }
        }
    }
}

==== 0
현재 상태:[money = 100, fruits = []]
소지금이 증가하였습니다.
소지금은 200원이 되었습니다.
    (많이 증가했으므로 현재 상태를 저장하자)
==== 1
현재 상태:[money = 200, fruits = []]
소지금이 절반이 되었습니다.
소지금은 100원이 되었습니다.
==== 2
현재 상태:[money = 100, fruits = []]
아무 일도 일어나지 않았다.
소지금은 100원이 되었습니다.
==== 3
현재 상태:[money = 100, fruits = []]
아무 일도 일어나지 않았다.
소지금은 100원이 되었습니다.
==== 4
현재 상태:[money = 100, fruits = []]
아무 일도 일어나지 않았다.
소지금은 100원이 되었습니다.
==== 5
현재 상태:[money = 100, fruits = []]
아무 일도 일어나지 않았다.
소지금은 100원이 되었습니다.
==== 6
현재 상태:[money = 100, fruits = []]
과일(맛있는 귤)을 받았습니다.
소지금은 100원이 되었습니다.
==== 7
현재 상태:[money = 100, fruits = [맛있는 귤]]
과일(맛있는 사과)을 받았습니다.
소지금은 100원이 되었습니다.
==== 8
현재 상태:[money = 100, fruits = [맛있는 귤, 맛있는 사과]]
과일(사과)을 받았습니다.
소지금은 100원이 되었습니다.
==== 9
현재 상태:[money = 100, fruits = [맛있는 귤, 맛있는 사과, 사과]]
아무 일도 일어나지 않았다.
소지금은 100원이 되었습니다.