80 lines
2.2 KiB
C#
80 lines
2.2 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Umbraco.Core.Hosting;
|
|
|
|
namespace Umbraco.Core
|
|
{
|
|
/// <summary>
|
|
/// Provides a simple implementation of <see cref="IMainDom"/>.
|
|
/// </summary>
|
|
public class SimpleMainDom : IMainDom, IDisposable
|
|
{
|
|
private readonly object _locko = new object();
|
|
private readonly List<KeyValuePair<int, Action>> _callbacks = new List<KeyValuePair<int, Action>>();
|
|
private bool _isStopping;
|
|
private bool _disposedValue;
|
|
|
|
/// <inheritdoc />
|
|
public bool IsMainDom { get; private set; } = true;
|
|
|
|
// always acquire
|
|
public bool Acquire(IApplicationShutdownRegistry hostingEnvironment) => true;
|
|
|
|
/// <inheritdoc />
|
|
public bool Register(Action install, Action release, int weight = 100)
|
|
{
|
|
lock (_locko)
|
|
{
|
|
if (_isStopping) return false;
|
|
install?.Invoke();
|
|
if (release != null)
|
|
_callbacks.Add(new KeyValuePair<int, Action>(weight, release));
|
|
return true;
|
|
}
|
|
}
|
|
|
|
public void Stop()
|
|
{
|
|
lock (_locko)
|
|
{
|
|
if (_isStopping) return;
|
|
if (IsMainDom == false) return; // probably not needed
|
|
_isStopping = true;
|
|
}
|
|
|
|
try
|
|
{
|
|
foreach (var callback in _callbacks.OrderBy(x => x.Key).Select(x => x.Value))
|
|
{
|
|
callback(); // no timeout on callbacks
|
|
}
|
|
}
|
|
finally
|
|
{
|
|
// in any case...
|
|
IsMainDom = false;
|
|
}
|
|
}
|
|
|
|
protected virtual void Dispose(bool disposing)
|
|
{
|
|
if (!_disposedValue)
|
|
{
|
|
if (disposing)
|
|
{
|
|
Stop();
|
|
}
|
|
_disposedValue = true;
|
|
}
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
// Do not change this code. Put cleanup code in 'Dispose(bool disposing)' method
|
|
Dispose(disposing: true);
|
|
GC.SuppressFinalize(this);
|
|
}
|
|
}
|
|
}
|