use std::marker::PhantomData;
use futures::{Future, Poll};
use error::{Error, Result};
use client::{AsyncSender, Client, Sender, SyncSender};
use client::requests::RequestBuilder;
use client::requests::params::{Id, Index, Type};
use client::requests::endpoints::DeleteRequest;
use client::requests::raw::RawRequestInner;
use client::responses::DeleteResponse;
use types::document::DocumentType;
pub type DeleteRequestBuilder<TSender, TDocument> = RequestBuilder<TSender, DeleteRequestInner<TDocument>>;
#[doc(hidden)]
pub struct DeleteRequestInner<TDocument> {
index: Index<'static>,
ty: Type<'static>,
id: Id<'static>,
_marker: PhantomData<TDocument>,
}
impl<TSender> Client<TSender>
where
TSender: Sender,
{
pub fn document_delete<TDocument>(&self, index: Index<'static>, id: Id<'static>) -> DeleteRequestBuilder<TSender, TDocument>
where
TDocument: DocumentType,
{
let ty = TDocument::name().into();
RequestBuilder::new(
self.clone(),
None,
DeleteRequestInner {
index: index,
ty: ty,
id: id,
_marker: PhantomData,
},
)
}
}
impl<TDocument> DeleteRequestInner<TDocument> {
fn into_request(self) -> DeleteRequest<'static> {
DeleteRequest::for_index_ty_id(self.index, self.ty, self.id)
}
}
impl<TSender, TDocument> DeleteRequestBuilder<TSender, TDocument>
where
TSender: Sender,
{
pub fn ty<I>(mut self, ty: I) -> Self
where
I: Into<Type<'static>>,
{
self.inner.ty = ty.into();
self
}
}
impl<TDocument> DeleteRequestBuilder<SyncSender, TDocument> {
pub fn send(self) -> Result<DeleteResponse> {
let req = self.inner.into_request();
RequestBuilder::new(self.client, self.params, RawRequestInner::new(req))
.send()?
.into_response()
}
}
impl<TDocument> DeleteRequestBuilder<AsyncSender, TDocument> {
pub fn send(self) -> Pending {
let req = self.inner.into_request();
let res_future = RequestBuilder::new(self.client, self.params, RawRequestInner::new(req))
.send()
.and_then(|res| res.into_response());
Pending::new(res_future)
}
}
pub struct Pending {
inner: Box<Future<Item = DeleteResponse, Error = Error>>,
}
impl Pending {
fn new<F>(fut: F) -> Self
where
F: Future<Item = DeleteResponse, Error = Error> + 'static,
{
Pending {
inner: Box::new(fut),
}
}
}
impl Future for Pending {
type Item = DeleteResponse;
type Error = Error;
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
self.inner.poll()
}
}
#[cfg(test)]
mod tests {
use serde_json::Value;
use prelude::*;
#[test]
fn default_request() {
let client = SyncClientBuilder::new().build().unwrap();
let req = client
.document_delete::<Value>(index("test-idx"), id("1"))
.inner
.into_request();
assert_eq!("/test-idx/value/1", req.url.as_ref());
}
#[test]
fn specify_ty() {
let client = SyncClientBuilder::new().build().unwrap();
let req = client
.document_delete::<Value>(index("test-idx"), id("1"))
.ty("new-ty")
.inner
.into_request();
assert_eq!("/test-idx/new-ty/1", req.url.as_ref());
}
}