Refactor: Same approach for cached_statement_size

ref: a66787d
launchbadge · Aug 12, 2024 · 7356c48 · 7356c48
1 parent a66787d
commit 7356c48
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/sqlx-sqlite/src/connection/mod.rs b/sqlx-sqlite/src/connection/mod.rs
@@ -212,10 +212,7 @@ impl Connection for SqliteConnection {
     }
 
     fn cached_statements_size(&self) -> usize {
-        self.worker
-            .shared
-            .cached_statements_size
-            .load(std::sync::atomic::Ordering::Acquire)
+        self.worker.shared.get_cached_statement_size()
     }
 
     fn clear_cached_statements(&mut self) -> BoxFuture<'_, Result<(), Error>> {

diff --git a/sqlx-sqlite/src/connection/worker.rs b/sqlx-sqlite/src/connection/worker.rs
@@ -35,14 +35,18 @@ pub(crate) struct ConnectionWorker {
 
 pub(crate) struct WorkerSharedState {
     transaction_depth: AtomicUsize,
-    pub(crate) cached_statements_size: AtomicUsize,
+    cached_statements_size: AtomicUsize,
     pub(crate) conn: Mutex<ConnectionState>,
 }
 
 impl WorkerSharedState {
     pub(crate) fn get_transaction_depth(&self) -> usize {
         self.transaction_depth.load(Ordering::Acquire)
     }
+
+    pub(crate) fn get_cached_statement_size(&self) -> usize {
+        self.cached_statements_size.load(Ordering::Acquire)
+    }
 }
 
 enum Command {