Microsoft MVP성태의 닷넷 이야기
글쓴 사람
홈페이지
첨부 파일

C# 컴파일러 대신 직접 구현하는 비동기(async/await) 코드

C# async/await 코드가 참 마법 같습니다. 단지 예약어 하나 썼을 뿐인데 어떻게 그걸 비동기로 처리해 주는지 신기할 따름인데요. 그저 그 신기함을 누리며 사용하는 것도 좋겠지만, 왠지 C# 컴파일러가 추상화한 부분을 걷어내고 싶어졌습니다. 혹시나 C# async/await 내부 동작을 이해하고 싶은 분들이 계시다면 이 글이 도움이 될 듯합니다. (또는, 수많은 소스코드 파일을 빌드하느라 힘에 겨운 C# 컴파일러의 일을 덜어주고 싶은, 착한 마음씨를 가진 개발자분들을 위해서도! ^^)




우선 다음과 같은 예제 코드를 보겠습니다.

using System;
using System.Threading.Tasks;

namespace ConsoleApp1
{
    class Program
    {
        // C# 7.1 async Main
        static async Task Main(string[] args)
        {
            Program pg = new Program();

            await pg.CallAsync();
        }

        private async Task CallAsync()
        {
            string title = DateTime.Now.ToString();
            string text = await GetFileContents();
            Console.WriteLine(title + ": " + text);
        }

        private async Task<string> GetFileContents()
        {
            return await new TaskFactory().StartNew(() => { return "test"; });
        }
    }
}

보는 바와 같이 async/await을 사용한 전형적인 비동기 함수 호출인데요. 실행하면 다음과 같은 식의 결과가 출력됩니다.

2017-11-07 오후 9:05:15: test

자, 그럼 이 부분에서 GetFileContents 비동기 함수를 C# 컴파일러가 아닌 우리가 직접 비동기 처리로 바꿔보겠습니다. 방법은 사실 매우 쉽습니다. .NET Reflector와 같은 역어셈블러를 이용해 GetFileContents를 어떻게 바꿨는지 살펴보면 됩니다. 다음은 실제로 C# 컴파일러가 비동기 처리를 위해 만든 내부 클래스입니다.

[CompilerGenerated]
private sealed class <GetFileContents>d__2 : IAsyncStateMachine
{
    // Fields
    public int <>1__state;
    public Program <>4__this;
    private string <>s__3;
    public AsyncTaskMethodBuilder<string> <>t__builder;
    private TaskAwaiter<string> <>u__1;
    private Task<string> <getStringTask>5__1;
    private string <urlContents>5__2;

    // Methods
    private void MoveNext()
    {
        string str;
        int num = this.<>1__state;
        try
        {
            TaskAwaiter<string> awaiter;
            if (num != 0)
            {
                this.<getStringTask>5__1 = new TaskFactory().StartNew<string>(Program.<>c.<>9__2_0 ?? (Program.<>c.<>9__2_0 = new Func<string>(Program.<>c.<>9.<GetFileContents>b__2_0)));
                awaiter = this.<getStringTask>5__1.GetAwaiter();
                if (!awaiter.IsCompleted)
                {
                    this.<>1__state = num = 0;
                    this.<>u__1 = awaiter;
                    Program.<GetFileContents>d__2 stateMachine = this;
                    this.<>t__builder.AwaitUnsafeOnCompleted<TaskAwaiter<string>, Program.<GetFileContents>d__2>(ref awaiter, ref stateMachine);
                    return;
                }
            }
            else
            {
                awaiter = this.<>u__1;
                this.<>u__1 = new TaskAwaiter<string>();
                this.<>1__state = num = -1;
            }
            this.<>s__3 = awaiter.GetResult();
            this.<urlContents>5__2 = this.<>s__3;
            this.<>s__3 = null;
            str = this.<urlContents>5__2;
        }
        catch (Exception exception)
        {
            this.<>1__state = -2;
            this.<>t__builder.SetException(exception);
            return;
        }
        this.<>1__state = -2;
        this.<>t__builder.SetResult(str);
    }

    [DebuggerHidden]
    private void SetStateMachine(IAsyncStateMachine stateMachine)
    {
    }
}

위의 소스코드는 지저분하니, 다듬어서 재작성해보겠습니다.

GetFileContents 메서드는 단 한 줄로 작성되어 있지만 다음과 같이 나눠볼 수 있습니다.

private async Task<string> GetFileContents()
{
    Task<string> getStringTask = new TaskFactory().StartNew(() => { return "test"; });

    string urlContents = await getStringTask;
    return urlContents;
}

C# 컴파일러는 위와 같은 async 메서드를 만나면 다음과 같이 2가지 단계로 분할합니다.

[Part A - 현재 스레드에서 실행할 코드]

    Task<string> getStringTask = new TaskFactory().StartNew(() => { return "test"; });

[Part B - 별도의 스레드에서 실행할 코드]

    string urlContents = [getStringTask 작업의 반환값];
    return urlContents;

그리고 저 코드들을 나눠 담을 IAsyncStateMachine 인터페이스를 상속한 별도의 내부 클래스를 정의합니다.

/*

public interface IAsyncStateMachine
{
    void MoveNext();
    void SetStateMachine(IAsyncStateMachine stateMachine);
}

*/

class GetFileContents_StateMachine : IAsyncStateMachine
{
    // ... [생략]...
}

GetFileContents_StateMachine 타입에는 내부 필드를 다음과 같은 식으로 포함하고 있습니다.

// [async 동작을 위한 필드 3개]
public int _state;
public AsyncTaskMethodBuilder _builder;
TaskAwaiter<string> _awaiter;

// [async 메서드를 구현하고 있는 클래스의 this 보관 필드]
public Program _this;

// [async 메서드의 반환값을 임시 보관하는 필드]
string _result;

// [async 메서드의 반환값을 보관하는 필드]
string _urlContents;

// [Part A 코드의 변수들]
Task<string> _getStringTask;

이런 내부 필드 중에서 public 필드의 경우에는 C# 컴파일러가 async 메서드를 일반 메서드로 바꾸는 과정에서 다음과 같은 초기화를 해줍니다.

private Task<string> GetFileContents()
{
    GetFileContents_StateMachine stateMachine = new GetFileContents_StateMachine
    {
        _this = this,
        _builder = AsyncTaskMethodBuilder<string>.Create(),
        _state = -1,
    };

    stateMachine._builder.Start(ref stateMachine);
    return stateMachine._builder.Task as Task<string>;
}

보는 바와 같이 C# 컴파일러는 async 메서드를 일반 메서드로 바꾸면서 내부 동작을 GetFileContents_StateMachine 타입 내에 넣어두고는 stateMachine._builder.Start 메서드를 호출하는 걸로 마무리를 합니다.

여기서 _builder.Start는 비동기 호출이 아닙니다. 현재 스레드에서 시작하는 동기 호출에 불과합니다. AsyncTaskMethodBuilder 타입의 _builder 인스턴스는 Start 메서드 내에서 인자로 들어온 stateMachine의 MoveNext 메서드를 실행하는데, 이 때문에 Part A로 분리한 코드들은 이때 실행이 되도록 MoveNext가 구성되어 있습니다.

void IAsyncStateMachine.MoveNext()
{
    string str;
    int num = this._state;

    try
    {
        TaskAwaiter<string> awaiter;
        // 원래는 if 문이지만 명확한 분리를 위해 switch로 바꿨습니다.
        switch (num)
        {
            case 0:
                // ...[생략]...
                break;

            default:
                this._getStringTask = new TaskFactory().StartNew(() => { return "test"; });
                awaiter = this._getStringTask.GetAwaiter();
                if (awaiter.IsCompleted == false)
                {
                    this._state = num = 0;
                    this._awaiter = awaiter;
                    GetFileContents_StateMachine stateMachine = this;
                    this._builder.AwaitUnsafeOnCompleted(ref awaiter, ref stateMachine);
                    return;
                }
                break;
        }

        // ...[생략]...
    }
    catch (Exception e)
    {
        // ...[생략]...
        return;
    }

    // ...[생략]...
}

C# 컴파일러가 GetFileContents 메서드를 async에서 일반 메서드로 변경하는 중에 _state 필드를 -1로 초기화했기 때문에 위의 MoveNext 메서드는 동기적으로 default 영역의 코드를 실행하게 됩니다.

보는 바와 같이 Part A로 분리되었던 영역의 코드가 default 영역에 추가되어 있고 await 코드의 대상이었던 _getStringTask에 대해 GetAwaiter()를 호출해 TaskAwaiter를 보관한 다음 해당 Task가 금방 끝나서 IsCompleted == true가 되면 더 이상 동작을 하지 않고 MoveNext를 반환합니다. 즉, 이런 경우에는 단순히 동기 메서드 호출한 것과 다를 바가 없습니다.

반면, 대개의 경우 awaiter.IsCompleted == false로 나오는데, 그럴 때는 다시 _builder.AwaitUnsafeOnCompleted 메서드를 호출해 작업이 완료된 경우의 알림을 등록합니다. 결국 stateMachine 객체의 MoveNext를 다시 호출하도록 만들고는 동기 호출을 마무리합니다. 그리고 이때의 _state 값은 0으로 설정했기 때문에 다음 번 MoveNext가 호출될 때는 switch의 case 0 영역의 코드가 실행됩니다.

그래서 Task의 작업이 완료되었을 때 실행되는 MoveNext의 _state == 0으로 실행할 코드에는 C# 컴파일러가 분리한 Part B 영역의 코드를 포함하게 됩니다.

void IAsyncStateMachine.MoveNext()
{
    string str;
    int num = this._state;

    try
    {
        TaskAwaiter<string> awaiter;
        switch (num)
        {
            case 0: // 작업 완료 후 실행되는 코드
                awaiter = this._awaiter;
                this._awaiter = new TaskAwaiter<string>();
                this._state = num = -1;
                break;

            default:
                // ...[생략]...
                return;
        }

        this._result = awaiter.GetResult();
        this._urlContents = this._result; // Part B의 코드
        this._result = null;

        str = this._urlContents;
    }
    catch (Exception e)
    {
        this._state = -2;
        this._builder.SetException(e);
        return;
    }

    this._state = -2;
    this._builder.SetResult(str);
}

이렇게 만들어두고 실행해 보면, C# 컴파일러가 제공했던 async/await 치환 코드와 정확히 동일한 결과를 얻게 됩니다.




하는 김에 CallAsync 메서드도 바꿔 볼까요? ^^

이것도 메서드를 다음과 같이 2부분으로 나눌 수 있습니다.

[Part A - 현재 스레드에서 실행할 코드]

    string title = DateTime.Now.ToString();
    [Task 객체 반환 = ]GetFileContents();

[Part B - 별도의 스레드에서 실행할 코드]

    string text = [GetFileContents 메서드의 반환값];
    Console.WriteLine(title + ": " + text);

마찬가지로 Part A, B 영역의 처리에 필요한 변수와 StateMachine 구현을 위한 필드를 가진 타입을 정의하고,

class CallAsync_StateMachine : IAsyncStateMachine
{
    public int _state;
    public AsyncTaskMethodBuilder _builder;
    TaskAwaiter<string> _awaiter;

    public Program _this;

    string _result;

    // 반환값
    string _text;

    // Part A 코드의 지역 변수
    string _title;

    //...[생략]...
}

마지막으로 Part A, B의 코드를 나눠서 실행할 MoveNext 메서드를 다음과 같이 구현해 주면 끝입니다.

void IAsyncStateMachine.MoveNext()
{
    int num = this._state;

    try
    {
        TaskAwaiter<string> awaiter;
        switch (num)
        {
            case 0: // 작업 완료 후 Task 스레드에서 실행
                awaiter = this._awaiter;
                this._awaiter = new TaskAwaiter<string>();
                this._state = num = -1;
                break;

            default: // 동기적으로 실행될 코드 - Part A 코드를 포함
                this._title = DateTime.Now.ToString();

                awaiter = _this.GetFileContents().GetAwaiter();
                if (awaiter.IsCompleted == false)
                {
                    this._state = num = 0;
                    this._awaiter = awaiter;
                    CallAsync_StateMachine stateMachine = this;
                    this._builder.AwaitUnsafeOnCompleted(ref awaiter, ref stateMachine);
                    return;
                }
                break;
        }

        // 작업 완료 후 Task 스레드에서 실행 - Part B 코드를 포함
        this._result = awaiter.GetResult();
        this._text = this._result;
        this._result = null;
        Console.WriteLine(this._title + ": " + this._text);
    }
    catch (Exception e)
    {
        this._state = -2;
        this._builder.SetException(e);
        return;
    }

    this._state = -2;
    this._builder.SetResult();
}

이것으로 완벽하게 C# 컴파일러가 대신 만들어 주었던 코드와 일치하므로 실행해 보면 정상적으로 다음과 같은 식의 결과를 볼 수 있습니다.

2017-11-07 오후 9:05:15: test

이렇게 풀어놓고 보니까... C# async/await 코드가 그다지 신기하지 않게 보입니다. ^^

(첨부 파일은 이 글의 모든 예제 코드를 포함합니다.)




[이 글에 대해서 여러분들과 의견을 공유하고 싶습니다. 틀리거나 미흡한 부분 또는 의문 사항이 있으시면 언제든 댓글 남겨주십시오.]

[연관 글]





[최초 등록일: ]
[최종 수정일: 12/22/2017 ]

Creative Commons License
이 저작물은 크리에이티브 커먼즈 코리아 저작자표시-비영리-변경금지 2.0 대한민국 라이센스에 따라 이용하실 수 있습니다.
by SeongTae Jeong, mailto:techsharer@outlook.com

비밀번호

댓글 쓴 사람
 




... 31  32  33  34  [35]  36  37  38  39  40  41  42  43  44  45  ...
NoWriterDateCnt.TitleFile(s)
11410정성태12/20/201711190.NET Framework: 714. SSL Socket 예제 - C/C++ 서버, C# 클라이언트파일 다운로드1
11409정성태6/6/201914512VC++: 122. 오픈 소스 라이브러리를 쉽게 빌드해 주는 "C++ Package Manager for Windows: vcpkg" [6]
11408정성태12/18/20173939.NET Framework: 713. C# - SharpDX + DXGI를 이용한 윈도우 화면 캡처 소스 코드 + Direct2D 출력 + OpenCV (2)파일 다운로드1
11407정성태12/18/20175212.NET Framework: 712. C# - SharpDX + DXGI를 이용한 윈도우 화면 캡처 소스 코드 + Direct2D 출력 + OpenCV [1]파일 다운로드1
11406정성태12/17/201715232.NET Framework: 711. C# - OpenCvSharp의 Mat 데이터 조작 방법 [5]파일 다운로드1
11405정성태12/17/20179893.NET Framework: 710. C# - OpenCvSharp을 이용한 Webcam 영상 처리 + Direct2D파일 다운로드1
11404정성태12/16/20177849.NET Framework: 709. C# - OpenCvSharp을 이용한 동영상(avi, mp4, ...) 처리 + Direct2D [4]파일 다운로드1
11403정성태12/16/20179837.NET Framework: 708. C# - OpenCvSharp을 이용한 동영상(avi, mp4, ...) 처리 [3]파일 다운로드1
11402정성태12/16/201714469.NET Framework: 707. OpenCV 응용 프로그램을 C#으로 구현 - OpenCvSharp [2]파일 다운로드1
11401정성태12/15/20176595.NET Framework: 706. C# - SharpDX + DXGI를 이용한 윈도우 화면 캡처 소스 코드 + Direct2D 출력파일 다운로드1
11400정성태12/14/20176542.NET Framework: 705. C# - SharpDX + DXGI를 이용한 윈도우 화면 캡처 소스 코드 [3]파일 다운로드1
11399정성태12/13/20172842.NET Framework: 704. Win32 API의 UnionRect를 닷넷 BCL의 Rectangle.Union으로 바꿀 때 주의 사항
11398정성태12/13/20173306오류 유형: 442. ASP.NET Core Web Application (on .NET Framework) 프로젝트에서 외부 라이브러리 동적 로드 시 런타임 버전 문제파일 다운로드1
11397정성태12/12/20174492.NET Framework: 703. 양자 컴퓨팅을 위한 마이크로소프트의 Q# 언어
11396정성태12/8/201714886개발 환경 구성: 343. Visual Studio - 리눅스 용 프로젝트의 인텔리센스를 위한 헤더 파일 처리 방법 [3]
11395정성태12/8/20173386오류 유형: 441. 이벤트 로그 - Time Provider NtpClient: No valid response has been received from domain controller
11394정성태12/8/20173165개발 환경 구성: 342. 비주얼 스튜디오에서 실행하던 ASP.NET Core (.NET Framework) 응용 프로그램을 명령행에서 실행하는 방법
11393정성태12/7/20174869Windows: 145. 윈도우 10 빌드 17046부터 WSL에서 백그라운드 작업 지원 [3]
11392정성태12/7/20172476개발 환경 구성: 341. openSUSE에 닷넷 코어 설치
11391정성태12/7/20173182개발 환경 구성: 340. WSL을 이용해 윈도우 PC 1대에서 openSUSE 응용 프로그램을 Visual Studio로 개발하는 방법 [1]
11390정성태12/7/20179140개발 환경 구성: 339. WSL을 이용해 윈도우 PC 1대에서 Linux 응용 프로그램을 Visual Studio로 개발하는 방법 [3]
11389정성태12/7/20173014오류 유형: 440. .NET Core 오류 - 0x80131620 Unable to load DLL 'libuv'
11388정성태10/4/20184723개발 환경 구성: 338. WSL 또는 Ubuntu에 닷넷 코어 설치 [2]
11387정성태12/6/20174665오류 유형: 439. 이벤트 로그 - Data Sharing Service 서비스의 %%3239247874 오류 메시지
11386정성태12/5/20173108오류 유형: 438. Hyper-V - '...' failed to add device 'Virtual CD/DVD Disk'
11385정성태12/14/20178088VC++: 121. DXGI를 이용한 윈도우 화면 캡처 소스 코드(Visual C++) [3]파일 다운로드1
... 31  32  33  34  [35]  36  37  38  39  40  41  42  43  44  45  ...