pub mod config;
pub mod http;
pub use config::{CallSpec, HttpReactionConfig, QueryConfig};
pub use http::HttpReaction;
use std::collections::HashMap;
pub struct HttpReactionBuilder {
id: String,
queries: Vec<String>,
base_url: String,
token: Option<String>,
timeout_ms: u64,
routes: HashMap<String, QueryConfig>,
priority_queue_capacity: Option<usize>,
auto_start: bool,
}
impl HttpReactionBuilder {
pub fn new(id: impl Into<String>) -> Self {
Self {
id: id.into(),
queries: Vec::new(),
base_url: "http://localhost".to_string(),
token: None,
timeout_ms: 5000,
routes: HashMap::new(),
priority_queue_capacity: None,
auto_start: true,
}
}
pub fn with_queries(mut self, queries: Vec<String>) -> Self {
self.queries = queries;
self
}
pub fn with_query(mut self, query_id: impl Into<String>) -> Self {
self.queries.push(query_id.into());
self
}
pub fn with_base_url(mut self, base_url: impl Into<String>) -> Self {
self.base_url = base_url.into();
self
}
pub fn with_token(mut self, token: impl Into<String>) -> Self {
self.token = Some(token.into());
self
}
pub fn with_timeout_ms(mut self, timeout_ms: u64) -> Self {
self.timeout_ms = timeout_ms;
self
}
pub fn with_route(mut self, query_id: impl Into<String>, config: QueryConfig) -> Self {
self.routes.insert(query_id.into(), config);
self
}
pub fn with_priority_queue_capacity(mut self, capacity: usize) -> Self {
self.priority_queue_capacity = Some(capacity);
self
}
pub fn with_auto_start(mut self, auto_start: bool) -> Self {
self.auto_start = auto_start;
self
}
pub fn with_config(mut self, config: HttpReactionConfig) -> Self {
self.base_url = config.base_url;
self.token = config.token;
self.timeout_ms = config.timeout_ms;
self.routes = config.routes;
self
}
pub fn build(self) -> anyhow::Result<HttpReaction> {
let config = HttpReactionConfig {
base_url: self.base_url,
token: self.token,
timeout_ms: self.timeout_ms,
routes: self.routes,
};
Ok(HttpReaction::from_builder(
self.id,
self.queries,
config,
self.priority_queue_capacity,
self.auto_start,
))
}
}
#[cfg(test)]
mod tests {
use super::*;
use drasi_lib::Reaction;
#[test]
fn test_http_builder_defaults() {
let reaction = HttpReactionBuilder::new("test-reaction").build().unwrap();
assert_eq!(reaction.id(), "test-reaction");
let props = reaction.properties();
assert_eq!(
props.get("base_url"),
Some(&serde_json::Value::String("http://localhost".to_string()))
);
assert_eq!(
props.get("timeout_ms"),
Some(&serde_json::Value::Number(5000.into()))
);
}
#[test]
fn test_http_builder_custom_values() {
let reaction = HttpReaction::builder("test-reaction")
.with_base_url("http://api.example.com") .with_token("secret-token")
.with_timeout_ms(10000)
.with_queries(vec!["query1".to_string()])
.build()
.unwrap();
assert_eq!(reaction.id(), "test-reaction");
let props = reaction.properties();
assert_eq!(
props.get("base_url"),
Some(&serde_json::Value::String(
"http://api.example.com".to_string() ))
);
assert_eq!(
props.get("timeout_ms"),
Some(&serde_json::Value::Number(10000.into()))
);
assert_eq!(reaction.query_ids(), vec!["query1".to_string()]);
}
#[test]
fn test_http_builder_with_query() {
let reaction = HttpReaction::builder("test-reaction")
.with_query("query1")
.with_query("query2")
.build()
.unwrap();
assert_eq!(reaction.query_ids(), vec!["query1", "query2"]);
}
#[test]
fn test_http_new_constructor() {
let config = HttpReactionConfig {
base_url: "http://test.example.com".to_string(), token: Some("test-token".to_string()),
timeout_ms: 3000,
routes: Default::default(),
};
let reaction = HttpReaction::new("test-reaction", vec!["query1".to_string()], config);
assert_eq!(reaction.id(), "test-reaction");
assert_eq!(reaction.query_ids(), vec!["query1".to_string()]);
}
}