Cleanup storage for empty account (#179)

It's helpful to remove account from storage if the account balance is 0 and there are no allowance set to any escrow.
near · Jun 5, 2020 · 6ca4b34 · 6ca4b34
1 parent e5295a3
commit 6ca4b34
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/examples/fungible-token/res/fungible_token.wasm b/examples/fungible-token/res/fungible_token.wasm
diff --git a/examples/fungible-token/src/lib.rs b/examples/fungible-token/src/lib.rs
@@ -238,7 +238,11 @@ impl FungibleToken {
     /// Helper method to set the account details for `owner_id` to the state.
     fn set_account(&mut self, owner_id: &AccountId, account: &Account) {
         let account_hash = env::sha256(owner_id.as_bytes());
-        self.accounts.insert(&account_hash, &account);
+        if account.balance > 0 || !account.allowances.is_empty() {
+            self.accounts.insert(&account_hash, &account);
+        } else {
+            self.accounts.remove(&account_hash);
+        }
     }
 
     fn refund_storage(&self, initial_storage: StorageUsage) {