naps2/NAPS2.Escl.Server/EsclServerState.cs

using System.Diagnostics.CodeAnalysis;

namespace NAPS2.Escl.Server;

internal class EsclServerState
{
    private const int CLEANUP_INTERVAL = 10_000;

    private Timer? _cleanupTimer;
    private readonly Dictionary<string, JobInfo> _jobDict = new();

    public bool IsProcessing { get; set; }

    public IEnumerable<JobInfo> Jobs => _jobDict.Values.ToList();

    public void AddJob(JobInfo jobInfo)
    {
        lock (this)
        {
            _jobDict.Add(jobInfo.Id, jobInfo);
            _cleanupTimer ??= new Timer(Cleanup, null, CLEANUP_INTERVAL, CLEANUP_INTERVAL);
        }
    }

    public bool TryGetJob(string id,  [MaybeNullWhen(false)] out JobInfo jobInfo)
    {
        lock (this)
        {
            return _jobDict.TryGetValue(id, out jobInfo);
        }
    }

    private void Cleanup(object? state)
    {
        lock (this)
        {
            foreach (var jobInfo in Jobs)
            {
                if (jobInfo.IsEligibleForCleanup)
                {
                    _jobDict.Remove(jobInfo.Id);
                    jobInfo.Job.Dispose();
                    if (_jobDict.Count == 0)
                    {
                        _cleanupTimer?.Dispose();
                        _cleanupTimer = null;
                    }
                }
            }
        }
    }
}