Let empty arrays being flattened not alloc additional state

ruby · Oct 25, 2023 · cd18b09 · cd18b09
1 parent a6a67b0
commit cd18b09
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/array.c b/array.c
@@ -6306,6 +6306,12 @@ flatten(VALUE ary, int level)
     st_table *memo = 0;
     st_data_t id;
 
+    if (UNLIKELY(RARRAY_LEN(ary) == 0)) {
+        result = ary_new(0, 0);
+        RBASIC_SET_CLASS(result, rb_obj_class(ary));
+        return result;
+    }
+
     for (i = 0; i < RARRAY_LEN(ary); i++) {
         elt = RARRAY_AREF(ary, i);
         tmp = rb_check_array_type(elt);