Merge branch '3.1.0'

Particular · May 21, 2012 · ac97738 · ac97738
2 parents a7021b1 + 65dcee2
commit ac97738
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/src/azure/Timeout/NServiceBus.Timeout.Hosting.Azure/TimeoutLogic/TimeoutPersister.cs b/src/azure/Timeout/NServiceBus.Timeout.Hosting.Azure/TimeoutLogic/TimeoutPersister.cs
@@ -24,11 +24,11 @@ IEnumerable<TimeoutData> IPersistTimeouts.GetAll()
                         SagaId = c.SagaId,
                         State = Deserialize(c.StateAddress),
                         Time = c.Time,
-                        CorrelationId = c.CorrelationId
+                        CorrelationId = c.CorrelationId,
+                        Id = c.RowKey
                     });
         }
 
-
         void IPersistTimeouts.Add(TimeoutData timeout)
         {
             var hash = Hash(timeout);
@@ -52,10 +52,8 @@ void IPersistTimeouts.Add(TimeoutData timeout)
 
         public void Remove(string timeoutId)
         {
-            //todo: we need to make this work using the Id of the timeout data
-            var hash = Hash(new TimeoutData());
             TimeoutDataEntity timeoutDataEntity;
-            if (!TryGetTimeoutData(hash, out timeoutDataEntity)) return;
+            if (!TryGetTimeoutData(timeoutId, out timeoutDataEntity)) return;
 
             RemoveSerializedState(timeoutDataEntity.StateAddress);
             context.DeleteObject(timeoutDataEntity);
@@ -87,9 +85,8 @@ public void ClearTimeoutsFor(Guid sagaId)
 
         public bool CanSend(TimeoutData data)
         {
-            var hash = Hash(data);
             TimeoutDataEntity timeoutDataEntity;
-            if (!TryGetTimeoutData(hash, out timeoutDataEntity)) return false;
+            if (!TryGetTimeoutData(data.Id, out timeoutDataEntity)) return false;
 
             var leaseBlob = container.GetBlockBlobReference(timeoutDataEntity.StateAddress);