From e828f9135d886dfb43b2284ada289124c7b5a1d3 Mon Sep 17 00:00:00 2001
From: ekexium <ekexium@gmail.com>
Date: Wed, 3 Mar 2021 12:39:30 +0800
Subject: [PATCH] feat: ExtractError also checks region_error

Signed-off-by: ekexium <ekexium@gmail.com>
---
 src/request/plan.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/request/plan.rs b/src/request/plan.rs
index 2f44d2e..07975dc 100644
--- a/src/request/plan.rs
+++ b/src/request/plan.rs
@@ -265,7 +265,7 @@ impl<P: Plan> Clone for ExtractError<P> {
 /// returns an `Err` wrapping the error.
 ///
 /// The errors come from two places: `Err` from inner plans, and `Ok(response)`
-/// where `response` contains unresolved errors.
+/// where `response` contains unresolved errors (`error` and `region_error`).
 #[async_trait]
 impl<P: Plan> Plan for ExtractError<P>
 where
@@ -277,6 +277,8 @@ where
         let mut result = self.inner.execute().await?;
         if let Some(error) = result.error() {
             Err(error)
+        } else if let Some(error) = result.region_error() {
+            Err(error)
         } else {
             Ok(result)
         }