perf:make txn share storage table catalog cache

KipData · Mar 17, 2024 · e2569a0 · e2569a0
1 parent f8488f4
commit e2569a0
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/storage/kip.rs b/src/storage/kip.rs
@@ -23,6 +23,7 @@ use std::sync::Arc;
 pub struct KipStorage {
     pub inner: Arc<storage::KipStorage>,
     pub(crate) meta_cache: Arc<ShardingLruCache<TableName, Vec<StatisticsMeta>>>,
+    pub(crate) table_cache: Arc<ShardingLruCache<String, TableCatalog>>,
 }
 
 impl KipStorage {
@@ -31,10 +32,12 @@ impl KipStorage {
             storage::KipStorage::open_with_config(Config::new(path).enable_level_0_memorization())
                 .await?;
         let meta_cache = Arc::new(ShardingLruCache::new(128, 16, RandomState::new()).unwrap());
+        let table_cache = Arc::new(ShardingLruCache::new(128, 16, RandomState::new()).unwrap());
 
         Ok(KipStorage {
             inner: Arc::new(storage),
             meta_cache,
+            table_cache,
         })
     }
 }
@@ -47,15 +50,15 @@ impl Storage for KipStorage {
 
         Ok(KipTransaction {
             tx,
-            table_cache: ShardingLruCache::new(8, 2, RandomState::default())?,
+            table_cache: Arc::clone(&self.table_cache),
             meta_cache: self.meta_cache.clone(),
         })
     }
 }
 
 pub struct KipTransaction {
     tx: mvcc::Transaction,
-    table_cache: ShardingLruCache<String, TableCatalog>,
+    table_cache: Arc<ShardingLruCache<String, TableCatalog>>,
     meta_cache: Arc<ShardingLruCache<TableName, Vec<StatisticsMeta>>>,
 }