mirror of
https://github.com/OMGeeky/tarpc.git
synced 2025-12-31 08:33:51 +01:00
e75193c191
This required the breaking change of removing the Client trait. The intent of the Client trait was to facilitate the decorator pattern by allowing users to create their own Clients that added behavior on top of the base client. Unfortunately, this trait had become a maintenance burden, consistently causing issues with lifetimes and the lack of generic associated types. Specifically, it meant that Client impls could not use async fns, which is no longer tenable today.
57 lines
1.4 KiB
Rust
57 lines
1.4 KiB
Rust
use futures::prelude::*;
|
|
use std::io;
|
|
use tarpc::serde_transport;
|
|
use tarpc::{
|
|
client, context,
|
|
server::{BaseChannel, Incoming},
|
|
};
|
|
use tokio_serde::formats::Json;
|
|
|
|
#[tarpc::derive_serde]
|
|
#[derive(Debug, PartialEq)]
|
|
pub enum TestData {
|
|
Black,
|
|
White,
|
|
}
|
|
|
|
#[tarpc::service]
|
|
pub trait ColorProtocol {
|
|
async fn get_opposite_color(color: TestData) -> TestData;
|
|
}
|
|
|
|
#[derive(Clone)]
|
|
struct ColorServer;
|
|
|
|
#[tarpc::server]
|
|
impl ColorProtocol for ColorServer {
|
|
async fn get_opposite_color(self, _: context::Context, color: TestData) -> TestData {
|
|
match color {
|
|
TestData::White => TestData::Black,
|
|
TestData::Black => TestData::White,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn test_call() -> io::Result<()> {
|
|
let transport = tarpc::serde_transport::tcp::listen("localhost:56797", Json::default).await?;
|
|
let addr = transport.local_addr();
|
|
tokio::spawn(
|
|
transport
|
|
.take(1)
|
|
.filter_map(|r| async { r.ok() })
|
|
.map(BaseChannel::with_defaults)
|
|
.execute(ColorServer.serve()),
|
|
);
|
|
|
|
let transport = serde_transport::tcp::connect(addr, Json::default).await?;
|
|
let client = ColorProtocolClient::new(client::Config::default(), transport).spawn()?;
|
|
|
|
let color = client
|
|
.get_opposite_color(context::current(), TestData::White)
|
|
.await?;
|
|
assert_eq!(color, TestData::Black);
|
|
|
|
Ok(())
|
|
}
|