1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
using System;
using System.Diagnostics;
using System.Linq;
namespace Jellyfin.Database.Implementations;
/// <summary>
/// Wrapper for progress reporting on Partition helpers.
/// </summary>
/// <typeparam name="TEntity">The entity to load.</typeparam>
public class ProgressablePartitionReporting<TEntity>
{
private readonly IOrderedQueryable<TEntity> _source;
private readonly Stopwatch _partitionTime = new();
private readonly Stopwatch _itemTime = new();
internal ProgressablePartitionReporting(IOrderedQueryable<TEntity> source)
{
_source = source;
}
internal Action<TEntity, int, int>? OnBeginItem { get; set; }
internal Action<int>? OnBeginPartition { get; set; }
internal Action<TEntity, int, int, TimeSpan>? OnEndItem { get; set; }
internal Action<int, TimeSpan>? OnEndPartition { get; set; }
internal IOrderedQueryable<TEntity> Source => _source;
internal void BeginItem(TEntity entity, int iteration, int itemIndex)
{
_itemTime.Restart();
OnBeginItem?.Invoke(entity, iteration, itemIndex);
}
internal void BeginPartition(int iteration)
{
_partitionTime.Restart();
OnBeginPartition?.Invoke(iteration);
}
internal void EndItem(TEntity entity, int iteration, int itemIndex)
{
OnEndItem?.Invoke(entity, iteration, itemIndex, _itemTime.Elapsed);
}
internal void EndPartition(int iteration)
{
OnEndPartition?.Invoke(iteration, _partitionTime.Elapsed);
}
}
|
