Horizon – Diff between revs 1 and 5

Subversion Repositories:

?pathlinks?

Rev 1 – Rev 6 – Go to most recent revision – Details – Blame – Last modification – View Log

 using System.Linq;
 using System.Net;
 using System.Reflection;
 using System.Threading;
 using System.Threading.Tasks;
+using System.Threading.Tasks.Dataflow;
 using System.Windows.Forms;
+using System.Windows.Shapes;
 using Horizon.Database;
 using Horizon.Snapshots;
 using Horizon.Utilities;
 using Horizon.Utilities.Serialization;
 using NetSparkleUpdater;
 using NetSparkleUpdater.SignatureVerifiers;
 using NetSparkleUpdater.UI.WinForms;
 using Serilog;
 using TrackedFolders;
 using CaptureMode = Configuration.CaptureMode;
+using Path = System.IO.Path;
 namespace Horizon
+{
     public partial class MainForm : Form
                 .WriteTo.Conditional(condition => MemorySinkEnabled, configureSink => configureSink.Sink(_memorySink))
                 .WriteTo.File(Path.Combine(Constants.UserApplicationDirectory, "Logs", $"{Constants.AssemblyName}.log"),
                     rollingInterval: RollingInterval.Day)
                 .CreateLogger();
-            _snapshotDatabase = new SnapshotDatabase();
+            _snapshotDatabase = new SnapshotDatabase(_cancellationToken);
             _snapshotDatabase.SnapshotRevert += SnapshotDatabase_SnapshotRevert;
             _snapshotDatabase.SnapshotCreate += SnapshotDatabase_SnapshotCreate;
                     color = folder.Color;
+                }
                 if (_changedFiles.Contains(e.FullPath))
                 {
-                    _changedFilesContinuation.Schedule(delay, () => TakeSnapshots(color), _cancellationToken);
+                    _changedFilesContinuation.Schedule(delay, async () => await TakeSnapshots(color, _cancellationToken), _cancellationToken);
                     return;
+                }
                 _changedFiles.Add(e.FullPath);
-                _changedFilesContinuation.Schedule(delay, () => TakeSnapshots(color), _cancellationToken);
+                _changedFilesContinuation.Schedule(delay, async () => await TakeSnapshots(color, _cancellationToken), _cancellationToken);
+            }
             catch (Exception exception)
                     break;
+            }
         }
-        private async void TakeSnapshots(Color color)
+        private async Task TakeSnapshots(Color color, CancellationToken cancellationToken)
         {
-            await _changedFilesLock.WaitAsync(_cancellationToken);
+            var bufferBlock = new BufferBlock<string>(new DataflowBlockOptions() {CancellationToken = cancellationToken});
+            var actionBlock = new ActionBlock<string>(async path =>
-            try
+            {
                 // In case files have vanished strictly due to the time specified by the tracked folders delay.
-                foreach (var path in _changedFiles)
-                {
-                    // In case files have vanished strictly due to the time specified by the tracked folders delay.
-                    if (!File.Exists(path))
+                if (!File.Exists(path))
-                    {
+                {
-                        Log.Warning("File vanished after tracked folder delay.", path);
+                    Log.Warning("File vanished after tracked folder delay.", path);
-                        continue;
+                    return;
-                    }
+                }
-                    try
+                try
-                    {
+                {
-                        var fileName = Path.GetFileName(path);
+                    var fileName = System.IO.Path.GetFileName(path);
-                        var screenCapture = ScreenCapture.Capture((Utilities.CaptureMode)Configuration.CaptureMode);
+                    var screenCapture = ScreenCapture.Capture((Utilities.CaptureMode)Configuration.CaptureMode);
+                    await _snapshotDatabase.CreateSnapshot(fileName, path, screenCapture, color,
                         _cancellationToken);
-                        await _snapshotDatabase.CreateSnapshot(fileName, path, screenCapture, color,
-                            _cancellationToken);
-                    }
+                }
-                    catch (SQLiteException exception)
                 catch (SQLiteException exception)
+                {
-                        if (exception.ResultCode == SQLiteErrorCode.Constraint)
+                    if (exception.ResultCode == SQLiteErrorCode.Constraint)
+                    {
+                        Log.Information(exception, "Snapshot already exists.");
+                    }
+                }
+                catch (Exception exception)
+                {
+                    Log.Error(exception, "Could not take snapshot.", path);
+                }
+            });
+            using (var snapshotLink =
+                   bufferBlock.LinkTo(actionBlock, new DataflowLinkOptions() { PropagateCompletion = true }))
+            {
-                            Log.Information(exception, "Snapshot already exists.");
+                await _changedFilesLock.WaitAsync(_cancellationToken);
                 try
+                {
+                    foreach (var path in _changedFiles)
+                    {
+                        await bufferBlock.SendAsync(path, cancellationToken);
+                    }
+                    bufferBlock.Complete();
+                    await bufferBlock.Completion;
+                }
+                catch (Exception exception)
+                {
+                    Log.Error(exception, "Could not take snapshots.");
+                }
-                    catch (Exception exception)
-                    {
-                        Log.Error(exception, "Could not take snapshot.", path);
-                    }
-                }
-            }
-            catch (Exception exception)
-            {
-                Log.Error(exception, "Could not take snapshots.");
                 finally
             finally
                     _changedFiles.Clear();
-                _changedFiles.Clear();
+                    _changedFilesLock.Release();