feat: the first resp may not contain schema. (#363)

feature: the first resp may not contain schema. Co-authored-by: sundyli <[email protected]>
databendlabs · Mar 13, 2024 · aa0ad1c · aa0ad1c
1 parent 73b4c8d
commit aa0ad1c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/driver/src/rest_api.rs b/driver/src/rest_api.rs
@@ -266,6 +266,9 @@ impl Stream for RestAPIRows {
             Some(ref mut next_page) => match Pin::new(next_page).poll(cx) {
                 Poll::Ready(Ok(resp)) => {
                     self.data = resp.data.into();
+                    if self.schema.fields().is_empty() {
+                        self.schema = Arc::new(resp.schema.try_into()?);
+                    }
                     self.query_id = resp.id;
                     self.next_uri = resp.next_uri;
                     self.next_page = None;