ARROW-3878: [Rust] Improve primitive types

Author: sunchao

.travis.yml

@@ -267,6 +267,13 @@ matrix:
     - $TRAVIS_BUILD_DIR/ci/travis_script_ruby.sh
   # Rust
   - language: rust
+    rust:
+      - stable
+      - nightly
+    matrix:
+      allow_failures:
+        - rust: stable
+      fast_finish: true
     cache: cargo
     addons:
       apt:
@@ -280,8 +287,7 @@ matrix:
     - if [ $ARROW_CI_RUST_AFFECTED != "1" ]; then exit; fi
     - $TRAVIS_BUILD_DIR/ci/travis_install_cargo.sh
     script:
-    - RUSTUP_TOOLCHAIN=stable $TRAVIS_BUILD_DIR/ci/travis_script_rust.sh
-    - RUSTUP_TOOLCHAIN=nightly $TRAVIS_BUILD_DIR/ci/travis_script_rust.sh || true
+    - $TRAVIS_BUILD_DIR/ci/travis_script_rust.sh
     after_success:
     - pushd ${TRAVIS_BUILD_DIR}/rust
     # Run coverage for codecov.io

rust/examples/builders.rs

@@ -18,7 +18,7 @@
 ///! Many builders are available to easily create different types of arrow arrays
 extern crate arrow;
 
-use arrow::builder::*;
+use arrow::builder::{ArrayBuilder, Int32Builder};
 
 fn main() {
     // Primitive Arrays
@@ -27,7 +27,7 @@ fn main() {
     // i32, i64, f32, f64)
 
     // Create a new builder with a capacity of 100
-    let mut primitive_array_builder = PrimitiveArrayBuilder::<i32>::new(100);
+    let mut primitive_array_builder = Int32Builder::new(100);
 
     // Push an individual primitive value
     primitive_array_builder.push(55).unwrap();

rust/examples/dynamic_types.rs

@@ -18,6 +18,7 @@
 ///! This example demonstrates dealing with mixed types dynamically at runtime
 use std::sync::Arc;
 
+#[macro_use(numeric_from_nonnull)]
 extern crate arrow;
 
 use arrow::array::*;
@@ -40,7 +41,7 @@ fn main() {
     ]);
 
     // create some data
-    let id = PrimitiveArray::from(vec![1, 2, 3, 4, 5]);
+    let id = numeric_from_nonnull!(Int32Array, 1, 2, 3, 4, 5);
 
     let nested = StructArray::from(vec![
         (
@@ -49,11 +50,11 @@ fn main() {
         ),
         (
             Field::new("b", DataType::Float64, false),
-            Arc::new(PrimitiveArray::from(vec![1.1, 2.2, 3.3, 4.4, 5.5])),
+            Arc::new(numeric_from_nonnull!(Float64Array, 1.1, 2.2, 3.3, 4.4, 5.5)),
         ),
         (
             Field::new("c", DataType::Float64, false),
-            Arc::new(PrimitiveArray::from(vec![2.2, 3.3, 4.4, 5.5, 6.6])),
+            Arc::new(numeric_from_nonnull!(Float64Array, 2.2, 3.3, 4.4, 5.5, 6.6)),
         ),
     ]);
 
@@ -75,12 +76,12 @@ fn process(batch: &RecordBatch) {
     let _nested_b = nested
         .column(1)
         .as_any()
-        .downcast_ref::<PrimitiveArray<f64>>()
+        .downcast_ref::<Float64Array>()
         .unwrap();
-    let nested_c: &PrimitiveArray<f64> = nested
+    let nested_c: &Float64Array = nested
         .column(2)
         .as_any()
-        .downcast_ref::<PrimitiveArray<f64>>()
+        .downcast_ref::<Float64Array>()
         .unwrap();
 
     let projected_schema = Schema::new(vec![
@@ -91,8 +92,8 @@ fn process(batch: &RecordBatch) {
     let _ = RecordBatch::new(
         Arc::new(projected_schema),
         vec![
-            id.clone(), //NOTE: this is cloning the Arc not the array data
-            Arc::new(PrimitiveArray::<f64>::from(nested_c.data())),
+            id.clone(), // NOTE: this is cloning the Arc not the array data
+            Arc::new(Float64Array::from(nested_c.data())),
         ],
     );
 }

rust/examples/read_csv.rs

@@ -17,7 +17,7 @@
 
 extern crate arrow;
 
-use arrow::array::{BinaryArray, PrimitiveArray};
+use arrow::array::{BinaryArray, Float64Array};
 use arrow::csv;
 use arrow::datatypes::{DataType, Field, Schema};
 use std::fs::File;
@@ -49,12 +49,12 @@ fn main() {
     let lat = batch
         .column(1)
         .as_any()
-        .downcast_ref::<PrimitiveArray<f64>>()
+        .downcast_ref::<Float64Array>()
         .unwrap();
     let lng = batch
         .column(2)
         .as_any()
-        .downcast_ref::<PrimitiveArray<f64>>()
+        .downcast_ref::<Float64Array>()
         .unwrap();
 
     for i in 0..batch.num_rows() {