add methods for manual cache interaction

JeremyOT · Jun 20, 2015 · 6fcbcf1 · 6fcbcf1
1 parent 9cfd5e0
commit 6fcbcf1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/TotoClient/TCDataCache.h b/TotoClient/TCDataCache.h
@@ -27,6 +27,8 @@ typedef void (^ImageCallback)(UIImage*);
 -(NSString*)cachePathForURL:(NSURL*)url;
 -(void)dataFromURL:(NSURL*)url block:(void (^)(NSData *data))block;
 -(void)dataFromURL:(NSURL*)url ignoreCache:(BOOL)ignoreCache block:(void (^)(NSData *data))block;
+-(UIImage*)cachedImageFromURL:(NSURL*)url;
+-(void)cacheImage:(UIImage*)image forURL:(NSURL*)url;
 -(void)imageFromURL:(NSURL*)url block:(void (^)(UIImage *image))block;
 -(void)imageFromURL:(NSURL*)url ignoreCache:(BOOL)ignoreCache block:(void (^)(UIImage *image))block;
 -(void)imageFromData:(NSData *)data forSourceURL:(NSURL *)url withCompletionHandler:(void(^)(UIImage *image)) completionHandler;

diff --git a/TotoClient/TCDataCache.m b/TotoClient/TCDataCache.m
@@ -109,6 +109,14 @@ -(NSData *)cachedDataForURL:(NSURL *)url cachePath:(NSString *)cachePath {
     return [NSData dataWithContentsOfFile:cachePath];
 }
 
+-(UIImage*)cachedImageFromURL:(NSURL*)url {
+    return [_cache objectForKey:url];
+}
+
+-(void)cacheImage:(UIImage*)image forURL:(NSURL*)url {
+    [_cache setObject:image forKey:url cost:[self cacheCostForImage:image]];
+}
+
 -(void)imageFromURL:(NSURL *)url block:(ImageCallback)block {
     [self imageFromURL:url ignoreCache:NO block:block];
 }
@@ -147,7 +155,7 @@ -(void)runCallbacksWithImage:(UIImage*)image forUrl:(NSURL*)url {
 }
 
 -(void)imageFromURL:(NSURL *)url ignoreCache:(BOOL)ignoreCache block:(ImageCallback)block {
-    UIImage *memoryCachedImage = [_cache objectForKey:url];
+    UIImage *memoryCachedImage = [self cachedImageFromURL:url];
     if (memoryCachedImage) {
         block(memoryCachedImage);
         return;