1 files changed, 64 insertions, 0 deletions
diff --git a/src/kv.rs b/src/kv.rs
new file mode 100644
index 0000000..6c6f15a
--- /dev/null
+++ b/src/kv.rs
@@ -0,0 +1,64 @@
+use anyhow::{anyhow, Result};
+use bytes::Bytes;
+use log::*;
+use reqwest::StatusCode;
+use serde::{Deserialize, Serialize};
+
+use crate::Consul;
+
+impl Consul {
+    pub async fn kv_get(&self, key: &str) -> Result<Option<Bytes>> {
+        debug!("kv_get {}", key);
+
+        let url = format!("{}/v1/kv/{}{}?raw", self.url, self.kv_prefix, key);
+        let http = self.client.get(&url).send().await?;
+        match http.status() {
+            StatusCode::OK => Ok(Some(http.bytes().await?)),
+            StatusCode::NOT_FOUND => Ok(None),
+            _ => Err(anyhow!(
+                "Consul request failed: {:?}",
+                http.error_for_status()
+            )),
+        }
+    }
+
+    pub async fn kv_get_json<T: for<'de> Deserialize<'de>>(&self, key: &str) -> Result<Option<T>> {
+        debug!("kv_get_json {}", key);
+
+        let url = format!("{}/v1/kv/{}{}?raw", self.url, self.kv_prefix, key);
+        let http = self.client.get(&url).send().await?;
+        match http.status() {
+            StatusCode::OK => Ok(Some(http.json().await?)),
+            StatusCode::NOT_FOUND => Ok(None),
+            _ => Err(anyhow!(
+                "Consul request failed: {:?}",
+                http.error_for_status()
+            )),
+        }
+    }
+
+    pub async fn kv_put(&self, key: &str, bytes: Bytes) -> Result<()> {
+        debug!("kv_put {}", key);
+
+        let url = format!("{}/v1/kv/{}{}", self.url, self.kv_prefix, key);
+        let http = self.client.put(&url).body(bytes).send().await?;
+        http.error_for_status()?;
+        Ok(())
+    }
+
+    pub async fn kv_put_json<T: Serialize>(&self, key: &str, value: &T) -> Result<()> {
+        debug!("kv_put_json {}", key);
+
+        let url = format!("{}/v1/kv/{}{}", self.url, self.kv_prefix, key);
+        let http = self.client.put(&url).json(value).send().await?;
+        http.error_for_status()?;
+        Ok(())
+    }
+
+    pub async fn kv_delete(&self, key: &str) -> Result<()> {
+        let url = format!("{}/v1/kv/{}{}", self.url, self.kv_prefix, key);
+        let http = self.client.delete(&url).send().await?;
+        http.error_for_status()?;
+        Ok(())
+    }
+}