1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
/// LucidKV Client
use reqwest::{Client, Error, Response};
use serde::Serialize;
use url::Url;

pub struct LucidKVClient {
    pub http_client: Client,
    pub base_url: Url,
    jwt: Option<String>,
}

impl LucidKVClient {
    /// Create a new basic Lucid KV Client
    pub fn new(option_base_url: Option<String>) -> Self {
        let http_client = Client::new();
        let mut base_url: Url =
            Url::parse(&option_base_url.unwrap_or(String::from("http://127.0.0.1:7020")))
                .expect("Invalid input url");

        base_url = base_url.join("/api/kv/").unwrap();

        Self {
            http_client,
            base_url,
            jwt: None,
        }
    }

    /// Stores data into Lucid DB
    /// Will panic if key is not URI safe
    pub async fn store_data<T>(&self, key: String, data: T) -> Result<Response, Error>
    where
        T: Serialize,
    {
        let url: Url = self.base_url.join(&key).unwrap();

        self.http_client.put(url).json(&data).send().await
    }

    /// Gets data from Lucid DB
    /// Will panic if key is not URI safe
    pub async fn get_data(&self, key: String) -> Result<Response, Error> {
        let url: Url = self.base_url.join(&key).unwrap();
        self.http_client.get(url).send().await
    }

    /// Delete data from Lucid DB
    /// Will panic if key is not URI safe
    pub async fn delete_data(&self, key: String) -> Result<Response, Error> {
        let url: Url = self.base_url.join(&key).unwrap();
        self.http_client.delete(url).send().await
    }

    /// Sends HEAD request to Lucid DB, will return true if key is present
    /// Will panic if key is not URI safe
    pub async fn is_key_present(&self, key: String) -> bool {
        let url: Url = self.base_url.join(&key).unwrap();
        match self.http_client.head(url).send().await {
            Ok(res) => res.status() == reqwest::StatusCode::OK,
            _ => false,
        }
    }
}