Added LRU cache to fix out of memory issues with color caching

2026-03-26 07:04:05 +08:00 · 2015-07-13 08:42:32 -07:00
parent 4beef5bec3
commit 01151f8c7a
17 changed files with 687 additions and 28 deletions
--- a/Examples/UIExplorer/UIExplorerUnitTests/RCTCacheTests.m
+++ b/Examples/UIExplorer/UIExplorerUnitTests/RCTCacheTests.m
@@ -0,0 +1,168 @@
+/**
+ * The examples provided by Facebook are for non-commercial testing and
+ * evaluation purposes only.
+ *
+ * Facebook reserves all rights not expressly granted.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
+ * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NON INFRINGEMENT. IN NO EVENT SHALL
+ * FACEBOOK BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN
+ * AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+ * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+
+#import <XCTest/XCTest.h>
+#import "RCTCache.h"
+
+// Silence silly sign warnings when using int literals
+#pragma clang diagnostic ignored "-Wsign-compare"
+
+@interface RCTCache (Private)
+
+- (void)cleanUpAllObjects;
+- (void)resequence;
+- (NSDictionary *)cache;
+- (void)setSequenceNumber:(NSInteger)number;
+
+@end
+
+@interface RCTCacheTests : XCTestCase
+
+@property (nonatomic, strong) RCTCache *cache;
+
+@end
+
+@implementation RCTCacheTests
+
+- (void)setUp
+{
+  self.cache = [[RCTCache alloc] init];
+  self.cache.countLimit = 3;
+  self.cache.totalCostLimit = 100;
+}
+
+- (void)tearDown
+{
+  self.cache = nil;
+}
+
+- (void)testInsertion
+{
+  [self.cache setObject:@1 forKey:@"foo" cost:1];
+  [self.cache setObject:@2 forKey:@"bar" cost:2];
+  [self.cache setObject:@3 forKey:@"baz" cost:3];
+
+  XCTAssertEqual([self.cache count], 3);
+  XCTAssertEqual([self.cache totalCost], 6);
+}
+
+- (void)testRemoval
+{
+  [self.cache setObject:@1 forKey:@"foo" cost:1];
+  [self.cache setObject:@2 forKey:@"bar" cost:2];
+  [self.cache setObject:@3 forKey:@"baz" cost:3];
+
+  [self.cache removeObjectForKey:@"bar"];
+
+  XCTAssertEqual([self.cache count], 2);
+  XCTAssertNil([self.cache objectForKey:@"bar"]);
+}
+
+- (void)testCountEviction
+{
+  [self.cache setObject:@1 forKey:@"foo"];
+  [self.cache setObject:@2 forKey:@"bar"];
+  [self.cache setObject:@3 forKey:@"baz"];
+  [self.cache setObject:@4 forKey:@"bam"];
+
+  XCTAssertEqual([self.cache count], 3);
+  XCTAssertNil([self.cache objectForKey:@"foo"]);
+
+  [self.cache setObject:@5 forKey:@"boo"];
+
+  XCTAssertEqual([self.cache count], 3);
+  XCTAssertNil([self.cache objectForKey:@"bar"]);
+}
+
+- (void)testCostEviction
+{
+  [self.cache setObject:@1 forKey:@"foo" cost:99];
+  [self.cache setObject:@2 forKey:@"bar" cost:2];
+
+  XCTAssertEqual([self.cache count], 1);
+  XCTAssertEqual([self.cache totalCost], 2);
+  XCTAssertNil([self.cache objectForKey:@"foo"]);
+
+  [self.cache setObject:@3 forKey:@"baz" cost:999];
+
+  XCTAssertEqual([self.cache count], 0);
+  XCTAssertEqual([self.cache totalCost], 0);
+}
+
+- (void)testCleanup
+{
+  [self.cache setObject:@1 forKey:@"foo"];
+  [self.cache setObject:@2 forKey:@"bar"];
+  [self.cache setObject:@3 forKey:@"baz"];
+
+  //simulate memory warning
+  [self.cache cleanUpAllObjects];
+
+  XCTAssertEqual([self.cache count], 0);
+  XCTAssertEqual([self.cache totalCost], 0);
+}
+
+- (void)testResequence
+{
+  [self.cache setObject:@1 forKey:@"foo"];
+  [self.cache setObject:@2 forKey:@"bar"];
+  [self.cache setObject:@3 forKey:@"baz"];
+
+  [self.cache resequence];
+
+  NSDictionary *innerCache = [self.cache cache];
+  XCTAssertEqualObjects([innerCache[@"foo"] valueForKey:@"sequenceNumber"], @0);
+  XCTAssertEqualObjects([innerCache[@"bar"] valueForKey:@"sequenceNumber"], @1);
+  XCTAssertEqualObjects([innerCache[@"baz"] valueForKey:@"sequenceNumber"], @2);
+
+  [self.cache removeObjectForKey:@"foo"];
+  [self.cache resequence];
+
+  XCTAssertEqualObjects([innerCache[@"bar"] valueForKey:@"sequenceNumber"], @0);
+  XCTAssertEqualObjects([innerCache[@"baz"] valueForKey:@"sequenceNumber"], @1);
+}
+
+- (void)testResequenceTrigger
+{
+  [self.cache setObject:@1 forKey:@"foo"];
+  [self.cache setObject:@2 forKey:@"bar"];
+
+  //first object should now be bar with sequence number of 1
+  [self.cache removeObjectForKey:@"foo"];
+
+  //should trigger resequence
+  [self.cache setSequenceNumber:NSIntegerMax];
+  [self.cache setObject:@3 forKey:@"baz"];
+
+  NSDictionary *innerCache = [self.cache cache];
+  XCTAssertEqualObjects([innerCache[@"bar"] valueForKey:@"sequenceNumber"], @0);
+  XCTAssertEqualObjects([innerCache[@"baz"] valueForKey:@"sequenceNumber"], @1);
+
+  //first object should now be baz with sequence number of 1
+  [self.cache removeObjectForKey:@"bar"];
+
+  //should also trigger resequence
+  [self.cache setSequenceNumber:NSIntegerMax];
+  [self.cache objectForKey:@"baz"];
+
+  XCTAssertEqualObjects([innerCache[@"baz"] valueForKey:@"sequenceNumber"], @0);
+}
+
+- (void)testName
+{
+  self.cache.name = @"Hello";
+  XCTAssertEqualObjects(self.cache.name, @"Hello");
+}
+
+@end