From 1e0d1676023d6433adb1ee63886fdd6d07152663 Mon Sep 17 00:00:00 2001
From: everpcpc <everpcpc@users.noreply.github.com>
Date: Sat, 15 Apr 2023 13:10:30 +0800
Subject: [PATCH] feat(driver): add affected rows with exec for RestAPI handler
 (#60)

---
 core/Cargo.toml        | 2 +-
 driver/Cargo.toml      | 4 ++--
 driver/src/rest_api.rs | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/core/Cargo.toml b/core/Cargo.toml
index cd4206c6d..5f9822014 100644
--- a/core/Cargo.toml
+++ b/core/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "databend-client"
-version = "0.1.8"
+version = "0.1.9"
 edition = "2021"
 license = "Apache-2.0"
 description = "Databend Client for Rust"
diff --git a/driver/Cargo.toml b/driver/Cargo.toml
index 82bcd94ff..bd1e4919c 100644
--- a/driver/Cargo.toml
+++ b/driver/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "databend-driver"
-version = "0.2.8"
+version = "0.2.9"
 edition = "2021"
 license = "Apache-2.0"
 description = "Databend Driver for Rust"
@@ -21,7 +21,7 @@ flight-sql = ["dep:arrow", "dep:arrow-array", "dep:arrow-cast", "dep:arrow-fligh
 [dependencies]
 async-trait = "0.1.68"
 chrono = { version = "0.4.24", default-features = false, features = ["clock"] }
-databend-client = { version = "0.1.8", path = "../core" }
+databend-client = { version = "0.1.9", path = "../core" }
 dyn-clone = "1.0.11"
 http = "0.2.9"
 serde = { version = "1.0.156", default-features = false, features = ["derive"] }
diff --git a/driver/src/rest_api.rs b/driver/src/rest_api.rs
index c70ebeed3..27fb34577 100644
--- a/driver/src/rest_api.rs
+++ b/driver/src/rest_api.rs
@@ -51,7 +51,7 @@ impl Connection for RestAPIConnection {
         while let Some(next_uri) = resp.next_uri {
             resp = self.client.query_page(&next_uri).await?;
         }
-        Ok(0)
+        Ok(resp.stats.progresses.write_progress.rows as i64)
     }
 
     async fn query_iter(&mut self, sql: &str) -> Result<RowIterator> {