chore: add admin_clear API in ic_object_store_canister

Author: zensh

Cargo.lock

@@ -100,6 +100,7 @@ service : (opt InstallArgs) -> {
   abort_multipart : (text, text) -> (Result);
   admin_add_auditors : (vec principal) -> (Result_1);
   admin_add_managers : (vec principal) -> (Result_1);
+  admin_clear : () -> (Result_1);
   admin_remove_auditors : (vec principal) -> (Result_1);
   admin_remove_managers : (vec principal) -> (Result_1);
   complete_multipart : (text, text, PutMultipartOpts) -> (Result_2);
@@ -122,6 +123,7 @@ service : (opt InstallArgs) -> {
   rename_if_not_exists : (text, text) -> (Result);
   validate_admin_add_auditors : (vec principal) -> (Result_13);
   validate_admin_add_managers : (vec principal) -> (Result_13);
+  validate_admin_clear : () -> (Result_13);
   validate_admin_remove_auditors : (vec principal) -> (Result_13);
   validate_admin_remove_managers : (vec principal) -> (Result_13);
 }

src/ic_object_store_canister/ic_object_store_canister.did

@@ -40,6 +40,12 @@ fn admin_remove_auditors(args: BTreeSet<Principal>) -> Result<(), String> {
     })
 }
 
+#[ic_cdk::update(guard = "is_controller")]
+fn admin_clear() -> Result<(), String> {
+    store::state::clear();
+    Ok(())
+}
+
 #[ic_cdk::update]
 fn validate_admin_add_managers(args: BTreeSet<Principal>) -> Result<String, String> {
     validate_principals(&args)?;
@@ -63,3 +69,8 @@ fn validate_admin_remove_auditors(args: BTreeSet<Principal>) -> Result<String, S
     validate_principals(&args)?;
     Ok("ok".to_string())
 }
+
+#[ic_cdk::update]
+fn validate_admin_clear() -> Result<String, String> {
+    Ok("ok".to_string())
+}

src/ic_object_store_canister/src/api_admin.rs

@@ -177,6 +177,17 @@ pub mod state {
             });
         });
     }
+
+    pub fn clear() {
+        MULTIPART_UPLOAD.with_borrow_mut(|mu| mu.clear());
+        OBJECT_META.with_borrow_mut(|om| om.clear_new());
+        OBJECT_DATA.with_borrow_mut(|od| od.clear_new());
+        STATE.with_borrow_mut(|s| {
+            s.locations.clear();
+            s.next_etag = 0;
+        });
+        save();
+    }
 }
 
 pub mod object {