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

비밀번호

댓글 쓴 사람
 




... 16  17  18  19  20  21  22  23  24  25  26  27  28  29  [30]  ...
NoWriterDateCnt.TitleFile(s)
11537정성태6/8/20183546.NET Framework: 761. C# - SmtpClient로 SMTP + SSL/TLS 서버를 이용하는 방법 [4]
11536정성태6/8/20183554.NET Framework: 760. Microsoft Build 2018 - The future of C# 동영상 내용 정리 [1]파일 다운로드1
11535정성태6/7/20183451.NET Framework: 759. C# - System.Span<T> 성능 [1]
11534정성태10/7/20194871.NET Framework: 758. C# 7.2 - Span<T> [3]
11533정성태6/8/20186915.NET Framework: 757. 포인터 형 매개 변수를 갖는 C++ DLL의 함수를 C#에서 호출하는 방법파일 다운로드1
11532정성태6/5/20183417.NET Framework: 756. JSON의 escape sequence 문자 처리 방식
11531정성태6/5/20186270오류 유형: 468. JSON.parse가 허용하지 않는 문자 [9]
11530정성태6/10/20184230.NET Framework: 755. C# 7.2 - 스택에만 생성할 수 있는 값 타입 지원 - "ref struct" [2]파일 다운로드1
11529정성태5/23/20183216.NET Framework: 754. 닷넷의 관리 포인터(Managed Pointer)와 System.TypedReference [5]파일 다운로드1
11528정성태6/10/20183221.NET Framework: 753. C# 7.2 - 3항 연산자에 ref 지원(conditional ref operator) [1]
11527정성태5/17/20182926오류 유형: 467. RDP 로그인 에러 - This could be due to CredSSP encryption oracle remediation.
11526정성태6/10/20182665.NET Framework: 752. C# 7.2 - 메서드의 반환값 및 로컬 변수에 ref readonly 기능 추가파일 다운로드1
11525정성태6/10/20182867.NET Framework: 751. C# 7.2 - 메서드의 매개 변수에 in 변경자 추가파일 다운로드1
11524정성태6/10/20183866.NET Framework: 750. C# 7.2 - readonly 구조체 [3]파일 다운로드1
11523정성태5/15/20183229.NET Framework: 749. C# - 값 형식의 readonly 인스턴스에 대한 메서드 호출 시 defensive copy 발생 [1]파일 다운로드1
11522정성태5/15/20182059개발 환경 구성: 378. Azure - VM 진단 설정 화면의 "This subscription is not registered with the Microsoft.Insights resource provider."
11521정성태5/15/20182076개발 환경 구성: 377. Azure - 원하는 성능 데이터로 모니터링 대시보드 구성
11520정성태5/12/20182719.NET Framework: 748. C# 7.1 - 참조 어셈블리(Ref Assemblies)
11519정성태3/13/20192916개발 환경 구성: 376. ASP.NET Web Application 프로젝트의 FileSystem 배포(Publish) 시 Before/After Task 설정 방법 [1]
11518정성태5/11/20182599.NET Framework: 747. C# 7.0에서도 부분적으로 가능해진 "타입 추론을 통한 튜플의 변수명 자동 지정"
11517정성태5/10/20182916.NET Framework: 746. Azure runbook 예제 - 6시간 동안 수행 중인 VM을 중지 [1]파일 다운로드1
11516정성태5/9/20182256.NET Framework: 745. Azure runbook을 PowerShell 또는 C# 코드로 실행하는 방법파일 다운로드1
11515정성태6/8/20182639.NET Framework: 744. C# 6 - Expression bodied function
11514정성태5/3/20184052오류 유형: 466. Bitvise - Error in component session/transport/kexHandler [2]
11513정성태5/10/20185096.NET Framework: 743. C# 언어의 공변성과 반공변성 [5]파일 다운로드1
11512정성태5/3/20182317개발 환경 구성: 375. Azure runbook 실행 시 "Errors", "All Logs"에 오류 메시지가 출력되는 경우
... 16  17  18  19  20  21  22  23  24  25  26  27  28  29  [30]  ...