fix: add missing exposed config

geofmureithi · Jul 10, 2024 · bd3cc5c · bd3cc5c
1 parent 9e71559
commit bd3cc5c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/packages/apalis-sql/src/mysql.rs b/packages/apalis-sql/src/mysql.rs
@@ -114,6 +114,11 @@ impl<T: Serialize + DeserializeOwned> MysqlStorage<T> {
     pub fn codec(&self) -> &BoxCodec<T, serde_json::Value> {
         &self.codec
     }
+
+    /// Get the config used by the storage
+    pub fn get_config(&self) -> &Config {
+        &self.config
+    }
 }
 
 impl<T: DeserializeOwned + Send + Unpin + Sync + 'static> MysqlStorage<T> {

diff --git a/packages/apalis-sql/src/sqlite.rs b/packages/apalis-sql/src/sqlite.rs
@@ -149,6 +149,11 @@ impl<T: Serialize + DeserializeOwned> SqliteStorage<T> {
     pub fn codec(&self) -> &SqliteCodec<T> {
         &self.codec
     }
+
+    /// Get the config used by the storage
+    pub fn get_config(&self) -> &Config {
+        &self.config
+    }
 }
 
 async fn fetch_next(