pengtaihao
/
UniTask
mirror of https://github.com/Cysharp/UniTask
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add overload in Progress.Create, Progress.CreateOnlyValueChanged to avoid closure allocation

pull/625/head
Kochoyume 2024-10-06 12:42:57 +09:00
parent 7c0f199fe0
commit d9f55b40a5
1 changed files with 65 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

65
src/UniTask/Assets/Plugins/UniTask/Runtime/Progress.cs
View File

@ -15,6 +15,12 @@ namespace Cysharp.Threading.Tasks
return new AnonymousProgress<T>(handler);
}
public static IProgress<T> Create<T, TState>(TState state, Action<T, TState> handler)
{
if (handler == null) return NullProgress<T>.Instance;
return new AnonymousProgress<T, TState>(handler, state);
}
public static IProgress<T> CreateOnlyValueChanged<T>(Action<T> handler, IEqualityComparer<T> comparer = null)
{
if (handler == null) return NullProgress<T>.Instance;
@ -25,6 +31,16 @@ namespace Cysharp.Threading.Tasks
#endif
}
public static IProgress<T> CreateOnlyValueChanged<T, TState>(TState state, Action<T, TState> handler, IEqualityComparer<T> comparer = null)
{
if (handler == null) return NullProgress<T>.Instance;
#if UNITY_2018_3_OR_NEWER
return new OnlyValueChangedProgress<T, TState>(handler, state, comparer ?? UnityEqualityComparer.GetDefault<T>());
#else
return new OnlyValueChangedProgress<T, TState>(handler, state, comparer ?? EqualityComparer<T>.Default);
#endif
}
sealed class NullProgress<T> : IProgress<T>
{
public static readonly IProgress<T> Instance = new NullProgress<T>();
@ -54,6 +70,23 @@ namespace Cysharp.Threading.Tasks
}
}
sealed class AnonymousProgress<T, TState> : IProgress<T>
{
readonly Action<T, TState> action;
readonly TState state;
public AnonymousProgress(Action<T, TState> action, TState state)
{
this.action = action;
this.state = state;
}
public void Report(T value)
{
action(value, state);
}
}
sealed class OnlyValueChangedProgress<T> : IProgress<T>
{
readonly Action<T> action;
@ -83,5 +116,37 @@ namespace Cysharp.Threading.Tasks
action(value);
}
}
sealed class OnlyValueChangedProgress<T, TState> : IProgress<T>
{
readonly Action<T, TState> action;
readonly TState state;
readonly IEqualityComparer<T> comparer;
bool isFirstCall;
T latestValue;
public OnlyValueChangedProgress(Action<T, TState> action, TState state, IEqualityComparer<T> comparer)
{
this.action = action;
this.state = state;
this.comparer = comparer;
this.isFirstCall = true;
}
public void Report(T value)
{
if (isFirstCall)
{
isFirstCall = false;
}
else if (comparer.Equals(value, latestValue))
{
return;
}
latestValue = value;
action(value, state);
}
}
}
}