From a8e9eded60e4d33c4d9e91d06505ef175b0a620f Mon Sep 17 00:00:00 2001
From: sakno <roman.sakno@gmail.com>
Date: Tue, 15 Oct 2024 18:56:49 +0300
Subject: [PATCH] Skip redundant dispose checks

---
 .../Threading/AsyncExclusiveLock.cs           |  7 +++++-
 .../Threading/AsyncReaderWriterLock.cs        | 24 +++++++++++++------
 2 files changed, 23 insertions(+), 8 deletions(-)

diff --git a/src/DotNext.Threading/Threading/AsyncExclusiveLock.cs b/src/DotNext.Threading/Threading/AsyncExclusiveLock.cs
index a28a93d72..a272c06d9 100644
--- a/src/DotNext.Threading/Threading/AsyncExclusiveLock.cs
+++ b/src/DotNext.Threading/Threading/AsyncExclusiveLock.cs
@@ -86,6 +86,11 @@ private void OnCompleted(DefaultWaitNode node)
     public bool TryAcquire()
     {
         ObjectDisposedException.ThrowIf(IsDisposed, this);
+        return TryAcquireCore();
+    }
+
+    private bool TryAcquireCore()
+    {
         Monitor.Enter(SyncRoot);
         var result = TryAcquire(ref manager);
         Monitor.Exit(SyncRoot);
@@ -104,7 +109,7 @@ public bool TryAcquire()
     public bool TryAcquire(TimeSpan timeout)
     {
         ObjectDisposedException.ThrowIf(IsDisposed, this);
-        return timeout == TimeSpan.Zero ? TryAcquire() : TryAcquire(new Timeout(timeout), ref manager);
+        return timeout == TimeSpan.Zero ? TryAcquireCore() : TryAcquire(new Timeout(timeout), ref manager);
     }
 
     /// <summary>
diff --git a/src/DotNext.Threading/Threading/AsyncReaderWriterLock.cs b/src/DotNext.Threading/Threading/AsyncReaderWriterLock.cs
index a0154a1b1..b73939370 100644
--- a/src/DotNext.Threading/Threading/AsyncReaderWriterLock.cs
+++ b/src/DotNext.Threading/Threading/AsyncReaderWriterLock.cs
@@ -284,8 +284,12 @@ public LockStamp TryOptimisticRead()
     /// </summary>
     /// <returns><see langword="true"/> if lock is taken successfully; otherwise, <see langword="false"/>.</returns>
     /// <exception cref="ObjectDisposedException">This object has been disposed.</exception>
-    public bool TryEnterReadLock() => TryEnter<ReadLockManager>();
-    
+    public bool TryEnterReadLock()
+    {
+        ObjectDisposedException.ThrowIf(IsDisposed, this);
+        return TryEnter<ReadLockManager>();
+    }
+
     /// <summary>
     /// Tries to obtain reader lock synchronously.
     /// </summary>
@@ -365,8 +369,12 @@ public bool TryEnterWriteLock(in LockStamp stamp)
     /// </summary>
     /// <returns><see langword="true"/> if lock is taken successfully; otherwise, <see langword="false"/>.</returns>
     /// <exception cref="ObjectDisposedException">This object has been disposed.</exception>
-    public bool TryEnterWriteLock() => TryEnter<WriteLockManager>();
-    
+    public bool TryEnterWriteLock()
+    {
+        ObjectDisposedException.ThrowIf(IsDisposed, this);
+        return TryEnter<WriteLockManager>();
+    }
+
     /// <summary>
     /// Tries to obtain writer lock synchronously.
     /// </summary>
@@ -425,13 +433,15 @@ public ValueTask EnterWriteLockAsync(TimeSpan timeout, CancellationToken token =
     /// </summary>
     /// <returns><see langword="true"/> if lock is taken successfully; otherwise, <see langword="false"/>.</returns>
     /// <exception cref="ObjectDisposedException">This object has been disposed.</exception>
-    public bool TryUpgradeToWriteLock() => TryEnter<UpgradeManager>();
+    public bool TryUpgradeToWriteLock()
+    {
+        ObjectDisposedException.ThrowIf(IsDisposed, this);
+        return TryEnter<UpgradeManager>();
+    }
 
     private bool TryEnter<TLockManager>()
         where TLockManager : struct, ILockManager<WaitNode>
     {
-        ObjectDisposedException.ThrowIf(IsDisposed, this);
-
         Monitor.Enter(SyncRoot);
         var result = TryAcquire(ref GetLockManager<TLockManager>());
         Monitor.Exit(SyncRoot);