wasm rdbms

Cargo.toml

@@ -66,6 +66,7 @@ exclude = [
     "test-components/variant-service",
     "test-components/write-stderr",
     "test-components/write-stdout",
+    "test-components/rdbms-service",
 ]
 
 [workspace.dependencies]
@@ -162,13 +163,20 @@ serde_yaml = { version = "0.9.34" }
 sha2 = "0.10.8"
 sozu-command-lib = { version = "1.0.5", default-features = false }
 sozu-lib = { version = "1.0.5", default-features = false }
+sqlx-core = { version = "0.8" }
 sqlx = { version = "0.8", features = [
     "runtime-tokio",
     "sqlite",
     "postgres",
+    "mysql",
     "uuid",
     "migrate",
     "chrono",
+    "json",
+    "bigdecimal",
+    "mac_address",
+    "bit-vec",
+    "ipnetwork"
 ] }
 strum = "0.26.3"
 strum_macros = "0.26.4"
@@ -179,6 +187,7 @@ testcontainers-modules = { version = "0.11.4", features = [
     "postgres",
     "redis",
     "minio",
+    "mysql"
 ] }
 test-r = { version = "2.0.1", default-features = true }
 thiserror = "2.0.6"

golem-common/src/cache.rs

@@ -156,35 +156,41 @@ impl<
     /// Tries to get a cached value for the given key. If the value is missing or is pending, it returns None.
     #[allow(unused)]
     pub fn try_get(&self, key: &K) -> Option<V> {
-        match self.state.items.try_get(key) {
+        let result = match self.state.items.try_get(key) {
             Present(item) => match item.deref() {
                 Item::Pending { .. } => None,
-                Item::Cached { value, .. } => {
-                    self.update_last_access(key);
-                    Some(value.clone())
-                }
+                Item::Cached { value, .. } => Some(value.clone()),
             },
             Absent | Locked => None,
+        };
+
+        if (result.is_some()) {
+            self.update_last_access(key);
         }
+
+        result
     }
 
     /// Gets a cached value for the given key. If the value is pending, it awaits it.
     /// If the pending value fails, it returns None.
     #[allow(unused)]
     pub async fn get(&self, key: &K) -> Option<V> {
-        match self.state.items.get(key) {
+        let result = match self.state.items.get(key) {
             Some(item) => match item.deref() {
                 Item::Pending { tx, .. } => {
                     let mut rx = tx.subscribe();
                     rx.recv().await.ok().and_then(|r| r.ok())
                 }
-                Item::Cached { value, .. } => {
-                    self.update_last_access(key);
-                    Some(value.clone())
-                }
+                Item::Cached { value, .. } => Some(value.clone()),
             },
             None => None,
+        };
+
+        if (result.is_some()) {
+            self.update_last_access(key);
         }
+
+        result
     }
 
     /// Gets a cached value for a given key, or inserts a new one with the given async function. If a value is pending,
@@ -336,6 +342,10 @@ impl<
         }
     }
 
+    pub fn contains_key(&self, key: &K) -> bool {
+        self.state.items.contains_key(key)
+    }
+
     pub fn create_weak_remover(&self, key: K) -> impl FnOnce() {
         let weak_state = Arc::downgrade(&self.state);
         let name = self.name;

golem-test-framework/Cargo.toml

@@ -40,10 +40,10 @@ redis = { workspace = true }
 serde = { workspace = true }
 serde_json = { workspace = true }
 serde_yaml = { workspace = true }
+sqlx = { workspace = true }
 testcontainers = { workspace = true }
 testcontainers-modules = { workspace = true }
 tokio = { workspace = true }
-tokio-postgres = { workspace = true }
 tokio-stream = { workspace = true }
 tonic = { workspace = true }
 tracing = { workspace = true }