update affected rows usage

Author: niebayes

go/util.go

@@ -19,7 +19,7 @@ func PrintAffectedRows(records []arrow.Record) {
 	defer releaseRecords(records)
 
 	// By Datalayers' design, the affected rows is the value at the first row and the first column.
-	affectedRows := records[0].Column(0).(*array.String).Value(0)
+	affectedRows := records[0].Column(0).(*array.Int64).Value(0)
 	fmt.Println("Affected rows: ", affectedRows)
 }
 

python/util.py

@@ -5,7 +5,7 @@
 
 
 def print_affected_rows(df: pandas.DataFrame) -> int:
-    print("Affected rows: {}".format(df["Count"][0]))
+    print("Affected rows: {}".format(df["affected_rows"][0]))
 
 
 def make_insert_binding() -> pa.RecordBatch:

rust/src/util.rs

@@ -2,7 +2,7 @@ use std::process::exit;
 use std::sync::Arc;
 
 use arrow_array::{
-    Float32Array, Int32Array, Int8Array, RecordBatch, StringArray, TimestampMillisecondArray,
+    Float32Array, Int32Array, Int64Array, Int8Array, RecordBatch, TimestampMillisecondArray,
 };
 use arrow_cast::pretty::pretty_format_batches;
 use arrow_schema::{DataType, Field, Schema, TimeUnit};
@@ -30,12 +30,9 @@ pub fn print_affected_rows(batches: &[RecordBatch]) {
         .unwrap()
         .column(0)
         .as_any()
-        .downcast_ref::<StringArray>()
+        .downcast_ref::<Int64Array>()
         .unwrap()
-        .value(0)
-        .to_string()
-        .parse::<u32>()
-        .unwrap();
+        .value(0);
     println!("Affected rows: {}", affected_rows);
 }