fix(driver): don't call wake_by_ref in Poll::Pending (#350)

databendlabs · Feb 2, 2024 · 2004b52 · 2004b52
1 parent 22eb764
commit 2004b52
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/driver/src/flight_sql.rs b/driver/src/flight_sql.rs
@@ -377,10 +377,7 @@ impl Stream for FlightSQLRows {
             }
             Poll::Ready(Some(Err(err))) => Poll::Ready(Some(Err(err.into()))),
             Poll::Ready(None) => Poll::Ready(None),
-            Poll::Pending => {
-                cx.waker().wake_by_ref();
-                Poll::Pending
-            }
+            Poll::Pending => Poll::Pending,
         }
     }
 }
diff --git a/driver/src/rest_api.rs b/driver/src/rest_api.rs
@@ -276,10 +276,7 @@ impl Stream for RestAPIRows {
                     self.next_page = None;
                     Poll::Ready(Some(Err(e)))
                 }
-                Poll::Pending => {
-                    cx.waker().wake_by_ref();
-                    Poll::Pending
-                }
+                Poll::Pending => Poll::Pending,
             },
             None => match self.next_uri {
                 Some(ref next_uri) => {