From f9724fa0ec8f1482fa911f24a0ce6c3a37931385 Mon Sep 17 00:00:00 2001
From: Nikolay Kim <fafhrd91@gmail.com>
Date: Mon, 11 Feb 2019 09:54:41 -0800
Subject: [PATCH] add ErrorResponse impl for TimeoutError

---
 Cargo.toml   |  3 +--
 src/error.rs | 11 +++++++++++
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index f940ae86c..5bcaea7c3 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -40,7 +40,7 @@ ssl = ["openssl"]
 actix-service = "0.2.1"
 actix-codec = "0.1.0"
 actix-connector = "0.2.0"
-actix-utils = "0.2.1"
+actix-utils = "0.2.2"
 
 base64 = "0.10"
 backtrace = "0.3"
@@ -80,7 +80,6 @@ openssl = { version="0.10", optional = true }
 actix-rt = "0.1.0"
 actix-server = { version="0.2", features=["ssl"] }
 actix-connector =  { version="0.2.0", features=["ssl"] }
-actix-utils = "0.2.1"
 actix-http-test = { path="test-server", features=["ssl"] }
 env_logger = "0.6"
 serde_derive = "1.0"
diff --git a/src/error.rs b/src/error.rs
index f71b429fd..cd5cabaa6 100644
--- a/src/error.rs
+++ b/src/error.rs
@@ -6,6 +6,7 @@ use std::string::FromUtf8Error;
 use std::{fmt, io, result};
 
 // use actix::MailboxError;
+use actix_utils::timeout::TimeoutError;
 use backtrace::Backtrace;
 use cookie;
 use derive_more::{Display, From};
@@ -187,6 +188,16 @@ impl<T: ResponseError + 'static> From<T> for Error {
 //     }
 // }
 
+/// Return `GATEWAY_TIMEOUT` for `TimeoutError`
+impl<E: ResponseError> ResponseError for TimeoutError<E> {
+    fn error_response(&self) -> Response {
+        match self {
+            TimeoutError::Service(e) => e.error_response(),
+            TimeoutError::Timeout => Response::new(StatusCode::GATEWAY_TIMEOUT),
+        }
+    }
+}
+
 /// `InternalServerError` for `JsonError`
 impl ResponseError for JsonError {}