Skip to content

Conversation

@lidezhu
Copy link
Collaborator

@lidezhu lidezhu commented Dec 26, 2025

What problem does this PR solve?

Issue Number: close #xxx

What is changed and how it works?

Check List

Tests

  • Unit test
  • Integration test
  • Manual test (add detailed scripts or steps below)
  • No code

Questions

Will it cause performance regression or break compatibility?
Do you need to update user documentation, design documentation or monitoring documentation?

Release note

Please refer to [Release Notes Language Style Guide](https://pingcap.github.io/tidb-dev-guide/contribute-to-tidb/release-notes-style-guide.html) to write a quality release note.

If you don't think this PR needs a release note then fill it with `None`.

@ti-chi-bot ti-chi-bot bot added do-not-merge/needs-linked-issue release-note Denotes a PR that will be considered when it comes time to generate release notes. do-not-merge/work-in-progress Indicates that a PR should not merge because it is a work in progress. labels Dec 26, 2025
@ti-chi-bot
Copy link

ti-chi-bot bot commented Dec 26, 2025

[APPROVALNOTIFIER] This PR is NOT APPROVED

This pull-request has been approved by:
Once this PR has been reviewed and has the lgtm label, please assign lidezhu for approval. For more information see the Code Review Process.
Please ensure that each of them provides their approval before proceeding.

The full list of commands accepted by this bot can be found here.

Details Needs approval from an approver in each of these files:

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment

@ti-chi-bot ti-chi-bot bot added the size/XXL Denotes a PR that changes 1000+ lines, ignoring generated files. label Dec 26, 2025
@gemini-code-assist
Copy link

Summary of Changes

Hello @lidezhu, 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 refactors the processing of resolved timestamp (ResolvedTs) events within the log puller component. By introducing a dedicated, configurable worker pool, the system can now handle these events asynchronously and in parallel. This change aims to enhance the efficiency, scalability, and robustness of ResolvedTs event processing, reducing potential bottlenecks in the main event handling pipeline and improving overall system performance.

Highlights

  • Dedicated ResolvedTs Worker Pool: Introduced a new resolvedTsWorkerPool to handle resolved timestamp (ResolvedTs) events asynchronously and in parallel, improving performance and decoupling this processing from the main event stream.
  • Refactored ResolvedTs Handling Logic: The core logic for processing individual ResolvedTs events has been extracted into a new handleResolvedTs function within resolved_ts_processor.go, which is now invoked by the dedicated worker pool.
  • Event Handler Renaming and Adaptation: The regionEventHandler has been renamed to subscriptionEventHandler and adapted to dispatch ResolvedTs events to the new worker pool rather than processing them directly, streamlining the event pipeline.
  • Configurable Worker Count: The number of workers in the resolvedTsWorkerPool can now be configured via SubscriptionClientConfig.ResolvedTsWorkerCount, allowing for better resource management and scalability.

🧠 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.

You can also get AI-powered code generation, chat, as well as code reviews directly in the IDE at no cost with the Gemini Code Assist IDE Extension.

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

@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

This pull request refactors the handling of resolved timestamp (resolvedTs) events by introducing a dedicated worker pool to process them in parallel, aiming to improve performance. The core logic for handleResolvedTs is moved to a new file, and the event handling is updated to accommodate this new asynchronous processing flow. The changes are well-structured and include corresponding test updates.

My review has identified a critical bug that could lead to a panic, and a potential memory leak due to a logic change in cleaning up the event matcher. I've provided suggestions to fix these issues.

Comment on lines 330 to 344
if span == nil {
span = state.region.subscribedSpan
} else if span != state.region.subscribedSpan {
log.Panic("resolved ts event contains multiple spans",
zap.Uint64("workerID", s.workerID),
zap.Uint64("subscriptionID", uint64(subscriptionID)),
zap.Uint64("regionID", regionID))
}
if span == nil {
log.Warn("region request worker receives a resolved ts event for region without span",
zap.Uint64("workerID", s.workerID),
zap.Uint64("subscriptionID", uint64(subscriptionID)),
zap.Uint64("regionID", regionID))
continue
}

Choose a reason for hiding this comment

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

critical

The logic to check for consistent spans across regions in a resolved-ts event has a bug. The condition span != state.region.subscribedSpan will cause a panic if span is non-nil and state.region.subscribedSpan is nil. This is incorrect, as it should only panic if they are two different, non-nil spans. This logic is also overly complex. It can be simplified to be more readable and correct.

        currentSpan := state.region.subscribedSpan
		if currentSpan == nil {
			log.Warn("region request worker receives a resolved ts event for region without span",
				zap.Uint64("workerID", s.workerID),
				zap.Uint64("subscriptionID", uint64(subscriptionID)),
				zap.Uint64("regionID", regionID))
			continue
		}

		if span == nil {
			span = currentSpan
		} else if span != currentSpan {
			log.Panic("resolved ts event contains multiple spans",
				zap.Uint64("workerID", s.workerID),
				zap.Uint64("subscriptionID", uint64(subscriptionID)),
				zap.Uint64("regionID", regionID))
		}

func handleResolvedTs(span *subscribedSpan, state *regionFeedState, resolvedTs uint64) uint64 {
if state.isStale() || !state.isInitialized() {
return 0
}

Choose a reason for hiding this comment

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

high

The call to state.matcher.tryCleanUnmatchedValue() was present in the original handleResolvedTs function but is missing here. This function was likely responsible for cleaning up stale, unmatched prewrite entries in the matcher using the advancing resolved timestamp as a watermark. By moving this call to handleEventEntries, the matcher will only be cleaned when there are KV events. If a region is idle and only receives resolved timestamp updates, the matcher may not be cleaned, potentially leading to a memory leak of stale prewrite entries. It seems this call should be restored here to prevent this potential memory leak.

Suggested change
}
}
state.matcher.tryCleanUnmatchedValue()


func handleEventEntries(span *subscribedSpan, state *regionFeedState, entries *cdcpb.Event_Entries_) {
regionID, _, _ := state.getRegionMeta()
state.matcher.tryCleanUnmatchedValue()

Choose a reason for hiding this comment

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

high

This call was moved here from handleResolvedTs. However, this means the matcher is only cleaned when KV events are processed. If a region is idle and only receives resolved timestamp updates, this could lead to a memory leak of stale prewrite entries. This call should be moved back to handleResolvedTs and removed from here.

@ti-chi-bot
Copy link

ti-chi-bot bot commented Dec 26, 2025

[FORMAT CHECKER NOTIFICATION]

Notice: To remove the do-not-merge/needs-linked-issue label, please provide the linked issue number on one line in the PR body, for example: Issue Number: close #123 or Issue Number: ref #456.

📖 For more info, you can check the "Contribute Code" section in the development guide.

@ti-chi-bot
Copy link

ti-chi-bot bot commented Dec 26, 2025

@lidezhu: The following tests failed, say /retest to rerun all failed tests or /retest-required to rerun all mandatory failed tests:

Test name Commit Details Required Rerun command
pull-check e49fc75 link true /test pull-check
pull-error-log-review e49fc75 link true /test pull-error-log-review

Full PR test history. Your PR dashboard.

Details

Instructions for interacting with me using PR comments are available here. If you have questions or suggestions related to my behavior, please file an issue against the kubernetes-sigs/prow repository. I understand the commands that are listed here.

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

Labels

do-not-merge/needs-linked-issue do-not-merge/work-in-progress Indicates that a PR should not merge because it is a work in progress. release-note Denotes a PR that will be considered when it comes time to generate release notes. size/XXL Denotes a PR that changes 1000+ lines, ignoring generated files.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants