Improve, Select/Where add UniTaskTracker

master
neuecc 2020-07-04 06:29:50 +09:00
parent 551128e64c
commit fee5518a82
2 changed files with 24 additions and 0 deletions

View File

@ -88,6 +88,7 @@ namespace Cysharp.Threading.Tasks.Linq
this.selector = selector; this.selector = selector;
this.cancellationToken = cancellationToken; this.cancellationToken = cancellationToken;
this.moveNextAction = MoveNext; this.moveNextAction = MoveNext;
TaskTracker.TrackActiveTask(this, 3);
} }
public TResult Current { get; private set; } public TResult Current { get; private set; }
@ -156,6 +157,7 @@ namespace Cysharp.Threading.Tasks.Linq
public UniTask DisposeAsync() public UniTask DisposeAsync()
{ {
TaskTracker.RemoveTracking(this);
return enumerator.DisposeAsync(); return enumerator.DisposeAsync();
} }
} }
@ -195,6 +197,7 @@ namespace Cysharp.Threading.Tasks.Linq
this.selector = selector; this.selector = selector;
this.cancellationToken = cancellationToken; this.cancellationToken = cancellationToken;
this.moveNextAction = MoveNext; this.moveNextAction = MoveNext;
TaskTracker.TrackActiveTask(this, 3);
} }
public TResult Current { get; private set; } public TResult Current { get; private set; }
@ -263,6 +266,7 @@ namespace Cysharp.Threading.Tasks.Linq
public UniTask DisposeAsync() public UniTask DisposeAsync()
{ {
TaskTracker.RemoveTracking(this);
return enumerator.DisposeAsync(); return enumerator.DisposeAsync();
} }
} }
@ -302,6 +306,7 @@ namespace Cysharp.Threading.Tasks.Linq
this.selector = selector; this.selector = selector;
this.cancellationToken = cancellationToken; this.cancellationToken = cancellationToken;
this.moveNextAction = MoveNext; this.moveNextAction = MoveNext;
TaskTracker.TrackActiveTask(this, 3);
} }
public TResult Current { get; private set; } public TResult Current { get; private set; }
@ -382,6 +387,7 @@ namespace Cysharp.Threading.Tasks.Linq
public UniTask DisposeAsync() public UniTask DisposeAsync()
{ {
TaskTracker.RemoveTracking(this);
return enumerator.DisposeAsync(); return enumerator.DisposeAsync();
} }
} }
@ -422,6 +428,7 @@ namespace Cysharp.Threading.Tasks.Linq
this.selector = selector; this.selector = selector;
this.cancellationToken = cancellationToken; this.cancellationToken = cancellationToken;
this.moveNextAction = MoveNext; this.moveNextAction = MoveNext;
TaskTracker.TrackActiveTask(this, 3);
} }
public TResult Current { get; private set; } public TResult Current { get; private set; }
@ -502,6 +509,7 @@ namespace Cysharp.Threading.Tasks.Linq
public UniTask DisposeAsync() public UniTask DisposeAsync()
{ {
TaskTracker.RemoveTracking(this);
return enumerator.DisposeAsync(); return enumerator.DisposeAsync();
} }
} }
@ -541,6 +549,7 @@ namespace Cysharp.Threading.Tasks.Linq
this.selector = selector; this.selector = selector;
this.cancellationToken = cancellationToken; this.cancellationToken = cancellationToken;
this.moveNextAction = MoveNext; this.moveNextAction = MoveNext;
TaskTracker.TrackActiveTask(this, 3);
} }
public TResult Current { get; private set; } public TResult Current { get; private set; }
@ -621,6 +630,7 @@ namespace Cysharp.Threading.Tasks.Linq
public UniTask DisposeAsync() public UniTask DisposeAsync()
{ {
TaskTracker.RemoveTracking(this);
return enumerator.DisposeAsync(); return enumerator.DisposeAsync();
} }
} }
@ -661,6 +671,7 @@ namespace Cysharp.Threading.Tasks.Linq
this.selector = selector; this.selector = selector;
this.cancellationToken = cancellationToken; this.cancellationToken = cancellationToken;
this.moveNextAction = MoveNext; this.moveNextAction = MoveNext;
TaskTracker.TrackActiveTask(this, 3);
} }
public TResult Current { get; private set; } public TResult Current { get; private set; }
@ -741,6 +752,7 @@ namespace Cysharp.Threading.Tasks.Linq
public UniTask DisposeAsync() public UniTask DisposeAsync()
{ {
TaskTracker.RemoveTracking(this);
return enumerator.DisposeAsync(); return enumerator.DisposeAsync();
} }
} }

View File

@ -88,6 +88,7 @@ namespace Cysharp.Threading.Tasks.Linq
this.predicate = predicate; this.predicate = predicate;
this.cancellationToken = cancellationToken; this.cancellationToken = cancellationToken;
this.moveNextAction = MoveNext; this.moveNextAction = MoveNext;
TaskTracker.TrackActiveTask(this, 3);
} }
public TSource Current { get; private set; } public TSource Current { get; private set; }
@ -165,6 +166,7 @@ namespace Cysharp.Threading.Tasks.Linq
public UniTask DisposeAsync() public UniTask DisposeAsync()
{ {
TaskTracker.RemoveTracking(this);
return enumerator.DisposeAsync(); return enumerator.DisposeAsync();
} }
} }
@ -204,6 +206,7 @@ namespace Cysharp.Threading.Tasks.Linq
this.predicate = predicate; this.predicate = predicate;
this.cancellationToken = cancellationToken; this.cancellationToken = cancellationToken;
this.moveNextAction = MoveNext; this.moveNextAction = MoveNext;
TaskTracker.TrackActiveTask(this, 3);
} }
public TSource Current { get; private set; } public TSource Current { get; private set; }
@ -281,6 +284,7 @@ namespace Cysharp.Threading.Tasks.Linq
public UniTask DisposeAsync() public UniTask DisposeAsync()
{ {
TaskTracker.RemoveTracking(this);
return enumerator.DisposeAsync(); return enumerator.DisposeAsync();
} }
} }
@ -320,6 +324,7 @@ namespace Cysharp.Threading.Tasks.Linq
this.predicate = predicate; this.predicate = predicate;
this.cancellationToken = cancellationToken; this.cancellationToken = cancellationToken;
this.moveNextAction = MoveNext; this.moveNextAction = MoveNext;
TaskTracker.TrackActiveTask(this, 3);
} }
public TSource Current { get; private set; } public TSource Current { get; private set; }
@ -410,6 +415,7 @@ namespace Cysharp.Threading.Tasks.Linq
public UniTask DisposeAsync() public UniTask DisposeAsync()
{ {
TaskTracker.RemoveTracking(this);
return enumerator.DisposeAsync(); return enumerator.DisposeAsync();
} }
} }
@ -450,6 +456,7 @@ namespace Cysharp.Threading.Tasks.Linq
this.predicate = predicate; this.predicate = predicate;
this.cancellationToken = cancellationToken; this.cancellationToken = cancellationToken;
this.moveNextAction = MoveNext; this.moveNextAction = MoveNext;
TaskTracker.TrackActiveTask(this, 3);
} }
public TSource Current { get; private set; } public TSource Current { get; private set; }
@ -540,6 +547,7 @@ namespace Cysharp.Threading.Tasks.Linq
public UniTask DisposeAsync() public UniTask DisposeAsync()
{ {
TaskTracker.RemoveTracking(this);
return enumerator.DisposeAsync(); return enumerator.DisposeAsync();
} }
} }
@ -579,6 +587,7 @@ namespace Cysharp.Threading.Tasks.Linq
this.predicate = predicate; this.predicate = predicate;
this.cancellationToken = cancellationToken; this.cancellationToken = cancellationToken;
this.moveNextAction = MoveNext; this.moveNextAction = MoveNext;
TaskTracker.TrackActiveTask(this, 3);
} }
public TSource Current { get; private set; } public TSource Current { get; private set; }
@ -669,6 +678,7 @@ namespace Cysharp.Threading.Tasks.Linq
public UniTask DisposeAsync() public UniTask DisposeAsync()
{ {
TaskTracker.RemoveTracking(this);
return enumerator.DisposeAsync(); return enumerator.DisposeAsync();
} }
} }
@ -709,6 +719,7 @@ namespace Cysharp.Threading.Tasks.Linq
this.predicate = predicate; this.predicate = predicate;
this.cancellationToken = cancellationToken; this.cancellationToken = cancellationToken;
this.moveNextAction = MoveNext; this.moveNextAction = MoveNext;
TaskTracker.TrackActiveTask(this, 3);
} }
public TSource Current { get; private set; } public TSource Current { get; private set; }
@ -799,6 +810,7 @@ namespace Cysharp.Threading.Tasks.Linq
public UniTask DisposeAsync() public UniTask DisposeAsync()
{ {
TaskTracker.RemoveTracking(this);
return enumerator.DisposeAsync(); return enumerator.DisposeAsync();
} }
} }