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

C# - AsyncLocal 기능을 CallContext만으로 구현하는 방법

전에 CallContext를 설명하면서,

HttpContext.Current를 통해 이해하는 CallContext와 ExecutionContext
; https://www.sysnet.pe.kr/2/0/1608

덧글로 잠시 소개를 했는데요,

Implicit Async Context ("AsyncLocal")
; https://blog.stephencleary.com/2013/04/implicit-async-context-asynclocal.html

위의 글에서는 AsyncLocal 타입을 사용하지 않고 (.NET Core에서는 제공되지 않는) 순수 CallContext만을 이용해 await 호출 간의 문맥을 전달하고 있습니다. 코드를 여기다 그대로 옮겨 볼까요? ^^ 근래의 현실적인 기준으로 볼 때, (.NET Core/5에서 다중 AppDomain을 지원하지 않으므로) MarshalByRefObject 처리를 없애면 다음과 같이 간단하게 정리할 수 있습니다.

// Install-Package System.Collections.Immutable

using System;
using System.Collections.Immutable;
using System.Linq;
using System.Runtime.CompilerServices;
using System.Runtime.Remoting.Messaging;

public static partial class MyStack
{
    private static readonly string name = Guid.NewGuid().ToString("N");

    private static ImmutableStack<string> CurrentContext
    {
        get
        {
            var ret = CallContext.LogicalGetData(name) as ImmutableStack<string>;
            return ret == null ? ImmutableStack.Create<string>() : ret;
        }

        set
        {
            CallContext.LogicalSetData(name, value);
        }
    }

    public static IDisposable Push([CallerMemberName] string context = "")
    {
        CurrentContext = CurrentContext.Push(context);
        return new PopWhenDisposed();
    }

    private static void Pop()
    {
        CurrentContext = CurrentContext.Pop();
    }

    private sealed class PopWhenDisposed : IDisposable
    {
        private bool disposed;

        public void Dispose()
        {
            if (disposed)
                return;
            Pop();
            disposed = true;
        }
    }

    public static string CurrentStack
    {
        get
        {
            return string.Join(" ", CurrentContext.Reverse());
        }
    }
}

이를 활용한 예제 코드는 변경 없이 원 글의 코드를 그대로 옮겨 봅니다.

using System;
using System.Threading.Tasks;

class Program
{
    static void Main(string[] args)
    {
        using (MyStack.Push("Main"))
        {
            Task.WhenAll(SomeWork("1"), SomeWork("2")).Wait();
        }

        Console.ReadKey();
    }

    static async Task SomeWork(string stackName)
    {
        using (MyStack.Push(stackName))
        {
            Log("<SomeWork>");
            await MoreWork("A");
            await MoreWork("B");
            Log("</SomeWork>");
        }
    }

    static async Task MoreWork(string stackName)
    {
        using (MyStack.Push(stackName))
        {
            Log("<MoreWork>");
            await Task.Delay(10);
            Log("</MoreWork>");
        }
    }

    static void Log(string message)
    {
        Console.WriteLine(MyStack.CurrentStack + ": " + message);
    }
}

실행하면 다음과 같은 결과를 얻을 수 있습니다.

Main 1: <SomeWork>
Main 1 A: <MoreWork>
Main 2: <SomeWork>
Main 2 A: <MoreWork>
Main 2 A: </MoreWork>
Main 1 A: </MoreWork>
Main 2 B: <MoreWork>
Main 1 B: <MoreWork>
Main 2 B: </MoreWork>
Main 2: </SomeWork>
Main 1 B: </MoreWork>
Main 1: </SomeWork>




그런데, 사실 해당 기능을 너무 어렵게 만든 것이 아닌가... 하는 느낌입니다. 왜냐하면, CallContext는 스레드를 넘어가면서 shallow copy가 되는데, 달리 말하면 메서드에 인자를 넘기는 것과 같습니다. 따라서, 굳이 Push/IDisposable/Pop을 사용할 필요 없이 다음과 같은 식으로 바꿔도 무방합니다.

using System;
using System.Runtime.Remoting.Messaging;

public static partial class MyStack
{
    private static readonly string idTitle = Guid.NewGuid().ToString("N");

    private static string CurrentTitle
    {
        get
        {
            object objValue = CallContext.LogicalGetData(idTitle);
            if (objValue == null)
            {
                return "";
            }

            return objValue as string;
        }

        set
        {
            CallContext.LogicalSetData(idTitle, value);
        }
    }

    public static void Push(string title)
    {
        CurrentTitle = CurrentTitle + " " + title;
    }

    public static string CurrentStack
    {
        get
        {
            return CurrentTitle;
        }
    }
}

또한 사용 측 코드도 using을 빼고 단순히 Push 메서드만 호출해 주면 됩니다.

using System;
using System.Threading.Tasks;

class Program
{
    static void Main(string[] args)
    {
        MyStack.Push("Main");
        Task.WhenAll(SomeWork("1"), SomeWork("2")).Wait();

        Console.ReadKey();
    }

    static async Task SomeWork(string stackName)
    {
        MyStack.Push(stackName);
        Log("<SomeWork>");
        await MoreWork("A");
        await MoreWork("B");
        Log("</SomeWork>");
    }

    static async Task MoreWork(string stackName)
    {
        MyStack.Push(stackName);
        Log("<MoreWork>");
        await Task.Delay(10);
        Log("</MoreWork>");
    }

    static void Log(string message)
    {
        Console.WriteLine(MyStack.CurrentStack + ": " + message);
    }
}

당연히 실행 결과는 이전과 다름없이 잘 나옵니다.




마지막으로, 이것을 AsyncLocal을 이용한 예제로 바꿔볼까요?

// .NET Framework 4.6 이상, .NET Core/5에서도 사용 가능

using System.Threading;

public static partial class MyStack
{
    private static AsyncLocal<string> CurrentTitle = new AsyncLocal<string>();

    public static void Push(string title)
    {
        CurrentTitle.Value = CurrentTitle.Value + " " + title;
    }

    public static string CurrentStack
    {
        get
        {
            return CurrentTitle.Value;
        }
    }
}

보는 바와 같이, AsyncLocal은 단순한 래퍼에 불과합니다. 단지, CallContext 타입이 .NET Framework에서만 허용되고 .NET Core/5에서는 사용할 수 없으므로 어쩔 수 없이 향후에는 AsyncLocal을 쓸 수밖에 없습니다.

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




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

[연관 글]


donaricano-btn



[최초 등록일: ]
[최종 수정일: 7/14/2021

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

비밀번호

댓글 쓴 사람
 



2021-07-16 09시06분
ExecutionContext의 또 다른 활용 사례

Exploring the async/await State Machine – Stack Traces and Refactoring Pitfalls
; https://vkontech.com/exploring-the-async-await-state-machine-stack-traces-and-refactoring-pitfalls/
정성태

1  [2]  3  4  5  6  7  8  9  10  11  12  13  14  15  ...
NoWriterDateCnt.TitleFile(s)
12815정성태8/30/2021191개발 환경 구성: 602. WSL 2 - docker-desktop-data, docker-desktop (%LOCALAPPDATA%\Docker\wsl\data\ext4.vhdx) 파일을 다른 디렉터리로 옮기는 방법
12814정성태8/30/2021254.NET Framework: 1110. C# 10 - (12) 인터페이스 내에 정적 추상 메서드 정의 가능파일 다운로드1
12813정성태8/29/2021246.NET Framework: 1109. C# 10 - (11) Lambda 개선파일 다운로드1
12812정성태8/28/2021222.NET Framework: 1108. C# 10 - (10) 개선된 #line 지시자
12811정성태8/27/2021193Linux: 44. 윈도우 개발자를 위한 리눅스 fork 동작 방식 설명 (파이썬 코드)
12810정성태8/27/2021131.NET Framework: 1107. .NET Core/5+에서 동적 컴파일한 C# 코드를 (Breakpoint도 활용하며) 디버깅하는 방법 - #line 지시자파일 다운로드1
12809정성태8/26/2021225.NET Framework: 1106. .NET Core/5+에서 C# 코드를 동적으로 컴파일/사용하는 방법파일 다운로드1
12808정성태8/25/2021121오류 유형: 758. go: ...: missing go.sum entry; to add it: go mod download ...
12807정성태8/25/2021215.NET Framework: 1105. C# 10 - (9) 비동기 메서드가 사용할 AsyncMethodBuilder 선택 가능파일 다운로드1
12806정성태8/24/2021169개발 환경 구성: 601. PyCharm - 다중 프로세스 디버깅 방법
12805정성태8/24/2021154.NET Framework: 1104. C# 10 - (8) 분해 구문에서 기존 변수의 재사용 가능파일 다운로드1
12804정성태8/24/2021160.NET Framework: 1103. C# 10 - (7)
12803정성태8/23/2021113개발 환경 구성: 600. pip cache 디렉터리 옮기는 방법
12802정성태8/23/2021218.NET Framework: 1102. .NET Conf Mini 21.08 - WinUI 3 따라해 보기
12801정성태8/23/2021128.NET Framework: 1101. C# 10 - (6) record class 타입의 ToString 메서드를 sealed 처리 허용파일 다운로드1
12800정성태8/22/2021127개발 환경 구성: 599. PyCharm - (반대로) 원격 프로세스가 PyCharm에 디버그 연결하는 방법
12799정성태8/22/2021215.NET Framework: 1100. C# 10 - (5) 속성 패턴의 개선파일 다운로드1
12798정성태8/21/2021308개발 환경 구성: 598. PyCharm - 원격 프로세스를 디버그하는 방법
12797정성태8/21/2021116Windows: 197. TCP의 MSS(Maximum Segment Size) 크기는 고정된 것일까요?
12796정성태8/21/2021210.NET Framework: 1099. C# 10 - (4) 상수 문자열에 포맷 식 사용 가능파일 다운로드1
12795정성태8/20/2021271.NET Framework: 1098. .NET 6에 포함된 신규 BCL API - 스레드 관련
12794정성태8/20/2021596스크립트: 23. 파이썬 - WSGI를 만족하는 최소한의 구현 코드 및 PyCharm에서의 디버깅 방법 [1]
12793정성태8/20/2021667.NET Framework: 1097. C# 10 - (3) 개선된 변수 초기화 판정파일 다운로드1
12792정성태8/19/2021204.NET Framework: 1096. C# 10 - (2) 전역 네임스페이스 선언파일 다운로드1
12791정성태8/19/2021155.NET Framework: 1095. C# COM 개체를 C++에서 사용하는 예제 [3]파일 다운로드1
12790정성태8/18/2021264.NET Framework: 1094. C# 10 - (1) 구조체를 생성하는 record struct파일 다운로드1
1  [2]  3  4  5  6  7  8  9  10  11  12  13  14  15  ...