Skip to content

⚡ Bolt: [performance improvement] optimize D1 query generation memory allocations#239

Open
bashandbone wants to merge 1 commit into
mainfrom
bolt/d1-query-optimization-1083225261666678753
Open

⚡ Bolt: [performance improvement] optimize D1 query generation memory allocations#239
bashandbone wants to merge 1 commit into
mainfrom
bolt/d1-query-optimization-1083225261666678753

Conversation

@bashandbone
Copy link
Copy Markdown
Contributor

@bashandbone bashandbone commented May 17, 2026

💡 What: Used String::with_capacity and std::fmt::Write instead of standard format! and intermediate Vec bindings.
🎯 Why: Cloudflare D1 query construction via build_upsert_stmt and build_delete_stmt had overhead from unnecessary intermediate memory allocations for loops string formatting operations.
📊 Impact: Expected to reduce memory overhead and latency during Cloudflare D1 edge database target mutations.
🔬 Measurement: Verify via executing target tests with cargo test -p thread-flow --test d1_target_tests --test d1_minimal_tests --test d1_cache_integration to confirm standard D1 functional parity.


PR created automatically by Jules for task 1083225261666678753 started by @bashandbone

Summary by Sourcery

Optimize Cloudflare D1 upsert and delete SQL generation to reduce memory allocations and improve performance.

Enhancements:

  • Replace intermediate vector-based SQL construction with preallocated String builders for D1 upsert statements.
  • Streamline DELETE statement generation by building SQL directly into a preallocated String while collecting parameters.

…llocations in D1 query construction

💡 What: Used `String::with_capacity` and `std::fmt::Write` instead of standard `format!` and intermediate `Vec` bindings.
🎯 Why: Cloudflare D1 query construction via `build_upsert_stmt` and `build_delete_stmt` had overhead from unnecessary intermediate memory allocations for loops string formatting operations.
📊 Impact: Expected to reduce memory overhead and latency during Cloudflare D1 edge database target mutations.
🔬 Measurement: Verify via executing target tests with `cargo test -p thread-flow --test d1_target_tests --test d1_minimal_tests --test d1_cache_integration` to confirm standard D1 functional parity.

Co-authored-by: bashandbone <89049923+bashandbone@users.noreply.github.com>
Copilot AI review requested due to automatic review settings May 17, 2026 17:35
@google-labs-jules
Copy link
Copy Markdown
Contributor

👋 Jules, reporting for duty! I'm here to lend a hand with this pull request.

When you start a review, I'll add a 👀 emoji to each comment to let you know I've read it. I'll focus on feedback directed at me and will do my best to stay out of conversations between you and other bots or reviewers to keep the noise down.

I'll push a commit with your requested changes shortly after. Please note there might be a delay between these steps, but rest assured I'm on the job!

For more direct control, you can switch me to Reactive Mode. When this mode is on, I will only act on comments where you specifically mention me with @jules. You can find this option in the Pull Request section of your global Jules UI settings. You can always switch back!

New to Jules? Learn more at jules.google/docs.


For security, I will only act on instructions from the user who triggered this task.

@sourcery-ai
Copy link
Copy Markdown
Contributor

sourcery-ai Bot commented May 17, 2026

Reviewer's Guide

Refactors D1 upsert and delete SQL construction to build query strings directly into preallocated Strings using std::fmt::Write, avoiding intermediate Vec allocations and format!/join overhead while preserving existing behavior and parameters ordering.

Flow diagram for optimized D1 delete SQL construction

flowchart TD
    A[Start build_delete_stmt] --> B[Init params Vec::with_capacity]
    B --> C[Init sql String::with_capacity and write DELETE FROM table_name WHERE]
    C --> D[Loop key_fields_schema]
    D -->|key_part present| E[Append column = ? to sql with AND]
    E --> F[Push key_part_to_json to params]
    D -->|no key_part| D
    F --> D
    D --> G[Return sql and params]
Loading

File-Level Changes

Change Details Files
Optimize D1 upsert statement construction by streaming SQL into a preallocated String and collecting parameters in a single pass.
  • Replaced intermediate vectors for columns, placeholders, and update clauses with direct writes into a String using std::fmt::Write
  • Preallocated the SQL buffer with String::with_capacity(256) and the params vector with combined key/value schema length
  • Inlined comma handling for column lists, value placeholders, and update clauses using counters to track when to insert separators
  • Preserved existing parameter push order for keys and values while changing the string-building strategy
crates/flow/src/targets/d1.rs
Optimize D1 delete statement construction by building the WHERE clause directly into a preallocated String.
  • Replaced vector of WHERE clauses and final format!/join with direct writes into a String using std::fmt::Write
  • Preallocated the SQL buffer with String::with_capacity(128) and the params vector with key schema length
  • Used a boolean flag to manage AND separator insertion while iterating key fields
  • Preserved behavior and parameter ordering while avoiding intermediate allocations
crates/flow/src/targets/d1.rs

Tips and commands

Interacting with Sourcery

  • Trigger a new review: Comment @sourcery-ai review on the pull request.
  • Continue discussions: Reply directly to Sourcery's review comments.
  • Generate a GitHub issue from a review comment: Ask Sourcery to create an
    issue from a review comment by replying to it. You can also reply to a
    review comment with @sourcery-ai issue to create an issue from it.
  • Generate a pull request title: Write @sourcery-ai anywhere in the pull
    request title to generate a title at any time. You can also comment
    @sourcery-ai title on the pull request to (re-)generate the title at any time.
  • Generate a pull request summary: Write @sourcery-ai summary anywhere in
    the pull request body to generate a PR summary at any time exactly where you
    want it. You can also comment @sourcery-ai summary on the pull request to
    (re-)generate the summary at any time.
  • Generate reviewer's guide: Comment @sourcery-ai guide on the pull
    request to (re-)generate the reviewer's guide at any time.
  • Resolve all Sourcery comments: Comment @sourcery-ai resolve on the
    pull request to resolve all Sourcery comments. Useful if you've already
    addressed all the comments and don't want to see them anymore.
  • Dismiss all Sourcery reviews: Comment @sourcery-ai dismiss on the pull
    request to dismiss all existing Sourcery reviews. Especially useful if you
    want to start fresh with a new review - don't forget to comment
    @sourcery-ai review to trigger a new review!

Customizing Your Experience

Access your dashboard to:

  • Enable or disable review features such as the Sourcery-generated pull request
    summary, the reviewer's guide, and others.
  • Change the review language.
  • Add, remove or edit custom review instructions.
  • Adjust other review settings.

Getting Help

Copy link
Copy Markdown
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

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

Pull request overview

Refactors build_upsert_stmt and build_delete_stmt in the D1 target to construct SQL strings directly via String::with_capacity + std::fmt::Write instead of building intermediate Vec<String>s and using format! + join. Behavior is preserved: the same columns/placeholders/update clauses and parameter ordering are produced.

Changes:

  • Rewrote build_upsert_stmt to incrementally write columns, placeholders, and ON CONFLICT DO UPDATE SET clauses into a single pre-sized String, pre-sizing the params Vec.
  • Rewrote build_delete_stmt similarly, joining WHERE clauses with AND via inline writes.
  • No changes to public signatures or semantics.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Copy link
Copy Markdown
Contributor

@sourcery-ai sourcery-ai Bot left a comment

Choose a reason for hiding this comment

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

Hey - I've left some high level feedback:

  • In both build_upsert_stmt and build_delete_stmt, the write! results are silently discarded (let _ = ...); since these functions already return a Result, consider propagating or at least asserting on the fmt::Result to avoid masking future formatting errors.
  • In build_upsert_stmt, you now iterate over values.fields twice (once for columns/placeholders and once for update clauses); you could fold the update-clause construction into the first loop to avoid duplication and keep the column and update logic tightly coupled.
Prompt for AI Agents
Please address the comments from this code review:

## Overall Comments
- In both `build_upsert_stmt` and `build_delete_stmt`, the `write!` results are silently discarded (`let _ = ...`); since these functions already return a `Result`, consider propagating or at least asserting on the `fmt::Result` to avoid masking future formatting errors.
- In `build_upsert_stmt`, you now iterate over `values.fields` twice (once for columns/placeholders and once for update clauses); you could fold the update-clause construction into the first loop to avoid duplication and keep the column and update logic tightly coupled.

Sourcery is free for open source - if you like our reviews please consider sharing them ✨
Help me be more useful! Please click 👍 or 👎 on each comment and I'll use the feedback to improve your reviews.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants