mirror of
https://github.com/Swiftgram/Telegram-iOS.git
synced 2025-09-04 11:50:38 +00:00
cac14e0bce
* Audit property attributes for core classes * Update style guide * Go crazy * Update changelog
87 lines
2.5 KiB
Objective-C
87 lines
2.5 KiB
Objective-C
//
|
|
// ASWeakMap.m
|
|
// Texture
|
|
//
|
|
// Copyright (c) 2014-present, Facebook, Inc. All rights reserved.
|
|
// This source code is licensed under the BSD-style license found in the
|
|
// LICENSE file in the /ASDK-Licenses directory of this source tree. An additional
|
|
// grant of patent rights can be found in the PATENTS file in the same directory.
|
|
//
|
|
// Modifications to this file made after 4/13/2017 are: Copyright (c) 2017-present,
|
|
// Pinterest, Inc. Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
|
|
#import <AsyncDisplayKit/ASWeakMap.h>
|
|
|
|
@interface ASWeakMapEntry ()
|
|
@property (nonatomic, readonly) id key;
|
|
@property id value;
|
|
@end
|
|
|
|
@implementation ASWeakMapEntry
|
|
|
|
- (instancetype)initWithKey:(id)key value:(id)value
|
|
{
|
|
self = [super init];
|
|
if (self) {
|
|
_key = key;
|
|
_value = value;
|
|
}
|
|
return self;
|
|
}
|
|
|
|
@end
|
|
|
|
|
|
@interface ASWeakMap ()
|
|
@property (nonatomic, readonly) NSMapTable<id, ASWeakMapEntry *> *hashTable;
|
|
@end
|
|
|
|
/**
|
|
* Implementation details:
|
|
*
|
|
* The retained size of our keys is potentially very large (for example, a UIImage is commonly part of a key).
|
|
* Unfortunately, NSMapTable does not make guarantees about how quickly it will dispose of entries where
|
|
* either the key or the value is weak and has been disposed. So, a NSMapTable with "strong key to weak value" is
|
|
* unsuitable for our purpose because the strong keys are retained longer than the value and for an indefininte period of time.
|
|
* More details here: http://cocoamine.net/blog/2013/12/13/nsmaptable-and-zeroing-weak-references/
|
|
*
|
|
* Our NSMapTable is "weak key to weak value" where each key maps to an Entry. The Entry object is responsible
|
|
* for retaining both the key and value. Our convention is that the caller must retain the Entry object
|
|
* in order to keep the key and the value in the cache.
|
|
*/
|
|
@implementation ASWeakMap
|
|
|
|
- (instancetype)init
|
|
{
|
|
self = [super init];
|
|
if (self) {
|
|
_hashTable = [NSMapTable weakToWeakObjectsMapTable];
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (ASWeakMapEntry *)entryForKey:(id)key
|
|
{
|
|
return [self.hashTable objectForKey:key];
|
|
}
|
|
|
|
- (ASWeakMapEntry *)setObject:(id)value forKey:(id)key
|
|
{
|
|
ASWeakMapEntry *entry = [self.hashTable objectForKey:key];
|
|
if (entry != nil) {
|
|
// Update the value in the existing entry.
|
|
entry.value = value;
|
|
} else {
|
|
entry = [[ASWeakMapEntry alloc] initWithKey:key value:value];
|
|
[self.hashTable setObject:entry forKey:key];
|
|
}
|
|
return entry;
|
|
}
|
|
|
|
@end
|