如何形成TaskCompletionSource <t>的集合并保持类型安全

时间:2019-03-29 09:46:31

标签: c# generics casting async-await taskcompletionsource

我有一些异步方法:

// not ideal
private TaskCompletionSource<A> _tcsA;
private TaskCompletionSource<A> _tcsB;
private TaskCompletionSource<A> _tcsC;
...

public Task<A> GetAAsync() {
  _currentTask = TaskType.A;
  _tcsA = new TaskCompletionSource<A>();
  // some complex non-sync task, using http/events that ends with Complete();
  // QueueRequest?.Invoke(this, taskInfo); // raise request -- this does not return
  return _tcsA.Task;
}

public Task<B> GetBAsync() {
  _currentTask = TaskType.B;
  _tcsB = new TaskCompletionSource<B>();
  // some complex non-sync task, using http/events that ends with Complete();
  // QueueRequest?.Invoke(this, taskInfo); // raise request -- this does not return
  return _tcsB.Task;
}

public Task<C> GetCAsync() {
  _currentTask = TaskType.C;
  _tcsC = new TaskCompletionSource<C>();
  // some complex non-sync task, using http/events that ends with Complete();
  // QueueRequest?.Invoke(this, taskInfo); // raise request -- this does not return
  return _tcsC.Task;
}

// called by an external source, a http listener / processor
// this should complete the async call by the client and return results
public void Complete(Result result) {
  switch (_currentTask) {
    case TaskType.A:
      _tcsA.SetResult(new A());
      break;
    case TaskType.B:
      _tcsB.SetResult(new B());
      break;
    case TaskType.C:
      _tcsC.SetResult(new C());
      break;
  }

  _currentTask = TaskType.None;
}

为简单起见,以上是半伪代码。我称这样的方法之一:

A a = await service.GetAAsync();

现在的问题是TaskCompletionSource<T>是泛型的,如果我有100种方法,则必须为每种返回类型创建一个变量。但是由于一次只能调用一个方法,因此最好使用单个TaskCompletionSource,而不是将其键入object (TaskCompletionSource<object>)

我不想做:

object a = await service.GetAAsync();

因为这将需要客户端强制转换。因此,最好的解决方案是只有一个TaskCompletionSource,但以某种方式键入它。或者,也可以具有TaskCompletionSource的字典。这两个对我来说似乎都是不可能的。

我应该如何解决?

更新:

有关我的情况的背景信息,请查看:Wrap synchronous code into async await in disconnected scenario

2 个答案:

答案 0 :(得分:3)

  

我不想做object a = await service.GetAAsync(),因为这需要客户强制转换。

不一定。保持相同的签名,但添加对ContinueWith的调用以强制转换为适当的类型:

public Task<A> GetAAsync() {
  _currentTask = TaskType.A;
  _tcs = new TaskCompletionSource<object>();
  // some complex task that ends with Complete();
  return _tcs.Task.ContinueWith(t => (A)t.Result);
}

或者您也可以使用async / await做同样的事情:

public async Task<A> GetAAsync() {
  _currentTask = TaskType.A;
  _tcs = new TaskCompletionSource<object>();
  // some complex task that ends with Complete();
  return (A)await _tcs.Task;
}

答案 1 :(得分:0)

使用TaskCompletionSourceDictionary <TKey>可以轻松实现:

目标:

  • 任何类型的键都可以使用-在此示例中,我使用了您的TaskType枚举
  • 如果每种类型只有一个实例,那么非通用版本会更简单
  • 生产者和消费者的动态数量
  • 结果的完成可以被多个消费者观察
  • 强制执行类型安全-生产者和消费者必须指定相同的类型,否则强制转换异常描述类型差异
  • 生产者可以在消费者要求任务之前设置结果
  • 消费者可以在生产者设置结果之前要求任务

我们可以在这个新类上使用GetOrAdd方法来完成所有这些操作:

TaskCompletionSource<TValue> GetOrAdd<TValue>(TKey key)

用法示例:

private readonly TaskCompletionSourceDictionary<TaskType> _tasks = new TaskCompletionSourceDictionary<TaskType>();

public Task<A> GetAAsync() =>
    _tasks.GetOrAdd<A>(TaskType.A).Task;

public Task<B> GetBAsync() =>
    _tasks.GetOrAdd<B>(TaskType.B).Task;

public Task<C> GetCAsync() =>
    _tasks.GetOrAdd<C>(TaskType.C).Task;

// called by an external source, a http listener / processor
// this should complete the async call by the client and return results
public void Complete(Result result)
{
    switch (_currentTask)
    {
        case TaskType.A:
            _tasks.GetOrAdd<A>(TaskType.A).TrySetResult(new A());
            break;
        case TaskType.B:
            _tasks.GetOrAdd<B>(TaskType.B).TrySetResult(new B());
            break;
        case TaskType.C:
            _tasks.GetOrAdd<C>(TaskType.C).TrySetResult(new C());
            break;
    }

    _currentTask = TaskType.None;
}

除了GetOrAdd外,还有TryAddTryRemove。不过,删除可能很危险-对于最后一个消费者来说,删除也是如此。