DreamineThreadJob.cs

이 파일의 문서화 페이지로 가기

using Dreamine.Threading.Interfaces;
using Dreamine.Threading.Models;
 
namespace Dreamine.Threading.Services;

public sealed class DreamineThreadJob : IDreamineThreadJob
{
    private readonly Func<CancellationToken, ValueTask> _action;
    private DateTimeOffset? _lastExecutedAt;
    private long _executeCount;

    public string Name { get; }

    public DreamineThreadJobOptions Options { get; }

    public long ExecuteCount => Interlocked.Read(ref _executeCount);

    public DateTimeOffset? LastExecutedAt => _lastExecutedAt;

    public Exception? LastException { get; private set; }

    public DreamineThreadJob(
        DreamineThreadJobOptions options,
        Func<CancellationToken, ValueTask> action)
    {
        ArgumentNullException.ThrowIfNull(options);
        ArgumentNullException.ThrowIfNull(action);
 
        Options = options.Normalize();
        Name = Options.Name;
        _action = action;
    }

    public bool ShouldRun(DateTimeOffset now)
    {
        if (!Options.IsEnabled)
        {
            return false;
        }
 
        if (Options.IntervalMs == 0)
        {
            return true;
        }
 
        if (_lastExecutedAt is null)
        {
            return true;
        }
 
        var elapsedMs = (now - _lastExecutedAt.Value).TotalMilliseconds;
        return elapsedMs >= Options.IntervalMs;
    }

    public async ValueTask ExecuteAsync(CancellationToken cancellationToken)
    {
        try
        {
            await _action(cancellationToken).ConfigureAwait(false);
 
            _lastExecutedAt = DateTimeOffset.Now;
            Interlocked.Increment(ref _executeCount);
            LastException = null;
        }
        catch (Exception ex) when (ex is not OperationCanceledException)
        {
            LastException = ex;
            throw;
        }
    }
}