some conversion

Author: aersam

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "lake2sql"
-version = "0.8.1"
+version = "0.8.3"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "maturin"
 [project]
 name = "lakeapi2sql"
 requires-python = ">=3.10"
-version = "0.8.2"
+version = "0.8.3"
 classifiers = [
     "Programming Language :: Rust",
     "Programming Language :: Python :: Implementation :: CPython",

src/arrow_convert.rs

@@ -352,11 +352,21 @@ pub(crate) fn get_token_rows<'a, 'b>(
             }
             arrow::datatypes::DataType::Float64 => {
                 let ba = col.as_any().downcast_ref::<Float64Array>().unwrap();
-
-                let mut rowindex = 0;
-                for val in ba.iter() {
-                    token_rows[rowindex].push(ColumnData::F64(val));
-                    rowindex += 1;
+                if coltype == &ColumnType::Int4 || coltype == &ColumnType::Intn {
+                    let mut rowindex = 0;
+                    for val in ba.iter() {
+                        token_rows[rowindex].push(ColumnData::I32(match val {
+                            Some(v) => Some(v as i32),
+                            None => None,
+                        }));
+                        rowindex += 1;
+                    }
+                } else {
+                    let mut rowindex = 0;
+                    for val in ba.iter() {
+                        token_rows[rowindex].push(ColumnData::F64(val));
+                        rowindex += 1;
+                    }
                 }
             }
             arrow::datatypes::DataType::Date32 => {