Add eraseUnreferencedEntries in RocksDBCache

GoodNotes · Jan 24, 2025 · d39bbd8 · d39bbd8
1 parent 386b0b0
commit d39bbd8
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/Code/RocksDBCache.h b/Code/RocksDBCache.h
@@ -35,6 +35,12 @@ NS_ASSUME_NONNULL_BEGIN
  */
 + (instancetype)LRUCacheWithCapacity:(size_t)capacity numShardsBits:(int)numShardBits;
 
+
+/**
+ Removes all unreferenced entries from the cache.
+ */
+- (void)eraseUnreferencedEntries;
+
 @end
 
 NS_ASSUME_NONNULL_END
diff --git a/Code/RocksDBCache.mm b/Code/RocksDBCache.mm
@@ -9,6 +9,7 @@
 #import "RocksDBCache.h"
 
 #import <rocksdb/cache.h>
+#import <rocksdb/advanced_cache.h>
 
 @interface RocksDBCache ()
 {
@@ -39,6 +40,15 @@ - (instancetype)initWithNativeCache:(std::shared_ptr<rocksdb::Cache>)cache
 	return self;
 }
 
+- (void)eraseUnreferencedEntries
+{
+    @synchronized(self) {
+        if (_cache != nullptr) {
+            _cache->EraseUnRefEntries();
+        }
+    }
+}
+
 - (void)dealloc
 {
 	@synchronized(self) {