AFNetworking源码中的读写锁

AF网络库里有一个叫做AFAutoPurgingImageCache的类,用来做网络图片的内存缓存。核心原理就是创建一个NSMutableDictionary字典,用键值对来存储图片和相应的URL路径,并且提供了自动管理缓存空间的特性(当缓存空间不足时,自动按时间顺序清理图片)。由于这个缓存类是用来在多线程的环境中使用的,除了使用Foundation提供的锁(如NSLock)以外,源码中采用了GCD的方式,确保字典的线程安全。

串行队列保护锁

如何用GCD来进行锁操作呢?可以使用串行队列,将任务依次添加到一个线程中,每个任务(不管是读/写操作)都要等前面的任务结束后,才可以执行,执行过程中也不用担心被后面的任务抢占资源。

比如创建一个名叫synchronizationQueue的图片存取队列,注意是串行队列!

self.cachedImages = [[NSMutableDictionary alloc] init];
self.synchronizationQueue = dispatch_queue_create("com.example.imagecache", DISPATCH_QUEUE_SERIAL);

然后实现存取方法:

- (void)setImage:(UIImage *)image withURLString:(NSString *)urlString {
    dispatch_async(self.synchronizationQueue, ^{
        self.cachedImages[urlString] = image;
    });
}

- (UIImage *)imageWithURLString:(NSString *)urlString {
    __block UIImage *image = nil;
    dispatch_sync(self.synchronizationQueue, ^{
        image = self.cachedImages[urlString];
    });
    return image;
}

其中保存图片的操作可以用dispatch_async异步将任务分发到队列中,然后立即返回,不会阻塞当前线程;但是读取图片时,就需要dispatch_sync同步获取图片,所以会立即执行block拿到图片,并且作为方法返回值。于是所有图片的存取操作都放在一个线程中执行,实现了字典的线程安全。

读写锁

不过AF源码并没有用串行队列来存取图片,而是使用了并行队列。下面是真正的源码:

self.cachedImages = [[NSMutableDictionary alloc] init];

NSString *queueName = [NSString stringWithFormat:@"com.alamofire.autopurgingimagecache-%@", [[NSUUID UUID] UUIDString]];
self.synchronizationQueue = dispatch_queue_create([queueName cStringUsingEncoding:NSASCIIStringEncoding], DISPATCH_QUEUE_CONCURRENT);

那么问题来了:

  1. 相比串行队列锁,改用并行队列有什么好处?
  2. 如何在并行队列中保证字典读写的线程安全?

先回答第一个问题,串行队列虽然保证了数据的读写安全,但是效率并不高,比如同一时间遇到大量对数据的访问操作时,就需要依次排队,按顺序使用资源。而数据的访问操作本身是安全的,只要数据没有被篡改,其实完全可以通过多线程并发访问。只有遇到写操作时,我们等之前所有数据的访问任务结束后,再开始写入操作,并且确保数据改动过程中,让新的访问操作排队等着就好。这其实是计算机科学中的读写锁(Reader/writer lock或者multiple readers/single-writer lock)的理念。

说完了理论,那么如何在并行队列中保证字典读写安全呢?答案是栅栏barrier,直接看源码,精简以后就是这样:

- (void)addImage:(UIImage *)image withIdentifier:(NSString *)identifier {
    // 异步栅栏派发操作
    dispatch_barrier_async(self.synchronizationQueue, ^{
        AFCachedImage *cacheImage = [[AFCachedImage alloc] initWithImage:image identifier:identifier];
        // 如果之前保存了重复图片,删除并从总缓存中减去该图片的占用空间值...
        // 保存图片
        self.cachedImages[identifier] = cacheImage;
        // 将新图片的占用空间计算到总缓存中...
    });
    // 接着又一个异步栅栏派发操作
    dispatch_barrier_async(self.synchronizationQueue, ^{
        // 缓存自动计算和清理:
        // 如果图片缓存空间大于预设值,就按照访问时间顺序,依次清理掉时间最久远的图片
        if (...) {
            // ...
            for (AFCachedImage *cachedImage in sortedImages) {
                [self.cachedImages removeObjectForKey:cachedImage.identifier];
                // ...
            }
        }
    });
}

在写入操作时,dispatch_barrier_async中的栅栏任务在添加到并行队列中,会等待队列中之前所有的任务结束后,才开始执行(也就是图片的缓存操作);而且在执行栅栏任务过程中,后续的新任务(无论读/写)必须等待,直到该栅栏任务结束后才能继续执行。源码在上面的写入方法中连续用了两个异步栅栏,于是队列会完成先前的所有任务,再依次执行栅栏任务1、栅栏任务2,以及后续任务。即使两个栅栏任务中都对字典数据做了改动,依然遵循先后顺序,不会出现资源竞争。再加上这是异步派发,所以写操作将任务加入队列中就会立即返回,而不会阻塞当前的调用线程。

下面是读取图片的实现:

- (nullable UIImage *)imageWithIdentifier:(NSString *)identifier {
    __block UIImage *image = nil;
    dispatch_sync(self.synchronizationQueue, ^{
        AFCachedImage *cachedImage = self.cachedImages[identifier];
        // 读取并更新图片的访问时间
        image = [cachedImage accessImage];
    });
    return image;
}

读取操作时,dispatch_sync会立即返回相应图片,由于这是在并发队列中,因此读取操作可以被指派到任意一个新的线程中完成,而不会被限制在单一线程中,这样就保证了并发访问的需求。

再看下缓存清理的实现:

- (BOOL)removeAllImages {
    __block BOOL removed = NO;
    dispatch_barrier_sync(self.synchronizationQueue, ^{
        if (self.cachedImages.count > 0) {
            [self.cachedImages removeAllObjects];
            self.currentMemoryUsage = 0;
            removed = YES;
        }
    });
    return removed;
}

这里使用了dispatch_barrier_sync,不仅保证了清理任务的排它性(也就是上面分析过的barrier特性),而且remove方法会等待block中的清理任务执行完毕后返回结果。

这里是源码传送门

选择Lock还是GCD

当然各种锁的API(比如NSLock, dispatch_semaphore)同样能够实现保护字典线程安全的任务,所以替换成lock()unlock()之类的方式应该也是可以的,不过像NSLock这类锁只是确保被保护的资源在任何时间只能被一个线程访问,并不具备读写锁的特性;而GCD实现的读写锁不仅允许并发访问,而且使用了异步的缓存操作也不会阻塞调用线程,比如缓存图片时还要及时更新缓存空间,甚至按清理条件删除图片的话,那么读写锁反而更有优势。像SDWebImageSDMemoryCache中实现了图片内存缓存,就是直接使用锁操作,原因是缓存时没有附加额外的任务(比如管理缓存空间之类的事情都交给了NSCache和内存警告通知处理),确保了缓存任务的快速和轻量,并没有像AFAutoPurgingImageCache对缓存空间个性化的控制要求。所以选择Lock还是GCD,还得看具体的需求。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,014评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,796评论 3 386
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 158,484评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,830评论 1 285
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,946评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,114评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,182评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,927评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,369评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,678评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,832评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,533评论 4 335
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,166评论 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,885评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,128评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,659评论 2 362
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,738评论 2 351

推荐阅读更多精彩内容