[c#] 태스크에서 동기식 연속을 방지하려면 어떻게해야합니까?

연속에 대한 명시적인 API 제어를 제공하는 TPL에는 아무것도 없다고 생각합니다 TaskCompletionSource.SetResult. 이 동작을 제어 하기 위해 초기 답변 을 유지하기로 결정했습니다.async/await시나리오 .

다음은 호출 된 동일한 스레드에서 트리거 된 연속이 발생하는 ContinueWith경우에 비동기를 부과하는 또 다른 솔루션입니다 .tcs.SetResultSetResult

public static class TaskExt
{
    static readonly ConcurrentDictionary<Task, Thread> s_tcsTasks =
        new ConcurrentDictionary<Task, Thread>();

    // SetResultAsync
    static public void SetResultAsync<TResult>(
        this TaskCompletionSource<TResult> @this,
        TResult result)
    {
        s_tcsTasks.TryAdd(@this.Task, Thread.CurrentThread);
        try
        {
            @this.SetResult(result);
        }
        finally
        {
            Thread thread;
            s_tcsTasks.TryRemove(@this.Task, out thread);
        }
    }

    // ContinueWithAsync, TODO: more overrides
    static public Task ContinueWithAsync<TResult>(
        this Task<TResult> @this,
        Action<Task<TResult>> action,
        TaskContinuationOptions continuationOptions = TaskContinuationOptions.None)
    {
        return @this.ContinueWith((Func<Task<TResult>, Task>)(t =>
        {
            Thread thread = null;
            s_tcsTasks.TryGetValue(t, out thread);
            if (Thread.CurrentThread == thread)
            {
                // same thread which called SetResultAsync, avoid potential deadlocks

                // using thread pool
                return Task.Run(() => action(t));

                // not using thread pool (TaskCreationOptions.LongRunning creates a normal thread)
                // return Task.Factory.StartNew(() => action(t), TaskCreationOptions.LongRunning);
            }
            else
            {
                // continue on the same thread
                var task = new Task(() => action(t));
                task.RunSynchronously();
                return Task.FromResult(task);
            }
        }), continuationOptions).Unwrap();
    }
}

댓글을 해결하도록 업데이트되었습니다.

나는 발신자를 통제하지 않는다-그들에게 특정한 continue-with 변형을 사용하도록 할 수 없다 : 가능하다면, 문제는 처음에 존재하지 않을 것이다.

나는 당신이 발신자를 통제하지 않는다는 것을 몰랐습니다. 그럼에도 불구하고, 당신이 그것을 제어하지 않는다면, 당신은 아마도 TaskCompletionSource객체 를 호출자에게 직접 전달하지 않을 것입니다 . 논리적으로, 당신 은 그것 의 토큰 부분, 즉 tcs.Task. 이 경우 위의 다른 확장 메서드를 추가하면 솔루션이 더 쉬울 수 있습니다.

// ImposeAsync, TODO: more overrides
static public Task<TResult> ImposeAsync<TResult>(this Task<TResult> @this)
{
    return @this.ContinueWith(new Func<Task<TResult>, Task<TResult>>(antecedent =>
    {
        Thread thread = null;
        s_tcsTasks.TryGetValue(antecedent, out thread);
        if (Thread.CurrentThread == thread)
        {
            // continue on a pool thread
            return antecedent.ContinueWith(t => t,
                TaskContinuationOptions.None).Unwrap();
        }
        else
        {
            return antecedent;
        }
    }), TaskContinuationOptions.ExecuteSynchronously).Unwrap();
}

사용하다:

// library code
var source = new TaskCompletionSource<int>();
var task = source.Task.ImposeAsync();
// ... 

// client code
task.ContinueWith(delegate
{
    Identify();
}, TaskContinuationOptions.ExecuteSynchronously);

// ...
// library code
source.SetResultAsync(123);

이것은 실제로 및 ( 바이올린 ) 모두에서 작동하며awaitContinueWith 반사 해킹이 없습니다.