update to latest tower and tower-hyper updates

tower-rs · Apr 25, 2019 · 0124184 · 0124184
1 parent f7aab6c
commit 0124184
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 27 deletions.
diff --git a/src/http-client.rs b/src/http-client.rs
@@ -4,10 +4,11 @@ use hyper::{
     Request, Response, Uri,
 };
 use std::time::Duration;
-use tower::{builder::ServiceBuilder, reconnect::Reconnect, Service, ServiceExt};
+use tower::{MakeService, Service};
+use tower::builder::ServiceBuilder;
 use tower_hyper::{
+    Body,
     client::{Builder, Connect},
-    retry::{Body, RetryPolicy},
     util::Connector,
 };
 
@@ -24,40 +25,30 @@ fn request() -> impl Future<Item = Response<hyper::Body>, Error = ()> {
     let connector = Connector::new(HttpConnector::new(1));
     let hyper = Connect::new(connector, Builder::new());
 
-    // RetryPolicy is a very simple policy that retries `n` times
-    // if the response has a 500 status code. Here, `n` is 5.
-    let policy = RetryPolicy::new(5);
     // We're calling the tower/examples/server.rs.
     let dst = Destination::try_from_uri(Uri::from_static("http://127.0.0.1:3000")).unwrap();
 
-    // Now, to build the service! We use two BufferLayers in order to:
-    // - provide backpressure for the RateLimitLayer, and ConcurrencyLimitLayer
-    // - meet `RetryLayer`'s requirement that our service implement `Service + Clone`
-    // - ..and to provide cheap clones on the service.
-    let maker = ServiceBuilder::new()
+    // Now, to build the service!
+    let mut maker = ServiceBuilder::new()
         .buffer(5)
         .rate_limit(5, Duration::from_secs(1))
         .concurrency_limit(5)
-        .retry(policy)
-        .buffer(5)
-        .make_service(hyper);
+        .service(hyper);
 
-    // `Reconnect` accepts a destination and a MakeService, creating a new service
-    // any time the connection encounters an error.
-    let client = Reconnect::new(maker, dst);
+    let client = maker
+        .make_service(dst)
+        .map_err(|err| eprintln!("Connect Error {:?}", err));
 
     let request = Request::builder()
         .method("GET")
-        .body(Body::from(Vec::new()))
+        .body(Vec::new())
         .unwrap();
 
-    // we check to see if the client is ready to accept requests.
     client
-        .ready()
         .map_err(|e| panic!("Service is not ready: {:?}", e))
         .and_then(|mut c| {
             c.call(request)
-                .map(|res| res.map(|b| b.into_inner()))
+                .map(|res| res.map(Body::into_inner))
                 .map_err(|e| panic!("{:?}", e))
         })
 }
diff --git a/src/http-server.rs b/src/http-server.rs
@@ -1,8 +1,8 @@
 use futures::{future, Future, Poll, Stream};
-use hyper::{self, Body, Request, Response};
+use hyper::{self, Request, Response};
 use tokio::net::TcpListener;
 use tower::{builder::ServiceBuilder, Service};
-use tower_hyper::{body::LiftBody, server::Server};
+use tower_hyper::{Body, server::Server};
 
 fn main() {
     hyper::rt::run(future::lazy(|| {
@@ -13,7 +13,7 @@ fn main() {
 
         let maker = ServiceBuilder::new()
             .concurrency_limit(5)
-            .make_service(MakeSvc);
+            .service(MakeSvc);
 
         let server = Server::new(maker);
 
@@ -37,17 +37,17 @@ fn main() {
 }
 
 struct Svc;
-impl Service<Request<LiftBody<Body>>> for Svc {
-    type Response = Response<&'static str>;
+impl Service<Request<Body>> for Svc {
+    type Response = Response<Body>;
     type Error = hyper::Error;
     type Future = future::FutureResult<Self::Response, Self::Error>;
 
     fn poll_ready(&mut self) -> Poll<(), Self::Error> {
         Ok(().into())
     }
 
-    fn call(&mut self, _req: Request<LiftBody<Body>>) -> Self::Future {
-        let res = Response::new("Hello World!");
+    fn call(&mut self, _req: Request<Body>) -> Self::Future {
+        let res = Response::new(Body::from(hyper::Body::from("Hello World!")));
         future::ok(res)
     }
 }