OMGeeky/tarpc
1
0
Fork 0
mirror of https://github.com/OMGeeky/tarpc.git synced 2025-12-31 08:33:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #7 from shaladdle/macro_try_clone

Browse Source 
Add TryClone to Client/AsyncClient
This commit is contained in:
Tim
2016-02-14 20:19:23 -08:00
parent 7d028f6076 04e513309a
commit 4b3e715313
1 changed files with 30 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
tarpc/src/macros.rs
View File

@@ -265,6 +265,12 @@ macro_rules! service_inner {
$fn_name($($arg: $in_),*) -> $out
)*
);
#[doc="Attempt to clone the client object. This might fail if the underlying TcpStream \
clone fails."]
pub fn try_clone(&self) -> ::std::io::Result<Self> {
::std::result::Result::Ok(Client(try!(self.0.try_clone())))
}
}
#[doc="The client stub that makes asynchronous RPC calls to the server."]
@@ -286,6 +292,12 @@ macro_rules! service_inner {
$fn_name($($arg: $in_),*) -> $out
)*
);
#[doc="Attempt to clone the client object. This might fail if the underlying TcpStream \
clone fails."]
pub fn try_clone(&self) -> ::std::io::Result<Self> {
::std::result::Result::Ok(AsyncClient(try!(self.0.try_clone())))
}
}
struct __Server<S: 'static + Service>(S);
@@ -386,6 +398,24 @@ mod functional_test {
handle.shutdown();
}
#[test]
fn try_clone() {
let handle = serve( "localhost:0", Server, test_timeout()).unwrap();
let client1 = Client::new(handle.local_addr(), None).unwrap();
let client2 = client1.try_clone().unwrap();
assert_eq!(3, client1.add(1, 2).unwrap());
assert_eq!(3, client2.add(1, 2).unwrap());
}
#[test]
fn async_try_clone() {
let handle = serve("localhost:0", Server, test_timeout()).unwrap();
let client1 = AsyncClient::new(handle.local_addr(), None).unwrap();
let client2 = client1.try_clone().unwrap();
assert_eq!(3, client1.add(1, 2).get().unwrap());
assert_eq!(3, client2.add(1, 2).get().unwrap());
}
// Tests that a server can be wrapped in an Arc; no need to run, just compile
#[allow(dead_code)]
fn serve_arc_server() {