Add subscript to MemoryCache

soffes · May 16, 2016 · 5828a2b · 5828a2b
1 parent e449240
commit 5828a2b
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/Sources/MemoryCache.swift b/Sources/MemoryCache.swift
@@ -64,4 +64,21 @@ public final class MemoryCache<T>: Cache {
 		cache.removeAllObjects()
 		completion?()
 	}
+
+
+	// MARK: - Synchronous
+
+	public subscript(key: String) -> T? {
+		get {
+			return (cache.objectForKey(key) as? Box<T>)?.value
+		}
+
+		set(newValue) {
+			if let newValue = newValue {
+				cache.setObject(Box(newValue), forKey: key)
+			} else {
+				cache.removeObjectForKey(key)
+			}
+		}
+	}
 }