fix(driver): drop next_page on page error (#77)

databendlabs · Apr 19, 2023 · cdf6de6 · cdf6de6
1 parent fec7658
commit cdf6de6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/driver/src/rest_api.rs b/driver/src/rest_api.rs
@@ -193,7 +193,10 @@ impl Stream for RestAPIRows {
                     let progress = QueryProgress::from(resp.stats.progresses);
                     Poll::Ready(Some(Ok(RowWithProgress::Progress(progress))))
                 }
-                Poll::Ready(Err(e)) => Poll::Ready(Some(Err(e))),
+                Poll::Ready(Err(e)) => {
+                    self.next_page = None;
+                    Poll::Ready(Some(Err(e)))
+                }
                 Poll::Pending => {
                     cx.waker().wake_by_ref();
                     Poll::Pending