From 8fcb6978447cb80d55272a45501471d998982891 Mon Sep 17 00:00:00 2001
From: "Frederick F. Kautz IV" <fkautz@alumni.cmu.edu>
Date: Wed, 29 Apr 2015 13:29:04 -0700
Subject: [PATCH] Simplifying memory sleep logic

---
 pkg/storage/drivers/memory/memory.go | 40 +++++++++++++---------------
 1 file changed, 18 insertions(+), 22 deletions(-)

diff --git a/pkg/storage/drivers/memory/memory.go b/pkg/storage/drivers/memory/memory.go
index 89b4768a3..bbebc0fb1 100644
--- a/pkg/storage/drivers/memory/memory.go
+++ b/pkg/storage/drivers/memory/memory.go
@@ -484,34 +484,30 @@ func (memory *memoryDriver) expireObjects() {
 			return
 		}
 		var keysToRemove []string
-		memory.lock.RLock()
-		var earliest time.Time
-		empty := true
-		for key, object := range memory.objectMetadata {
-			if empty {
-				empty = false
-			}
-			if time.Now().Add(-memory.expiration).After(object.metadata.Created) {
-				keysToRemove = append(keysToRemove, key)
-			} else {
-				if object.metadata.Created.Before(earliest) {
-					earliest = object.metadata.Created
+		if len(memory.objectMetadata) > 0 {
+			memory.lock.RLock()
+			var earliest time.Time
+			for key, object := range memory.objectMetadata {
+				if time.Now().Add(-memory.expiration).After(object.metadata.Created) {
+					keysToRemove = append(keysToRemove, key)
+				} else {
+					if object.metadata.Created.Before(earliest) {
+						earliest = object.metadata.Created
+					}
 				}
 			}
-		}
-		memory.lock.RUnlock()
-		memory.lock.Lock()
-		for _, key := range keysToRemove {
-			memory.objects.Remove(key)
-		}
-		memory.lock.Unlock()
-		if empty {
-			time.Sleep(memory.expiration)
-		} else {
+			memory.lock.RUnlock()
+			memory.lock.Lock()
+			for _, key := range keysToRemove {
+				memory.objects.Remove(key)
+			}
+			memory.lock.Unlock()
 			sleepFor := earliest.Sub(time.Now())
 			if sleepFor > 0 {
 				time.Sleep(sleepFor)
 			}
+		} else {
+			time.Sleep(memory.expiration)
 		}
 	}
 }