Nikita Tsukanov
4 years ago
9 changed files with 151 additions and 395 deletions
@ -1,300 +0,0 @@ |
|||||
using System; |
|
||||
using System.Collections.Generic; |
|
||||
using System.Linq; |
|
||||
using System.Threading; |
|
||||
using System.Threading.Tasks; |
|
||||
using Avalonia.Platform; |
|
||||
|
|
||||
namespace Avalonia.Threading |
|
||||
{ |
|
||||
/// <summary>
|
|
||||
/// A main loop in a <see cref="Dispatcher"/>.
|
|
||||
/// </summary>
|
|
||||
internal class JobRunner |
|
||||
{ |
|
||||
private IPlatformThreadingInterface? _platform; |
|
||||
|
|
||||
private readonly Queue<IJob>[] _queues = Enumerable.Range(0, (int)DispatcherPriority.MaxValue + 1) |
|
||||
.Select(_ => new Queue<IJob>()).ToArray(); |
|
||||
|
|
||||
public JobRunner(IPlatformThreadingInterface? platform) |
|
||||
{ |
|
||||
_platform = platform; |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Runs continuations pushed on the loop.
|
|
||||
/// </summary>
|
|
||||
/// <param name="priority">Priority to execute jobs for. Pass null if platform doesn't have internal priority system</param>
|
|
||||
public void RunJobs(DispatcherPriority? priority) |
|
||||
{ |
|
||||
var minimumPriority = priority ?? DispatcherPriority.MinimumActiveValue; |
|
||||
while (true) |
|
||||
{ |
|
||||
var job = GetNextJob(minimumPriority); |
|
||||
if (job == null) |
|
||||
return; |
|
||||
|
|
||||
job.Run(); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Invokes a method on the main loop.
|
|
||||
/// </summary>
|
|
||||
/// <param name="action">The method.</param>
|
|
||||
/// <param name="priority">The priority with which to invoke the method.</param>
|
|
||||
/// <returns>A task that can be used to track the method's execution.</returns>
|
|
||||
public Task InvokeAsync(Action action, DispatcherPriority priority) |
|
||||
{ |
|
||||
var job = new Job(action, priority, false); |
|
||||
AddJob(job); |
|
||||
return job.Task!; |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Invokes a method on the main loop.
|
|
||||
/// </summary>
|
|
||||
/// <param name="function">The method.</param>
|
|
||||
/// <param name="priority">The priority with which to invoke the method.</param>
|
|
||||
/// <returns>A task that can be used to track the method's execution.</returns>
|
|
||||
public Task<TResult> InvokeAsync<TResult>(Func<TResult> function, DispatcherPriority priority) |
|
||||
{ |
|
||||
var job = new JobWithResult<TResult>(function, priority); |
|
||||
AddJob(job); |
|
||||
return job.Task; |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Post action that will be invoked on main thread
|
|
||||
/// </summary>
|
|
||||
/// <param name="action">The method.</param>
|
|
||||
///
|
|
||||
/// <param name="priority">The priority with which to invoke the method.</param>
|
|
||||
internal void Post(Action action, DispatcherPriority priority) |
|
||||
{ |
|
||||
AddJob(new Job(action, priority, true)); |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Post action that will be invoked on main thread
|
|
||||
/// </summary>
|
|
||||
/// <param name="action">The method to call.</param>
|
|
||||
/// <param name="parameter">The parameter of method to call.</param>
|
|
||||
/// <param name="priority">The priority with which to invoke the method.</param>
|
|
||||
internal void Post(SendOrPostCallback action, object? parameter, DispatcherPriority priority) |
|
||||
{ |
|
||||
AddJob(new JobWithArg(action, parameter, priority, true)); |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Allows unit tests to change the platform threading interface.
|
|
||||
/// </summary>
|
|
||||
internal void UpdateServices() |
|
||||
{ |
|
||||
_platform = AvaloniaLocator.Current.GetService<IPlatformThreadingInterface>(); |
|
||||
} |
|
||||
|
|
||||
private void AddJob(IJob job) |
|
||||
{ |
|
||||
bool needWake; |
|
||||
var queue = _queues[(int)job.Priority]; |
|
||||
lock (queue) |
|
||||
{ |
|
||||
needWake = queue.Count == 0; |
|
||||
queue.Enqueue(job); |
|
||||
} |
|
||||
if (needWake) |
|
||||
_platform?.Signal(job.Priority); |
|
||||
} |
|
||||
|
|
||||
private IJob? GetNextJob(DispatcherPriority minimumPriority) |
|
||||
{ |
|
||||
for (int c = (int)DispatcherPriority.MaxValue; c >= (int)minimumPriority; c--) |
|
||||
{ |
|
||||
var q = _queues[c]; |
|
||||
lock (q) |
|
||||
{ |
|
||||
if (q.Count > 0) |
|
||||
return q.Dequeue(); |
|
||||
} |
|
||||
} |
|
||||
return null; |
|
||||
} |
|
||||
|
|
||||
public bool HasJobsWithPriority(DispatcherPriority minimumPriority) |
|
||||
{ |
|
||||
for (int c = (int)minimumPriority; c < (int)DispatcherPriority.MaxValue; c++) |
|
||||
{ |
|
||||
var q = _queues[c]; |
|
||||
lock (q) |
|
||||
{ |
|
||||
if (q.Count > 0) |
|
||||
return true; |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
return false; |
|
||||
} |
|
||||
|
|
||||
private interface IJob |
|
||||
{ |
|
||||
/// <summary>
|
|
||||
/// Gets the job priority.
|
|
||||
/// </summary>
|
|
||||
DispatcherPriority Priority { get; } |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Runs the job.
|
|
||||
/// </summary>
|
|
||||
void Run(); |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// A job to run.
|
|
||||
/// </summary>
|
|
||||
private sealed class Job : IJob |
|
||||
{ |
|
||||
/// <summary>
|
|
||||
/// The method to call.
|
|
||||
/// </summary>
|
|
||||
private readonly Action _action; |
|
||||
/// <summary>
|
|
||||
/// The task completion source.
|
|
||||
/// </summary>
|
|
||||
private readonly TaskCompletionSource<object?>? _taskCompletionSource; |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Initializes a new instance of the <see cref="Job"/> class.
|
|
||||
/// </summary>
|
|
||||
/// <param name="action">The method to call.</param>
|
|
||||
/// <param name="priority">The job priority.</param>
|
|
||||
/// <param name="throwOnUiThread">Do not wrap exception in TaskCompletionSource</param>
|
|
||||
public Job(Action action, DispatcherPriority priority, bool throwOnUiThread) |
|
||||
{ |
|
||||
_action = action; |
|
||||
Priority = priority; |
|
||||
_taskCompletionSource = throwOnUiThread ? null : new TaskCompletionSource<object?>(); |
|
||||
} |
|
||||
|
|
||||
/// <inheritdoc/>
|
|
||||
public DispatcherPriority Priority { get; } |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// The task.
|
|
||||
/// </summary>
|
|
||||
public Task? Task => _taskCompletionSource?.Task; |
|
||||
|
|
||||
/// <inheritdoc/>
|
|
||||
void IJob.Run() |
|
||||
{ |
|
||||
if (_taskCompletionSource == null) |
|
||||
{ |
|
||||
_action(); |
|
||||
return; |
|
||||
} |
|
||||
try |
|
||||
{ |
|
||||
_action(); |
|
||||
_taskCompletionSource.SetResult(null); |
|
||||
} |
|
||||
catch (Exception e) |
|
||||
{ |
|
||||
_taskCompletionSource.SetException(e); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// A typed job to run.
|
|
||||
/// </summary>
|
|
||||
private sealed class JobWithArg : IJob |
|
||||
{ |
|
||||
private readonly SendOrPostCallback _action; |
|
||||
private readonly object? _parameter; |
|
||||
private readonly TaskCompletionSource<bool>? _taskCompletionSource; |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Initializes a new instance of the <see cref="Job"/> class.
|
|
||||
/// </summary>
|
|
||||
/// <param name="action">The method to call.</param>
|
|
||||
/// <param name="parameter">The parameter of method to call.</param>
|
|
||||
/// <param name="priority">The job priority.</param>
|
|
||||
/// <param name="throwOnUiThread">Do not wrap exception in TaskCompletionSource</param>
|
|
||||
|
|
||||
public JobWithArg(SendOrPostCallback action, object? parameter, DispatcherPriority priority, bool throwOnUiThread) |
|
||||
{ |
|
||||
_action = action; |
|
||||
_parameter = parameter; |
|
||||
Priority = priority; |
|
||||
_taskCompletionSource = throwOnUiThread ? null : new TaskCompletionSource<bool>(); |
|
||||
} |
|
||||
|
|
||||
/// <inheritdoc/>
|
|
||||
public DispatcherPriority Priority { get; } |
|
||||
|
|
||||
/// <inheritdoc/>
|
|
||||
void IJob.Run() |
|
||||
{ |
|
||||
if (_taskCompletionSource == null) |
|
||||
{ |
|
||||
_action(_parameter); |
|
||||
return; |
|
||||
} |
|
||||
try |
|
||||
{ |
|
||||
_action(_parameter); |
|
||||
_taskCompletionSource.SetResult(default); |
|
||||
} |
|
||||
catch (Exception e) |
|
||||
{ |
|
||||
_taskCompletionSource.SetException(e); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// A job to run thath return value.
|
|
||||
/// </summary>
|
|
||||
/// <typeparam name="TResult">Type of job result</typeparam>
|
|
||||
private sealed class JobWithResult<TResult> : IJob |
|
||||
{ |
|
||||
private readonly Func<TResult> _function; |
|
||||
private readonly TaskCompletionSource<TResult> _taskCompletionSource; |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// Initializes a new instance of the <see cref="Job"/> class.
|
|
||||
/// </summary>
|
|
||||
/// <param name="function">The method to call.</param>
|
|
||||
/// <param name="priority">The job priority.</param>
|
|
||||
public JobWithResult(Func<TResult> function, DispatcherPriority priority) |
|
||||
{ |
|
||||
_function = function; |
|
||||
Priority = priority; |
|
||||
_taskCompletionSource = new TaskCompletionSource<TResult>(); |
|
||||
} |
|
||||
|
|
||||
/// <inheritdoc/>
|
|
||||
public DispatcherPriority Priority { get; } |
|
||||
|
|
||||
/// <summary>
|
|
||||
/// The task.
|
|
||||
/// </summary>
|
|
||||
public Task<TResult> Task => _taskCompletionSource.Task; |
|
||||
|
|
||||
/// <inheritdoc/>
|
|
||||
void IJob.Run() |
|
||||
{ |
|
||||
try |
|
||||
{ |
|
||||
var result = _function(); |
|
||||
_taskCompletionSource.SetResult(result); |
|
||||
} |
|
||||
catch (Exception e) |
|
||||
{ |
|
||||
_taskCompletionSource.SetException(e); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
@ -0,0 +1,17 @@ |
|||||
|
using Avalonia.Threading; |
||||
|
|
||||
|
namespace Avalonia.Utilities; |
||||
|
|
||||
|
public class DispatcherTimerHelper |
||||
|
{ |
||||
|
|
||||
|
} |
||||
|
|
||||
|
public static class DispatcherTimerUtils |
||||
|
{ |
||||
|
public static void ForceFire(this DispatcherTimer timer) |
||||
|
{ |
||||
|
timer.Promote(); |
||||
|
timer.Dispatcher.RunJobs(); |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue