Fix kv purge with prefix

Signed-off-by: Tomasz Pietrek <[email protected]>
nats-io · Jul 24, 2023 · 24fb246 · 24fb246
1 parent 625d1da
commit 24fb246
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/async-nats/src/jetstream/kv/mod.rs b/async-nats/src/jetstream/kv/mod.rs
@@ -689,7 +689,13 @@ impl Store {
             return Err(PurgeError::new(PurgeErrorKind::InvalidKey));
         }
 
-        let subject = format!("{}{}", self.prefix.as_str(), key.as_ref());
+        let mut subject = String::new();
+        if self.use_jetstream_prefix {
+            subject.push_str(&self.stream.context.prefix);
+            subject.push('.');
+        }
+        subject.push_str(self.put_prefix.as_ref().unwrap_or(&self.prefix));
+        subject.push_str(key.as_ref());
 
         let mut headers = crate::HeaderMap::default();
         headers.insert(KV_OPERATION, HeaderValue::from(KV_OPERATION_PURGE));

diff --git a/async-nats/tests/kv_tests.rs b/async-nats/tests/kv_tests.rs
@@ -757,6 +757,10 @@ mod kv {
         let name = test.get("name").await.unwrap();
         assert_eq!(from_utf8(&name.unwrap()).unwrap(), "ivan".to_string());
 
+        test.purge("name").await.unwrap();
+        let name = test.get("name").await.unwrap();
+        assert!(name.is_none());
+
         // Shutdown HUB and test get still work.
         drop(hub_server);