Skip to content

Further optimize Array#product#6753

Merged
matz merged 1 commit intomruby:masterfrom
dearblue:array.product
Mar 21, 2026
Merged

Further optimize Array#product#6753
matz merged 1 commit intomruby:masterfrom
dearblue:array.product

Conversation

@dearblue
Copy link
Contributor

Replace __product_group method with __product_generate and __product_next.
This change eliminates the need for Ruby to perform internal state calculations, allowing it to simply receive the results.

Replace `__product_group` method with `__product_generate` and `__product_next`.
This change eliminates the need for Ruby to perform internal state calculations, allowing it to simply receive the results.
@dearblue dearblue requested a review from matz as a code owner March 20, 2026 12:31
@gemini-code-assist
Copy link
Contributor

Summary of Changes

Hello, I'm Gemini Code Assist1! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed!

This pull request significantly optimizes the Array#product method by shifting the complex iteration logic and state management from Ruby to C. By introducing a C-level generator, the Ruby implementation of product now simply requests the next product group, eliminating the need for Ruby to perform internal state calculations and improving overall performance.

Highlights

  • Ruby Array#product Refactoring: The Array#product method in mrblib/array.rb was refactored to delegate its core logic to new C-level helper methods, __product_generate and __product_next.
  • C-level Generator Implementation: New C functions, ary_product_generate and ary_product_next, were introduced in src/array.c to manage the iteration state and yield product groups more efficiently. This includes a new ary_product_generator struct for state management.
  • Optimized Group Fetching: The internal ary_product_group C function was renamed to ary_product_fetch and its signature and logic were updated to serve as a helper for fetching individual product groups, now utilized by the new generator functions.
  • Reduced Ruby Overhead: The change eliminates the need for Ruby to perform internal state calculations for Array#product, allowing it to simply receive the results from the C-level generator, leading to performance optimization.

🧠 New Feature in Public Preview: You can now enable Memory to help Gemini Code Assist learn from your team's feedback. This makes future code reviews more consistent and personalized to your project's style. Click here to enable Memory in your admin console.

Using Gemini Code Assist

The full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips.

Invoking Gemini

You can request assistance from Gemini at any point by creating a comment using either /gemini <command> or @gemini-code-assist <command>. Below is a summary of the supported commands on the current page.

Feature Command Description
Code Review /gemini review Performs a code review for the current pull request in its current state.
Pull Request Summary /gemini summary Provides a summary of the current pull request in its current state.
Comment @gemini-code-assist Responds in comments when explicitly tagged, both in pull request comments and review comments.
Help /gemini help Displays a list of available commands.

Customization

To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a .gemini/ folder in the base of the repository. Detailed instructions can be found here.

Limitations & Feedback

Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for GitHub and other Google products, sign up here.

Footnotes

  1. Review the Privacy Notices, Generative AI Prohibited Use Policy, Terms of Service, and learn how to configure Gemini Code Assist in GitHub here. Gemini can make mistakes, so double check it and use code with caution.

Copy link
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

The code changes refactor the product method in mruby-array-ext to use a generator pattern for improved performance, especially when used with a block. The reviewer suggests further optimization by storing the arys array within the generator object to reduce overhead when calling __product_next.

Comment on lines +1457 to +1468
ary_product_next(mrb_state *mrb, mrb_value self)
{
mrb_value arys;
struct ary_product_generator *g;
mrb_get_args(mrb, "Ad", &arys, &g, &ary_product_generator_type);

if (g->cursor >= g->total) {
return mrb_nil_value();
}

return ary_product_fetch(mrb, self, arys, g->cursor++);
}
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

For further optimization, consider storing the arys array within the generator object itself rather than passing it from Ruby on every call to __product_next. This would reduce the overhead of passing the array across the Ruby/C boundary in each iteration of the while loop.

This would involve:

  1. Adding a mrb_value arys; field to the struct ary_product_generator.
  2. Storing arys_ary in this new field within ary_product_generate and adding a write barrier.
  3. Adding a mark function to ary_product_generator_type to ensure arys is not garbage collected.
  4. Modifying ary_product_next to retrieve arys from the generator struct instead of from its arguments.
  5. Updating the call in array.rb to __product_next(gen) and the method signature in the C method table.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@gemini-code-assist
In the current version of mruby, there is no way to protect an object from the garbage collector once it has been placed in a user-defined structure.
However, placing it in a hidden instance variable is not memory-efficient.
Therefore, I believe the current patch strikes a good balance.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you for clarifying the mruby garbage collector's limitations regarding protecting objects placed directly in user-defined C structures. Given this constraint, and the potential memory inefficiency of using hidden instance variables, I agree that the current implementation strikes a good balance between performance and memory management. It's important to work within the existing GC capabilities.

@matz matz merged commit ff5e6a2 into mruby:master Mar 21, 2026
18 of 19 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants