forked from matter-labs/zksync
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
8 changed files
with
259 additions
and
227 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,159 @@ | ||
| // Built-in deps | ||
| use std::time::Instant; | ||
| // External imports | ||
| // Workspace imports | ||
| use zksync_types::BlockNumber; | ||
| // Local imports | ||
| use self::records::AccountTreeCache; | ||
| use crate::{QueryResult, StorageProcessor}; | ||
|
|
||
| pub mod records; | ||
|
|
||
| /// Tree cache schema contains methods to store/load Merkle tree cache. | ||
| #[derive(Debug)] | ||
| pub struct TreeCacheSchema<'a, 'c>(pub &'a mut StorageProcessor<'c>); | ||
|
|
||
| impl<'a, 'c> TreeCacheSchema<'a, 'c> { | ||
| /// Stores account tree cache for a block. | ||
| /// Expects `tree_cache` to be a valid encoded JSON. | ||
| pub async fn store_account_tree_cache( | ||
| &mut self, | ||
| block: BlockNumber, | ||
| tree_cache: String, | ||
| ) -> QueryResult<()> { | ||
| let start = Instant::now(); | ||
| if *block == 0 { | ||
| return Ok(()); | ||
| } | ||
|
|
||
| sqlx::query!( | ||
| " | ||
| INSERT INTO account_tree_cache (block, tree_cache) | ||
| VALUES ($1, $2) | ||
| ON CONFLICT (block) | ||
| DO NOTHING | ||
| ", | ||
| *block as i64, | ||
| tree_cache, | ||
| ) | ||
| .execute(self.0.conn()) | ||
| .await?; | ||
|
|
||
| metrics::histogram!("sql.chain.block.store_account_tree_cache", start.elapsed()); | ||
| Ok(()) | ||
| } | ||
|
|
||
| /// Gets the number of the latest block that has a stored cache. | ||
| /// Returns `None` if there are no caches in the database. | ||
| pub async fn get_last_block_with_account_tree_cache( | ||
| &mut self, | ||
| ) -> QueryResult<Option<BlockNumber>> { | ||
| let start = Instant::now(); | ||
|
|
||
| let last_block_with_cache = sqlx::query!("SELECT MAX(block) FROM account_tree_cache") | ||
| .fetch_one(self.0.conn()) | ||
| .await? | ||
| .max; | ||
|
|
||
| metrics::histogram!( | ||
| "sql.chain.block.get_last_block_with_account_tree_cache", | ||
| start.elapsed() | ||
| ); | ||
| Ok(last_block_with_cache.map(|block| BlockNumber(block as u32))) | ||
| } | ||
|
|
||
| /// Gets the latest stored account tree cache. | ||
| /// Returns `None` if there are no caches in the database. | ||
| /// Returns the block number and associated cache otherwise. | ||
| pub async fn get_account_tree_cache( | ||
| &mut self, | ||
| ) -> QueryResult<Option<(BlockNumber, serde_json::Value)>> { | ||
| let start = Instant::now(); | ||
| let account_tree_cache = sqlx::query_as!( | ||
| AccountTreeCache, | ||
| " | ||
| SELECT * FROM account_tree_cache | ||
| ORDER BY block DESC | ||
| LIMIT 1 | ||
| ", | ||
| ) | ||
| .fetch_optional(self.0.conn()) | ||
| .await?; | ||
|
|
||
| metrics::histogram!("sql.chain.block.get_account_tree_cache", start.elapsed()); | ||
| Ok(account_tree_cache.map(|w| { | ||
| ( | ||
| BlockNumber(w.block as u32), | ||
| serde_json::from_str(&w.tree_cache) | ||
| .expect("Failed to deserialize Account Tree Cache"), | ||
| ) | ||
| })) | ||
| } | ||
|
|
||
| /// Gets stored account tree cache for a certain block. | ||
| /// Returns `None` if there is no cache for requested block. | ||
| pub async fn get_account_tree_cache_block( | ||
| &mut self, | ||
| block: BlockNumber, | ||
| ) -> QueryResult<Option<serde_json::Value>> { | ||
| let start = Instant::now(); | ||
| let account_tree_cache = sqlx::query_as!( | ||
| AccountTreeCache, | ||
| " | ||
| SELECT * FROM account_tree_cache | ||
| WHERE block = $1 | ||
| ", | ||
| *block as i64 | ||
| ) | ||
| .fetch_optional(self.0.conn()) | ||
| .await?; | ||
|
|
||
| metrics::histogram!( | ||
| "sql.chain.block.get_account_tree_cache_block", | ||
| start.elapsed() | ||
| ); | ||
| Ok(account_tree_cache.map(|w| { | ||
| serde_json::from_str(&w.tree_cache).expect("Failed to deserialize Account Tree Cache") | ||
| })) | ||
| } | ||
|
|
||
| // Removes account tree cache for blocks with number greater than `last_block` | ||
| pub async fn remove_new_account_tree_cache( | ||
| &mut self, | ||
| last_block: BlockNumber, | ||
| ) -> QueryResult<()> { | ||
| let start = Instant::now(); | ||
| sqlx::query!( | ||
| "DELETE FROM account_tree_cache WHERE block > $1", | ||
| *last_block as i64 | ||
| ) | ||
| .execute(self.0.conn()) | ||
| .await?; | ||
|
|
||
| metrics::histogram!( | ||
| "sql.chain.block.remove_new_account_tree_cache", | ||
| start.elapsed() | ||
| ); | ||
| Ok(()) | ||
| } | ||
|
|
||
| // Removes account tree cache for blocks with number less than `last_block` | ||
| pub async fn remove_old_account_tree_cache( | ||
| &mut self, | ||
| last_block: BlockNumber, | ||
| ) -> QueryResult<()> { | ||
| let start = Instant::now(); | ||
| sqlx::query!( | ||
| "DELETE FROM account_tree_cache WHERE block < $1", | ||
| *last_block as i64 | ||
| ) | ||
| .execute(self.0.conn()) | ||
| .await?; | ||
|
|
||
| metrics::histogram!( | ||
| "sql.chain.block.remove_old_account_tree_cache", | ||
| start.elapsed() | ||
| ); | ||
| Ok(()) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,10 @@ | ||
| // External imports | ||
| use serde::{Deserialize, Serialize}; | ||
| // Workspace imports | ||
| // Local imports | ||
|
|
||
| #[derive(Debug, Clone, Serialize, Deserialize)] | ||
| pub struct AccountTreeCache { | ||
| pub block: i64, | ||
| pub tree_cache: String, | ||
| } |
Oops, something went wrong.