Make sure and check if we are over the wait time after we sleep in Get() to ensure we exit quickly instead of doing another needless findAndModify.

With lots of Queues objects and running in loops the extra query is expensive.

Author: Jonathan Gaillard

DominionEnterprises.Mongo.Tests/QueueTests.cs

@@ -280,12 +280,21 @@ public void GetWait()
         {
             var start = DateTime.Now;
 
-            queue.Get(new QueryDocument(), TimeSpan.MaxValue, TimeSpan.FromMilliseconds(200), TimeSpan.MinValue, false);
+            queue.Get(new QueryDocument(), TimeSpan.MaxValue, TimeSpan.FromMilliseconds(200), TimeSpan.FromMilliseconds(201), false);
 
             var end = DateTime.Now;
 
             Assert.IsTrue(end - start >= TimeSpan.FromMilliseconds(200));
             Assert.IsTrue(end - start < TimeSpan.FromMilliseconds(400));
+
+            start = DateTime.Now;
+
+            queue.Get(new QueryDocument(), TimeSpan.MaxValue, TimeSpan.FromMilliseconds(200), TimeSpan.MinValue, false);
+
+            end = DateTime.Now;
+
+            Assert.IsTrue(end - start >= TimeSpan.FromMilliseconds(200));
+            Assert.IsTrue(end - start < TimeSpan.FromMilliseconds(400));
         }
 
         [Test]

DominionEnterprises.Mongo/Queue.cs

@@ -248,6 +248,9 @@ public BsonDocument Get(QueryDocument query, TimeSpan resetRunning, TimeSpan wai
 
                     Thread.Sleep(poll);
                 }
+
+                if (DateTime.UtcNow >= end)
+                    return null;
             }
         }
         #endregion