1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

#![feature(generators)]

mod error;
mod pooling;
mod socket;
mod stream;
mod sink;
mod sink_pool;


#[cfg(test)]
pub mod test_request;

pub use self::error::KfSocketError;
pub use self::socket::KfSocket;
pub use pooling::SocketPool;
pub use sink_pool::SinkPool;
pub use sink_pool::SharedSinkPool;
pub use stream::KfStream;
pub use sink::KfSink;
pub use sink::ExclusiveKfSink;


use std::net::SocketAddr;

use kf_protocol::api::Request;
use kf_protocol::api::RequestMessage;
use kf_protocol::api::ResponseMessage;


/// send request and return response from calling server at socket addr
pub async fn send_and_receive<R>(
    addr: SocketAddr,
    request: &RequestMessage<R>,
) -> Result<ResponseMessage<R::Response>, KfSocketError>
where
    R: Request,
{
    let mut client = KfSocket::connect(&addr).await?;

    let msgs: ResponseMessage<R::Response> = client.send(&request).await?;

    Ok(msgs)
}