fix(driver): kill instead of finish query for query row

databendlabs · Oct 6, 2023 · 7fd0bde · 7fd0bde
1 parent 479d4a2
commit 7fd0bde
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/driver/src/rest_api.rs b/driver/src/rest_api.rs
@@ -75,7 +75,7 @@ impl Connection for RestAPIConnection {
     async fn query_row(&self, sql: &str) -> Result<Option<Row>> {
         let resp = self.client.query(sql).await?;
         let resp = self.wait_for_data(resp).await?;
-        self.finish_query(resp.final_uri).await?;
+        self.kill_query(&resp).await?;
         let schema = resp.schema.try_into()?;
         if resp.data.is_empty() {
             Ok(None)
@@ -152,10 +152,10 @@ impl<'o> RestAPIConnection {
         Ok(result)
     }
 
-    async fn finish_query(&self, final_uri: Option<String>) -> Result<QueryResponse> {
-        match final_uri {
-            Some(uri) => self.client.query_page(&uri).await.map_err(|e| e.into()),
-            None => Err(Error::InvalidResponse("final_uri is empty".to_string())),
+    async fn kill_query(&self, resp: &QueryResponse) -> Result<QueryResponse> {
+        match &resp.kill_uri {
+            Some(uri) => self.client.query_page(uri).await.map_err(|e| e.into()),
+            None => Err(Error::InvalidResponse("kill_uri is empty".to_string())),
         }
     }