Merge branch '8.5.4-repair' into xcframework

GoodNotes · Jan 18, 2024 · 991b46a · 991b46a
2 parents a3e7c00 + a685be6
commit 991b46a
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/Code/RocksDB.h b/Code/RocksDB.h
@@ -590,6 +590,10 @@ typedef void (^RestoreBlock)(NSError * __autoreleasing *);
 - (nullable id)withBulkLoad:(id (NS_NOESCAPE ^)(void))actionBlock
                       error:(NSError * __autoreleasing *)error;
 
+#pragma mark - Repair
+
++ (BOOL)repairDB:(NSString *)dbPath options:(RocksDBOptions *)options error:(NSError **)error;
+
 @end
 
 NS_ASSUME_NONNULL_END
diff --git a/Code/RocksDB.mm b/Code/RocksDB.mm
@@ -804,4 +804,19 @@ - (nullable id)withBulkLoad:(id (NS_NOESCAPE ^)(void))actionBlock
     return result;
 }
 
+#pragma mark - Repair
+
++ (BOOL)repairDB:(NSString *)dbPath options:(RocksDBOptions *)options error:(NSError **)error {
+    rocksdb::Status s = rocksdb::RepairDB(dbPath.UTF8String, options.options);
+
+    if (!s.ok()) {
+        if (error != NULL) {
+            *error =  [RocksDBError errorWithRocksStatus:s];
+        }
+        return NO;
+    }
+
+    return YES;
+}
+
 @end