use super::edges::EdgeKey;
use super::types::Type;
use chrono::DateTime;
use chrono::offset::Utc;
use uuid::Uuid;
#[derive(Eq, PartialEq, Clone, Debug, Serialize, Deserialize, Hash, Copy)]
pub enum EdgeDirection {
#[serde(rename = "outbound")] Outbound,
#[serde(rename = "inbound")] Inbound,
}
#[derive(Eq, PartialEq, Clone, Debug, Serialize, Deserialize, Hash)]
#[serde(tag = "type", rename_all = "snake_case")]
pub enum VertexQuery {
All {
start_id: Option<Uuid>,
limit: u32,
},
Vertices {
ids: Vec<Uuid>,
},
Pipe {
edge_query: Box<EdgeQuery>,
converter: EdgeDirection,
limit: u32,
},
}
impl VertexQuery {
pub fn outbound_edges(
self,
t: Option<Type>,
high: Option<DateTime<Utc>>,
low: Option<DateTime<Utc>>,
limit: u32,
) -> EdgeQuery {
EdgeQuery::Pipe {
vertex_query: Box::new(self),
converter: EdgeDirection::Outbound,
type_filter: t,
high_filter: high,
low_filter: low,
limit: limit,
}
}
pub fn inbound_edges(
self,
t: Option<Type>,
high: Option<DateTime<Utc>>,
low: Option<DateTime<Utc>>,
limit: u32,
) -> EdgeQuery {
EdgeQuery::Pipe {
vertex_query: Box::new(self),
converter: EdgeDirection::Inbound,
type_filter: t,
high_filter: high,
low_filter: low,
limit: limit,
}
}
}
#[derive(Eq, PartialEq, Clone, Debug, Serialize, Deserialize, Hash)]
#[serde(tag = "type", rename_all = "snake_case")]
pub enum EdgeQuery {
Edges {
keys: Vec<EdgeKey>,
},
Pipe {
vertex_query: Box<VertexQuery>,
converter: EdgeDirection,
type_filter: Option<Type>,
high_filter: Option<DateTime<Utc>>,
low_filter: Option<DateTime<Utc>>,
limit: u32,
},
}
impl EdgeQuery {
pub fn outbound_vertices(self, limit: u32) -> VertexQuery {
VertexQuery::Pipe {
edge_query: Box::new(self),
converter: EdgeDirection::Outbound,
limit: limit,
}
}
pub fn inbound_vertices(self, limit: u32) -> VertexQuery {
VertexQuery::Pipe {
edge_query: Box::new(self),
converter: EdgeDirection::Inbound,
limit: limit,
}
}
}