tarpc/examples/two_clients.rs

// Copyright 2016 Google Inc. All Rights Reserved.
//
// Licensed under the MIT License, <LICENSE or http://opensource.org/licenses/MIT>.
// This file may not be copied, modified, or distributed except according to those terms.

#![feature(conservative_impl_trait, plugin)]
#![plugin(tarpc_plugins)]

#[macro_use]
extern crate log;
#[macro_use]
extern crate tarpc;
extern crate bincode;
extern crate env_logger;
extern crate futures;

use bar::FutureServiceExt as BarExt;
use baz::FutureServiceExt as BazExt;
use futures::Future;
use tarpc::{client, server};
use tarpc::client::sync::Connect;
use tarpc::util::{FirstSocketAddr, Never};

mod bar {
    service! {
        rpc bar(i: i32) -> i32;
    }
}

#[derive(Clone)]
struct Bar;
impl bar::FutureService for Bar {
    type BarFut = Result<i32, Never>;

    fn bar(&self, i: i32) -> Self::BarFut {
        Ok(i)
    }
}

mod baz {
    service! {
        rpc baz(s: String) -> String;
    }
}

#[derive(Clone)]
struct Baz;
impl baz::FutureService for Baz {
    type BazFut = Result<String, Never>;

    fn baz(&self, s: String) -> Self::BazFut {
        Ok(format!("Hello, {}!", s))
    }
}

macro_rules! pos {
    () => (concat!(file!(), ":", line!()))
}

fn main() {
    let _ = env_logger::init();
    let bar_addr = Bar.listen("localhost:0".first_socket_addr(),
                server::Options::default())
        .wait()
        .unwrap();
    let baz_addr = Baz.listen("localhost:0".first_socket_addr(),
                server::Options::default())
        .wait()
        .unwrap();

    let bar_client = bar::SyncClient::connect(bar_addr, client::Options::default()).unwrap();
    let baz_client = baz::SyncClient::connect(baz_addr, client::Options::default()).unwrap();

    info!("Result: {:?}", bar_client.bar(17));

    let total = 20;
    for i in 1..(total + 1) {
        if i % 2 == 0 {
            info!("Result 1: {:?}", bar_client.bar(i));
        } else {
            info!("Result 2: {:?}", baz_client.baz(i.to_string()));
        }
    }

    info!("Done.");
}