Microsoft MVP성태의 닷넷 이야기
글쓴 사람
정성태 (techsharer at outlook.com)
홈페이지
첨부 파일

(시리즈 글이 4개 있습니다.)
.NET Framework: 2008. C# - async/await 그리고 스레드 (1) MyTask로 재현
; https://www.sysnet.pe.kr/2/0/13055

.NET Framework: 2009. C# - async/await 그리고 스레드 (2) MyTask의 호출 흐름
; https://www.sysnet.pe.kr/2/0/13056

.NET Framework: 2012. C# - async/await 그리고 스레드 (3) Task.Delay 재현
; https://www.sysnet.pe.kr/2/0/13060

.NET Framework: 2014. C# - async/await 그리고 스레드 (4) 비동기 I/O 재현
; https://www.sysnet.pe.kr/2/0/13062




C# - async/await 그리고 스레드 (2) MyTask의 호출 흐름

자, 그럼 지난번 글에 이어서, 왜 "Thread.Sleep(16)"을 두었는지 설명해 보겠습니다. 우선, Task.Run을 사용한 예제에서 Sleep 코드를 제거해볼까요?

using System;

internal class Program
{
    static async Task Main(string[] args)
    {
        Console.WriteLine($"1단계: threadid == {Thread.CurrentThread.ManagedThreadId}");
        await DoAsync();
        Console.WriteLine($"2단계: threadid == {Thread.CurrentThread.ManagedThreadId}");
    }

    private static Task DoAsync()
    {
        return Task.Run(
            () =>
            {
                Console.WriteLine($"DoAsync called! (threadid == {Thread.CurrentThread.ManagedThreadId})");
            });
    }
}

위의 코드를 수행하면 보통 다음과 같은 출력을 볼 수 있습니다.

1단계: threadid == 1
DoAsync called! (threadid == 7)
2단계: threadid == 7

하지만 가끔씩, 이런 식의 출력도 나옵니다.

1단계: threadid == 1
DoAsync called! (threadid == 6)
2단계: threadid == 7

아니, Task.Run 스레드가 await 이후의 코드를 수행한다면서요? 그런데, 위의 결과는 (스레드 풀에 있던 것으로 짐작되는) 별도의 스레드가 await 이후의 코드를 실행하고 있습니다. 이유가 뭘까요?




이것을 밝혀내려면 (물론, Task.cs 소스 코드를 분석해도 되지만) 우리가 만든 MyTask의 소스 코드에서 마찬가지로 Thread.Sleep을 제거해 봐야 합니다.

using System.Runtime.CompilerServices;

internal class Program
{
    static async Task Main(string[] args)
    {
        Console.WriteLine($"1단계: threadid == {Thread.CurrentThread.ManagedThreadId}");
        await new MyDummy().DoAsync();
        Console.WriteLine($"2단계: threadid == {Thread.CurrentThread.ManagedThreadId}");
    }
}

class MyDummy
{
    public MyTask DoAsync()
    {
        MyTask _task = new MyTask(() =>
        {
            Console.WriteLine($"DoAsync called! (threadid == {Thread.CurrentThread.ManagedThreadId})");
        });

        return _task;
    }
}

public class MyTask
{
    Action? _continuation;

    bool _completed = false;
    public bool Completed => _completed;

    public MyTask(Action action)
    {
        ParameterizedThreadStart taskAction = (arg) =>
        {
            action();
            this.SetComplete();
        };

        Thread t = new Thread(taskAction);
        t.Start();
    }

    public MyTaskAwaiter GetAwaiter()
    {
        MyTaskAwaiter ta = new MyTaskAwaiter(this);
        return ta;
    }

    public void SetComplete()
    {
        _completed = true;

        if (_continuation != null)
        {
            _continuation();
        }
    }

    private void SetContinuation(Action continuation)
    {
        _continuation = continuation;
    }

    public struct MyTaskAwaiter : ICriticalNotifyCompletion, INotifyCompletion
    {
        MyTask _task;

        public MyTaskAwaiter(MyTask task)
        {
            _task = task;
        }

        public bool IsCompleted
        {
            get { return _task.Completed; }
        }

        public void GetResult()
        {
        }

        public void OnCompleted(Action continuation)
        {
        }

        public void UnsafeOnCompleted(Action continuation)
        {
            _task.SetContinuation(continuation);
        }
    }
}

이렇게 바꾸고 실행하면, 이제는 다음과 같이 await 이후의 코드가 실행되지 않는 것을 볼 수 있습니다.

1단계: threadid == 1
DoAsync called! (threadid == 7)

무엇이 문제일까요? 간단합니다. _continuation을 호출하지 못하는 상황이 발생하기 때문입니다.

public MyTask(Action action)
{
    ParameterizedThreadStart taskAction = (arg) =>
    {
        action();
        this.SetComplete();
    };

    Thread t = new Thread(taskAction);
    t.Start();
}

public void SetComplete()
{
    _completed = true;

    if (_continuation != null)
    {
        _continuation();
    }
}

위의 동작과 함께 원래의 await 코드가 다음과 같이 실행되는 형식이라는 것을 감안하면,

Task task = new MyDummy().DoAsync();
if (task.IsCompleted == true)
{
    // "분할 2" 코드 실행
}
else
{
    task.GetAwaiter().UnsafeOnCompleted(...[분할 2] 코드...);
}

Thread.Start 실행은 되었지만 아직 SetComplete를 호출하기 전이라고 가정해 보겠습니다. 그럼, if (task.IsCompleted == true)는 아니기 때문에 "분할 2" 코드는 실행되지 않고, 향후 IsCompleted가 true인 상황에서 실행되도록 UnsafeOnCompleted를 통해 보관하는 else 절을 타게 됩니다.

하지만, else 절까지 이동했지만 그사이 SetComplete의 코드가 실행되는 경우라면 어떨까요? 그럼 _completed = true가 되지만 아직 _continuation이 들어온 것은 아니므로 SetComplete 메서드에서는 실행이 안 될 것입니다. 결국 "분할 2" 코드는 어디에서도 실행이 안 된 것입니다.

따라서, 저런 식의 상황을 만들지 않기 위해 일부러 Thread.Sleep(16) 정도의 시간을 주면 우리가 원하는 대로 항상 스레드가 _continuation 코드를 실행할 수 있었던 것입니다.




(확실하게 분석한 것은 아니지만) 닷넷의 Task는 이런 문제를 해결하기 위해 실행되지 않는 UnsafeOnCompleted의 action 코드를 스레드 풀의 스레드가 실행하는 대비책을 마련해두었습니다.

이를 이용해 우리가 만든 MyTask의 동작을 다음과 같이 변경할 수 있습니다.

public class MyTask
{
    // ...[생략]...

    bool _executed = false;

    public void SetComplete()
    {
        _completed = true;

        _executed = _continuation != null;

        if (_executed == true)
        {
            _continuation!();
        }
    }

    private void SetContinuation(Action continuation)
    {
        _continuation = continuation;

        if (_completed == true && _executed == false)
        {
            ThreadPool.QueueUserWorkItem((obj) =>
           {
               _continuation();
           });
        }
    }

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

이렇게 하면, SetComplete 시점에 _continuation을 실행할 수 없어도 (UnsafeOnCompleted를 경유한) SetContinuation에서 스레드 풀에 넣어 두는 작업을 한 번 거치기 때문에 이런 실행 결과를 보입니다.

1단계: threadid == 1
DoAsync called! (threadid == 7)
2단계: threadid == 8




유의할 점은, 제가 이 글에서 구현하는 MyTask와 MyTaskAwaiter는 임의로 재현을 위해 코드를 만든 것일 뿐 이 동작이 BCL의 Task, TaskAwaiter와 정확히 같지는 않습니다. 그냥 전체적인 흐름이 그런 식으로 된다는 정도로 이해하시면 되겠습니다.

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




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







[최초 등록일: ]
[최종 수정일: 9/3/2023]

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

비밀번호

댓글 작성자
 



2022-05-13 12시18분
[한예지] 선생님, 코드에서 arg 매개변수를 사용하지 않는 것 같은데
MyTask 생성자에서 ParameterizedThreadStart 델리게이트를 사용하신 이유를 알 수 있을까요?
[guest]
2022-05-13 09시00분
아... 이유 없습니다. 습관적으로 스레드 생성자에 그걸 사용해서 그렇게 된 것입니다. ^^; (당연히 arg를 사용하지 않으면 ThreadStart 델리게이트를 사용해도 무방합니다.)
정성태

... 61  62  63  64  65  66  67  [68]  69  70  71  72  73  74  75  ...
NoWriterDateCnt.TitleFile(s)
12110정성태1/11/202015313디버깅 기술: 154. Patch Guard로 인해 블루 스크린(BSOD)가 발생하는 사례 [5]파일 다운로드1
12109정성태1/10/202012725오류 유형: 588. Driver 프로젝트 빌드 오류 - Inf2Cat error -2: "Inf2Cat, signability test failed."
12108정성태1/10/202013240오류 유형: 587. Kernel Driver 시작 시 127(The specified procedure could not be found.) 오류 메시지 발생
12107정성태1/10/202014246.NET Framework: 877. C# - 프로세스의 모든 핸들을 열람 - 두 번째 이야기
12106정성태1/8/202015870VC++: 136. C++ - OSR Driver Loader와 같은 Legacy 커널 드라이버 설치 프로그램 제작 [1]
12105정성태1/8/202014325디버깅 기술: 153. C# - PEB를 조작해 로드된 DLL을 숨기는 방법
12104정성태1/7/202014901DDK: 9. 커널 메모리를 읽고 쓰는 NT Legacy driver와 C# 클라이언트 프로그램 [4]
12103정성태1/7/202017833DDK: 8. Visual Studio 2019 + WDK Legacy Driver 제작- Hello World 예제 [1]파일 다운로드2
12102정성태1/6/202015077디버깅 기술: 152. User 권한(Ring 3)의 프로그램에서 _ETHREAD 주소(및 커널 메모리를 읽을 수 있다면 _EPROCESS 주소) 구하는 방법
12101정성태1/5/202014661.NET Framework: 876. C# - PEB(Process Environment Block)를 통해 로드된 모듈 목록 열람
12100정성태1/3/202012309.NET Framework: 875. .NET 3.5 이하에서 IntPtr.Add 사용
12099정성태1/3/202014786디버깅 기술: 151. Windows 10 - Process Explorer로 확인한 Handle 정보를 windbg에서 조회 [1]
12098정성태1/2/202014799.NET Framework: 874. C# - 커널 구조체의 Offset 값을 하드 코딩하지 않고 사용하는 방법 [3]
12097정성태1/2/202012924디버깅 기술: 150. windbg - Wow64, x86, x64에서의 커널 구조체(예: TEB) 구조체 확인
12096정성태12/30/201915995디버깅 기술: 149. C# - DbgEng.dll을 이용한 간단한 디버거 제작 [1]
12095정성태12/27/201917081VC++: 135. C++ - string_view의 동작 방식
12094정성태12/26/201914744.NET Framework: 873. C# - 코드를 통해 PDB 심벌 파일 다운로드 방법
12093정성태12/26/201914510.NET Framework: 872. C# - 로딩된 Native DLL의 export 함수 목록 출력파일 다운로드1
12092정성태12/25/201914158디버깅 기술: 148. cdb.exe를 이용해 (ntdll.dll 등에 정의된) 커널 구조체 출력하는 방법
12091정성태12/25/201915540디버깅 기술: 147. pdb 파일을 다운로드하기 위한 symchk.exe 실행에 필요한 최소 파일 [1]
12090정성태12/24/201915114.NET Framework: 871. .NET AnyCPU로 빌드된 PE 헤더의 로딩 전/후 차이점 [1]파일 다운로드1
12089정성태12/23/201914791디버깅 기술: 146. gflags와 _CrtIsMemoryBlock을 이용한 Heap 메모리 손상 여부 체크
12088정성태12/23/201913792Linux: 28. Linux - 윈도우의 "Run as different user" 기능을 shell에서 실행하는 방법
12087정성태12/21/201914232디버깅 기술: 145. windbg/sos - Dictionary의 entries 배열 내용을 모두 덤프하는 방법 (do_hashtable.py) [1]
12086정성태12/20/201916164디버깅 기술: 144. windbg - Marshal.FreeHGlobal에서 발생한 덤프 분석 사례
12085정성태12/20/201914105오류 유형: 586. iisreset - The data is invalid. (2147942413, 8007000d) 오류 발생 - 두 번째 이야기 [1]
... 61  62  63  64  65  66  67  [68]  69  70  71  72  73  74  75  ...