feat: added query function in aw-client

aw-client-rust/src/blocking.rs

@@ -62,6 +62,12 @@ impl AwClient {
         stop: Option<DateTime<Utc>>,
         limit: Option<u64>
     );
+    proxy_method!(
+        query,
+        serde_json::Value,
+        query: &str,
+        timeperiods: Vec<(DateTime<Utc>, DateTime<Utc>)>
+    );
     proxy_method!(insert_event, (), bucketname: &str, event: &Event);
     proxy_method!(insert_events, (), bucketname: &str, events: Vec<Event>);
     proxy_method!(

aw-client-rust/src/lib.rs

@@ -11,7 +11,7 @@ use std::vec::Vec;
 use std::{collections::HashMap, error::Error};
 
 use chrono::{DateTime, Utc};
-use serde_json::Map;
+use serde_json::{Map, json};
 
 pub use aw_models::{Bucket, BucketMetadata, Event};
 
@@ -98,6 +98,31 @@ impl AwClient {
         Ok(())
     }
 
+    pub async fn query(
+        &self,
+        query: &str,
+        timeperiods: Vec<(DateTime<Utc>, DateTime<Utc>)>,
+    ) -> Result<serde_json::Value, reqwest::Error> {
+        let url = reqwest::Url::parse(
+            format!("{}/api/0/query", self.baseurl).as_str(),
+        )
+        .unwrap();
+
+        // Format timeperiods as ISO8601 strings, separated by /
+        let timeperiods_str: Vec<String> = timeperiods
+            .iter()
+            .map(|(start, stop)| (start.to_rfc3339(), stop.to_rfc3339()))
+            .map(|(start, stop)| format!("{}/{}", start, stop))
+            .collect();
+
+        self.client.post(url)
+            .json(&json!({
+                "query": query,
+                "timeperiods": timeperiods_str,
+            }))
+            .send().await?.json().await
+    }
+
     pub async fn get_events(
         &self,
         bucketname: &str,