using array

Author: Jiayu Liu

datafusion/src/physical_plan/math_expressions.rs

@@ -16,13 +16,13 @@
 // under the License.
 
 //! Math expressions
-
 use super::{ColumnarValue, ScalarValue};
 use crate::error::{DataFusionError, Result};
 use arrow::array::{make_array, Array, ArrayData, Float32Array, Float64Array};
 use arrow::buffer::Buffer;
 use arrow::datatypes::{DataType, ToByteSlice};
 use rand::{thread_rng, Rng};
+use std::sync::Arc;
 
 macro_rules! compute_op {
     ($ARRAY:expr, $FUNC:ident, $TYPE:ident) => {{
@@ -118,10 +118,15 @@ math_unary_function!("log2", log2);
 math_unary_function!("log10", log10);
 
 /// random SQL function
-pub fn random(_: &[ColumnarValue]) -> Result<ColumnarValue> {
+pub fn random(args: &[ColumnarValue]) -> Result<ColumnarValue> {
+    let len = match &args[0] {
+        ColumnarValue::Array(array) => array.len(),
+        _ => return Err(DataFusionError::Internal("d".to_string())),
+    };
     let mut rng = thread_rng();
-    let value: f64 = rng.gen_range(0.0..1.0);
-    let scalar = ScalarValue::Float64(Some(value));
-    let column_value = ColumnarValue::Scalar(scalar);
-    Ok(column_value)
+    let mut array = Vec::with_capacity(len);
+    for _ in 0..len {
+        array.push(Some(rng.gen_range(0.0..1.0)))
+    }
+    Ok(ColumnarValue::Array(Arc::new(Float64Array::from(array))))
 }