How to dispose a MemoryStream that is used in a Task?

Question

I have the following method:

public void Write() {
     var tasks = new List<Task>();
     while(...) {
         var memoryStream = new MemoryStream(...);
         var task = _pageBlob.WritePagesAsync(memoryStream, ... );
         tasks.Add(task);
     }    
     Task.WaitAll(tasks.ToArray());
}

How to correctly dispose memoryStream that in Task? I need to dispose the memoryStream object when the task is finished.

Answer 1

You have two options:

1-Encapsulate all the process inside a task:

while(...) {

     var task = Task.Run(async () => {

         var memoryStream = new MemoryStream(...);
         var res = await _pageBlob.WritePagesAsync(memoryStream, ... );
         memoryStream.Dispose();

     });

     tasks.Add(task);
 }    

2-Use a Continuation:

 while(...) {
     var memoryStream = new MemoryStream(...);
     var task = _pageBlob.WritePagesAsync(memoryStream, ... )
                .ContinueWith((PrevTask) => memoryStream.Dispose());

     tasks.Add(task);
 }    

Answer 2

Split out your while loop body into a separate async method:

private async Task WriteAsync(...)
{
  using (var memoryStream = new MemoryStream(...))
  {
    await _pageBlob.WritePagesAsync(memoryStream, ...);
  }
}

Then use your new method:

public void Write() {
  var tasks = new List<Task>();
  while(...) {
    tasks.Add(WriteAsync(...));
  }    
  Task.WaitAll(tasks.ToArray());
}

On a side note, blocking on asynchronous code (Task.WaitAll) is not generally a good idea. A more natural approach is to keep it asynchronous:

public async Task WriteAsync() {
  var tasks = new List<Task>();
  while(...) {
    tasks.Add(WriteAsync(...));
  }    
  await Task.WhenAll(tasks);
}