Revert "return Int64 instead of UInt64 when returning count (#17)"

This reverts commit 314a726.

Author: YaroslavLitvinov

datafusion/core/tests/user_defined/insert_operation.rs

@@ -184,7 +184,7 @@ impl ExecutionPlan for TestInsertExec {
 fn make_count_schema() -> SchemaRef {
     Arc::new(Schema::new(vec![Field::new(
         "count",
-        DataType::Int64, // should return signed int for snowflake
+        DataType::UInt64,
         false,
     )]))
 }

datafusion/expr/src/logical_plan/dml.rs

@@ -203,8 +203,7 @@ impl Display for InsertOp {
 
 fn make_count_schema() -> DFSchemaRef {
     Arc::new(
-        // should return signed int for snowflake
-        Schema::new(vec![Field::new("count", DataType::Int64, false)])
+        Schema::new(vec![Field::new("count", DataType::UInt64, false)])
             .try_into()
             .unwrap(),
     )

datafusion/physical-plan/src/insert.rs

@@ -32,7 +32,7 @@ use crate::ExecutionPlanProperties;
 
 use arrow::datatypes::SchemaRef;
 use arrow::record_batch::RecordBatch;
-use arrow_array::{ArrayRef, Int64Array};
+use arrow_array::{ArrayRef, UInt64Array};
 use arrow_schema::{DataType, Field, Schema};
 use datafusion_common::{internal_err, Result};
 use datafusion_execution::TaskContext;
@@ -271,15 +271,16 @@ impl ExecutionPlan for DataSinkExec {
 /// +-------+,
 /// ```
 fn make_count_batch(count: u64) -> RecordBatch {
-    let array = Arc::new(Int64Array::from(vec![count as i64])) as ArrayRef;
+    let array = Arc::new(UInt64Array::from(vec![count])) as ArrayRef;
+
     RecordBatch::try_from_iter_with_nullable(vec![("count", array, false)]).unwrap()
 }
 
 fn make_count_schema() -> SchemaRef {
     // Define a schema.
     Arc::new(Schema::new(vec![Field::new(
         "count",
-        DataType::Int64, // should return signed int for snowflake
+        DataType::UInt64,
         false,
     )]))
 }