[Unity] Coroutine 의 yield return 뒤 함수 확장

1. IEnumerator 구현

 소스코드

using System.Collections;
using UnityEngine;
public class UseIEnumerator : IEnumerator {
    public object Current
    {
        get
        {
            return null;
        }
    }
 
 
    public bool MoveNext()
    {
        return !Input.GetKey(KeyCode.Space);
 
    }
 
    public void Reset()
    {
    }
 
}
 

2. CustomYieldInstruction 구현

 소스코드

using UnityEngine;
 
public class UseCustomYieldInstruction : CustomYieldInstruction
{
    public override bool keepWaiting
    {
        get
        {
            return !Input.GetKey(KeyCode.Space);
        }
    }
    
}
 

using System.Collections;
using UnityEngine;
 
public class Main : MonoBehaviour {
 
    // Use this for initialization
    void Start () {
 
        StartCoroutine(CoroutineTest());
    }
    
    IEnumerator CoroutineTest()
    {
        while(true)
        {
            yield return new UseCustomYieldInstruction();
            Debug.Log("Running");
        }
    }
 
 
}
 

3. WaitUntil

 소스코드

using System.Collections;
using UnityEngine;
 
public class Main : MonoBehaviour {
 
    // Use this for initialization
    void Start () {
 
        StartCoroutine(CoroutineTest());
    }
    
    IEnumerator CoroutineTest()
    {
        while(true)
        {
            yield return new WaitUntil(() => Input.GetKey(KeyCode.Space));
            Debug.Log("Running");
        }
    }
}
 

지금처럼 단순히 스페이스바를 눌렀을 때 멈춘다 정도의 기능을 원한다면 WaitUntil을 사용하는것이 제일 단순하고 깔끔해보인다. 

위 코드에서는 yield return 뒤에 new 생성자를 통해 인스턴스화 하고 있는데 미리 캐싱해놓는게 실제 사용에서는 바람직할 것 같다.  

IEnumerator > CustomYieldInstruction > WaitUntil 순으로 자유도가 높다.