设计模式-状态模式（四）

public class SoldState implements State {
    private VendingMachine vendingMachine;

    public SoldState(VendingMachine vendingMachine) {
        this.vendingMachine = vendingMachine;
    }

    @Override
    public void insertCoin() {
        System.out.println("Please wait, we're already giving you a product.");
    }

    @Override
    public void pressButton() {
        System.out.println("You have not inserted a coin yet.");
    }

    @Override
    public void dispense() {
        if (vendingMachine.getCount() > 0) {
            vendingMachine.setCount(vendingMachine.getCount() - 1);
            System.out.println("Product dispensed.");
            if (vendingMachine.getCount() == 0) {
                vendingMachine.setCurrentState(new NoStockState(vendingMachine));
            } else {
                vendingMachine.setCurrentState(new IdleState(vendingMachine));
            }
        } else {
            vendingMachine.setCurrentState(new NoStockState(vendingMachine));
        }
    }
}