use std::sync::Arc;
use crate::prelude::{thirdparty::arrow_array::Array, *};
pub struct Query<T: ArrowMessage, F: ArrowMessage> {
pub raw: RawQuery,
_phantom: std::marker::PhantomData<(T, F)>,
}
impl<T: ArrowMessage, F: ArrowMessage> Query<T, F> {
pub fn new(
tx: MessageSender,
rx: MessageReceiver,
clock: Arc<HLC>,
source: NodeID,
layout: QueryID,
) -> Self {
Self {
raw: RawQuery::new(tx, rx, clock, source, layout),
_phantom: std::marker::PhantomData,
}
}
pub async fn query(&mut self, data: T) -> Result<TypedDataflowMessage<F>> {
self.raw
.query(
data.try_into_arrow()
.wrap_err(report_failed_conversion_to_arrow::<T>(
&self.raw.source,
&self.raw.layout,
))?
.into_data(),
)
.await?
.try_into()
.wrap_err(report_failed_conversion_from_arrow::<F>(
&self.raw.source,
&self.raw.layout,
))
}
}