From 6336cc3d7b8de71b15ff9ee9e11d537e443519df Mon Sep 17 00:00:00 2001
From: Sean McArthur <sean@seanmonstar.com>
Date: Thu, 6 Jan 2022 10:12:29 -0800
Subject: [PATCH] Add Error::is_go_away() and Error::is_remote()

---
 src/error.rs | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/src/error.rs b/src/error.rs
index fdbfc0d..6c8f6ed 100644
--- a/src/error.rs
+++ b/src/error.rs
@@ -57,7 +57,7 @@ impl Error {
         }
     }
 
-    /// Returns the true if the error is an io::Error
+    /// Returns true if the error is an io::Error
     pub fn is_io(&self) -> bool {
         match self.kind {
             Kind::Io(_) => true,
@@ -86,6 +86,21 @@ impl Error {
             kind: Kind::Io(err),
         }
     }
+
+    /// Returns true if the error is from a `GOAWAY`.
+    pub fn is_go_away(&self) -> bool {
+        matches!(self.kind, Kind::GoAway(..))
+    }
+
+    /// Returns true if the error was received in a frame from the remote.
+    ///
+    /// Such as from a received `RST_STREAM` or `GOAWAY` frame.
+    pub fn is_remote(&self) -> bool {
+        matches!(
+            self.kind,
+            Kind::GoAway(_, _, Initiator::Remote) | Kind::Reset(_, _, Initiator::Remote)
+        )
+    }
 }
 
 impl From<proto::Error> for Error {