chore: add mutex in tasks

Author: mateof

TelegramDownloader/Models/DownloadModel.cs

@@ -7,12 +7,14 @@
 using System.ComponentModel;
 using System.Runtime.CompilerServices;
 using BlazorBootstrap;
+using System.Threading;
 
 namespace TelegramDownloader.Models
 {
 
     public class InfoDownloadTaksModel
     {
+        public Mutex mutex = new Mutex();
         public event EventHandler<InfoTaskEventArgs> EventChanged;
         public StateTask state { get; set; } = StateTask.Pending;
         public DateTime creationDate { get; set; } = DateTime.Now;
@@ -38,10 +40,11 @@ public async void AddOne(long size)
         {
             if (state == StateTask.Canceled)
                 cancelTask();
-            
+            mutex.WaitOne();
             executed += 1;
             progress = Convert.ToInt32(executed * 100 / total);
             executedSize += size;
+            mutex.ReleaseMutex();
             EventChanged?.Invoke(this, new InfoTaskEventArgs());
 
         }
@@ -66,16 +69,20 @@ public void addUpload(UploadModel um)
         {
             if (state == StateTask.Canceled)
                 return;
+            mutex.WaitOne();
             deleteNotWorkingTasks();
             currentUploads.Add(um);
+            mutex.ReleaseMutex();
         }
 
         public void addDownloads(DownloadModel dm)
         {
             if (state == StateTask.Canceled)
                 return;
+            mutex.WaitOne();
             deleteNotWorkingTasks();
             currentDownloads.Add(dm);
+            mutex.ReleaseMutex();
         }
 
         public void Retry()
@@ -159,6 +166,7 @@ public void ProgressCallback(long transmitted, long totalSize)
                 throw new Exception($"Paused {name}");
             }
             tis.addDownloadBytes(transmitted - _transmitted);
+            mutex.WaitOne();
             _transmitted = transmitted;
             _sizeString = HelperService.SizeSuffix(totalSize);
             _transmittedString = HelperService.SizeSuffix(transmitted);
@@ -172,18 +180,23 @@ public void ProgressCallback(long transmitted, long totalSize)
                 nm.sendEvent(new Notification($"Download {name} completed", "Download Completed", NotificationTypes.Success));
                 tis.CheckPendingDownloads();
             }
+            mutex.ReleaseMutex();
         }
 
         public void Cancel()
         {
+            mutex.WaitOne();
             state = StateTask.Canceled;
+            mutex.ReleaseMutex();
             tis.CheckPendingDownloads();
             EventChanged?.Invoke(this, new DownloadEventArgs());
         }
 
         public void Pause()
         {
+            mutex.WaitOne();
             state = StateTask.Paused;
+            mutex.ReleaseMutex();
             EventChanged?.Invoke(this, new DownloadEventArgs());
         }
 
@@ -195,6 +208,7 @@ public void RetryCallback()
 
     public class UploadModel
     {
+        public Mutex mutex = new Mutex();
         public event EventHandler<UploadEventArgs> EventChanged;
         public Guid id = Guid.NewGuid();
         public string action { get; set; } = "Upload";
@@ -219,6 +233,7 @@ public virtual void ProgressCallback(long transmitted, long totalSize)
             if (state == StateTask.Canceled)
                 throw new Exception($"Canceled {name}");
             tis.addUploadBytes(transmitted - _transmitted);
+            mutex.WaitOne();
             _transmitted = transmitted;
             _sizeString = HelperService.SizeSuffix(totalSize);
             _transmittedString = HelperService.SizeSuffix(transmitted);
@@ -230,6 +245,7 @@ public virtual void ProgressCallback(long transmitted, long totalSize)
                 NotificationModel nm = new NotificationModel();
                 nm.sendEvent(new Notification($"Upload {name} completed", "Upload Completed", NotificationTypes.Success));
             }
+            mutex.ReleaseMutex();
 
         }
 
@@ -240,13 +256,17 @@ public void SendNotification()
 
         public void SetState(StateTask newState)
         {
+            mutex.WaitOne();
             state = newState;
+            mutex.ReleaseMutex();
             EventChanged?.Invoke(this, new UploadEventArgs());
         }
 
         public void Cancel()
         {
+            mutex.WaitOne();
             state = StateTask.Canceled;
+            mutex.ReleaseMutex();
             EventChanged?.Invoke(this, new UploadEventArgs());
         }
 
@@ -264,6 +284,7 @@ public SplitModel(): base()
         }
         public override void ProgressCallback(long transmitted, long totalSize)
         {
+            mutex.WaitOne();
             _transmitted = transmitted;
             _sizeString = HelperService.SizeSuffix(totalSize);
             _transmittedString = HelperService.SizeSuffix(transmitted);
@@ -275,7 +296,7 @@ public override void ProgressCallback(long transmitted, long totalSize)
                 NotificationModel nm = new NotificationModel();
                 nm.sendEvent(new Notification($"Split {name} completed", "Split Completed", NotificationTypes.Success));
             }
-
+            mutex.ReleaseMutex();
         }
     }
 
@@ -287,15 +308,19 @@ public Md5Model() : base()
         }
         public virtual void Init(long size, string filename)
         {
+            mutex.WaitOne();
             _sizeString = HelperService.SizeSuffix(size);
             name = filename;
+            mutex.ReleaseMutex();
             base.InvokeEvent(new UploadEventArgs());
         }
 
         public virtual void Finish()
         {
+            mutex.WaitOne();
             state = StateTask.Completed;
             progress = 100;
+            mutex.ReleaseMutex();
             base.InvokeEvent(new UploadEventArgs());
         }
     }