Merge pull request #1275 from froydnj/froydnj-ast-ary-capa

preallocate arrays when building the Ruby AST
ruby · Aug 17, 2023 · bf10d91 · bf10d91
2 parents c57436c + c2e92e9
commit bf10d91
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/templates/ext/yarp/api_node.c.erb b/templates/ext/yarp/api_node.c.erb
@@ -146,22 +146,22 @@ yp_ast_new(yp_parser_t *parser, yp_node_t *node, rb_encoding *encoding) {
                     <%- when NodeParam, OptionalNodeParam -%>
                     argv[<%= index %>] = rb_ary_pop(value_stack);
                     <%- when NodeListParam -%>
-                    argv[<%= index %>] = rb_ary_new();
+                    argv[<%= index %>] = rb_ary_new_capa(cast-><%= param.name %>.size);
                     for (size_t index = 0; index < cast-><%= param.name %>.size; index++) {
                         rb_ary_push(argv[<%= index %>], rb_ary_pop(value_stack));
                     }
                     <%- when StringParam -%>
                     argv[<%= index %>] = yp_string_new(&cast-><%= param.name %>, encoding);
                     <%- when LocationListParam -%>
-                    argv[<%= index %>] = rb_ary_new();
+                    argv[<%= index %>] = rb_ary_new_capa(cast-><%= param.name %>.size);
                     for (size_t index = 0; index < cast-><%= param.name %>.size; index++) {
                         yp_location_t location = cast-><%= param.name %>.locations[index];
                         rb_ary_push(argv[<%= index %>], yp_location_new(parser, location.start, location.end, source));
                     }
                     <%- when ConstantParam -%>
                     argv[<%= index %>] = rb_id2sym(constants[cast-><%= param.name %> - 1]);
                     <%- when ConstantListParam -%>
-                    argv[<%= index %>] = rb_ary_new();
+                    argv[<%= index %>] = rb_ary_new_capa(cast-><%= param.name %>.size);
                     for (size_t index = 0; index < cast-><%= param.name %>.size; index++) {
                         rb_ary_push(argv[<%= index %>], rb_id2sym(constants[cast-><%= param.name %>.ids[index] - 1]));
                     }