Fix array_sort for empty record batch

Author: joroKr21

datafusion/functions-nested/src/sort.rs

@@ -171,6 +171,12 @@ pub fn array_sort_inner(args: &[ArrayRef]) -> Result<ArrayRef> {
         return Ok(Arc::clone(&args[0]));
     }
 
+    let list_array = as_list_array(&args[0])?;
+    let row_count = list_array.len();
+    if row_count == 0 || list_array.value_type().is_null() {
+        return Ok(Arc::clone(&args[0]));
+    }
+
     if args[1..].iter().any(|array| array.is_null(0)) {
         return Ok(new_null_array(args[0].data_type(), args[0].len()));
     }
@@ -195,12 +201,6 @@ pub fn array_sort_inner(args: &[ArrayRef]) -> Result<ArrayRef> {
         _ => return exec_err!("array_sort expects 1 to 3 arguments"),
     };
 
-    let list_array = as_list_array(&args[0])?;
-    let row_count = list_array.len();
-    if row_count == 0 || list_array.value_type().is_null() {
-        return Ok(Arc::clone(&args[0]));
-    }
-
     let mut array_lengths = vec![];
     let mut arrays = vec![];
     let mut valid = NullBufferBuilder::new(row_count);

datafusion/sqllogictest/test_files/array.slt

@@ -2348,6 +2348,11 @@ NULL
 [NULL, 51, 52, 54, 55, 56, 57, 58, 59, 60]
 [61, 62, 63, 64, 65, 66, 67, 68, 69, 70]
 
+# test with empty table
+query ?
+select array_sort(column1, 'DESC', 'NULLS FIRST') from arrays_values where false;
+----
+
 # test with empty array
 query ?
 select array_sort([]);