init

Common/DTS.Common.Import/DatabaseLocks/LockImportSensors.cs

@@ -0,0 +1,130 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using DTS.Common.Classes.Locking;
+using DTS.Common.Import.Interfaces;
+using DTS.Common.SharedResource.Strings;
+using DTS.Common.Storage;
+using DTS.Common.Utilities.Logging;
+using DTS.SensorDB;
+using DTS.Slice.Users;
+
+//FB 36740
+namespace DTS.Common.Import.DatabaseLocks
+{
+    public class LockImportSensors : ILockImport
+    {
+        private readonly User _currentUser;
+        private readonly double _strandedLockTimeoutMinutes;
+
+        /// <summary>
+        /// list of all sensors the user has locked
+        /// </summary>
+        private readonly List<LockRecord> _lockedSensors = new List<LockRecord>();
+        /// <summary>
+        /// list of all sensors the user attempted to lock but another user has the lock
+        /// </summary>
+        private readonly List<LockRecord> _contendedSensors = new List<LockRecord>();
+        public bool Contended { get => _contendedSensors.Any(); }
+
+        public LockImportSensors(User currentUser, double strandedLockTimeoutMinutes)
+        {
+            _currentUser = currentUser;
+            _strandedLockTimeoutMinutes = strandedLockTimeoutMinutes;
+        }
+
+        public void FreeLock(ref ImportObject importObject)
+        {
+            if (!_lockedSensors.Any())
+            {
+                return;
+            }
+            foreach (var record in _lockedSensors)
+            {
+                LockManager.FreeLock(record.ItemId, _currentUser.UserName, _currentUser.Id, out var lockError, record.ItemCategory);
+            }
+        }
+
+        public void SetLock(ref ImportObject importObject, ref StringBuilder message)
+        {
+            var sensorLookup = SensorsCollection.SensorsList.GetAllSensors(true).ToDictionary(s => s.SerialNumber);
+
+            _lockedSensors.Clear();
+            _contendedSensors.Clear();
+
+            foreach (var sensorSerialNumber in importObject.Sensors().Select(s => s.SerialNumber))
+            {
+                if (!sensorLookup.ContainsKey(sensorSerialNumber)) { continue; } //no lock needed
+                try
+                {
+                    var wasAbleToLock = LockManager.LockItem(sensorSerialNumber, sensorLookup[sensorSerialNumber].DatabaseId, LockManager.ItemCategories.Sensor, _currentUser.UserName,
+                        _currentUser.Id, out var existingLock, out var lockError);
+                    if (!wasAbleToLock)
+                    {
+                        var lockOutOfDate = existingLock.LastUpdated.AddMinutes(_strandedLockTimeoutMinutes) < DateTime.Now;
+                        var IHaveTheLock = existingLock.LockingUserName == _currentUser.UserName && existingLock.LockingMachineName == Environment.MachineName;
+
+                        if (lockOutOfDate || IHaveTheLock)
+                        {
+                            //lock is expired (or already belongs to us), either way we can claim it
+                            LockManager.FreeLock(existingLock.ItemId, _currentUser.UserName, _currentUser.Id, out lockError, existingLock.ItemCategory);
+                            LockManager.LockItem(existingLock.ItemKey, existingLock.ItemId, LockManager.ItemCategories.Sensor, _currentUser.UserName,
+                                _currentUser.Id, out existingLock, out lockError);
+                            _lockedSensors.Add(existingLock);
+                        }
+                        else
+                        {
+                            //lock is not expired, we need admin + confirmation to steal the lock
+                            _contendedSensors.Add(existingLock);
+                        }
+                    }
+                    else
+                    {
+                        _lockedSensors.Add(existingLock); //successfully locked
+                    }
+                }
+                catch (Exception ex) { APILogger.Log(ex); }
+            }
+            if (!_contendedSensors.Any())
+            {
+                return;
+            }
+
+            message.AppendLine(StringResources.ImportTestSetup_TestsLocked);
+            foreach (var contended in _contendedSensors)
+            {
+                message.AppendLine($"{contended.ItemKey} by {contended.LockingUserName} on {contended.LockingMachineName}");
+            }
+
+        }
+
+        public bool StealLock(bool proceed)
+        {
+            if (!_contendedSensors.Any())
+            {
+                return true;
+            }
+
+            if (!_currentUser.IsAdmin)
+            {
+                return false;
+            }
+
+            //steal locks if user oks it
+            if (!proceed)
+            {
+                return false;
+            }
+
+            foreach (var contendedSensor in _contendedSensors)
+            {
+                LockManager.FreeLock(contendedSensor.ItemId, _currentUser.UserName, _currentUser.Id, out var lockError, contendedSensor.ItemCategory);
+                LockManager.LockItem(contendedSensor.ItemKey, contendedSensor.ItemId, LockManager.ItemCategories.Sensor, _currentUser.UserName, _currentUser.Id, out var newLockRecord, out lockError);
+                _lockedSensors.Add(newLockRecord);
+            }
+
+            return true;
+        }
+    }
+}