Máy trạng thái

Phương thức SetState của Beacon sẽ giống nhau trong mọi đối tượng có các trạng thái: tháp pháo, kẻ địch, cánh cửa. Đoạn code đó có thể nằm trong một lớp riêng, máy trạng thái, giữ trạng thái hiện tại và thay đổi trạng thái đó. Mỗi đối tượng tạo một máy trạng thái và gọi máy đó, thay vì lặp lại đoạn code:

public class StateMachine
{
    public IState Current { get; private set; }

    public void ChangeState(IState next)
    {
        Current?.Exit();
        Current = next;
        Current.Enter();
    }

    public void Tick()
    {
        Current?.Tick();
    }
}

public class Beacon : MonoBehaviour
{
    public Renderer lamp;
    private StateMachine machine = new StateMachine();
    private IState blink;
    private IState steady;

    void Start()
    {
        blink = new BlinkState(lamp);
        steady = new SteadyState();
        machine.ChangeState(steady);
    }

    void Update()
    {
        if (Keyboard.current.spaceKey.wasPressedThisFrame) machine.ChangeState(machine.Current == steady ? blink : steady);
        machine.Tick();
    }
}

Lần gọi ChangeState đầu tiên không có trạng thái nào để thoát: Current là null, và ?. bỏ qua lời gọi Exit. Phần 1 đã lưu ý rằng ?. không phát hiện được giá trị null giả của Unity; điều đó áp dụng cho các đối tượng Unity đã bị hủy, còn các trạng thái là đối tượng C# thông thường, nên ?. an toàn ở đây. Current có private set, nên không script nào có thể đổi trạng thái nếu không thông qua ChangeState cùng Exit và Enter.

Thứ tự bên trong ChangeState rất quan trọng mỗi khi hai trạng thái tác động lên cùng một thứ. Exit hoàn tác những gì trạng thái cũ đã thay đổi, sau đó Enter thiết lập trạng thái mới, nên trạng thái mới có quyền quyết định cuối cùng. Nếu Enter của DarkState tắt đèn và chạy trước Exit của BlinkState, thì Exit đó sẽ bật đèn trở lại trong trạng thái tối. StateMachine cũng là một lớp thông thường: đèn hiệu, tháp pháo và kẻ địch mỗi đối tượng tạo máy trạng thái riêng bằng new và gọi Tick từ Update.