Expose method to repair the DB

GoodNotes · Dec 5, 2023 · a685be6 · a685be6
1 parent 315e45d
commit a685be6
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