Skip to content

chore(tools/oracle,tools/postgres,tools/redis,tools/singlestore,tools/spanner): decouple source config from tools#2068

Merged
Yuan325 merged 2 commits into
config-branchfrom
config-oracle-etc
Dec 13, 2025
Merged

chore(tools/oracle,tools/postgres,tools/redis,tools/singlestore,tools/spanner): decouple source config from tools#2068
Yuan325 merged 2 commits into
config-branchfrom
config-oracle-etc

Conversation

@Yuan325

@Yuan325 Yuan325 commented Nov 27, 2025

Copy link
Copy Markdown
Contributor

Within Invoke() and RequiresClientAuthorization(), get Source from ResourceManager and use it to call functions directly.

@Yuan325 Yuan325 requested a review from a team November 27, 2025 17:28
@gemini-code-assist

Copy link
Copy Markdown
Contributor

Summary of Changes

Hello @Yuan325, 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 introduces a significant architectural refactoring across various database and client tools (Oracle, Postgres, Redis, SingleStore, Spanner). The core change involves decoupling the direct management of data source connections from the tools themselves. Instead of holding direct references to database pools or clients, tools now dynamically obtain these resources from a centralized ResourceManager when they need to perform an operation. This change promotes a more flexible and maintainable design by centralizing resource management and reducing tight coupling.

Highlights

  • Decoupled Source Configuration: The direct dependency of tools on specific database or client source configurations has been removed. Tools now dynamically retrieve their required source (e.g., OracleDB, PostgresPool, RedisClient, SingleStorePool, SpannerClient) from a ResourceManager during their Invoke method execution.
  • Improved Modularity and Testability: By fetching sources at runtime via an interface (tools.SourceProvider), the tools are no longer tightly coupled to concrete source implementations. This enhances modularity, simplifies testing, and allows for more flexible configuration of data sources.
  • Streamlined Tool Initialization: The Initialize methods for all affected tools (Oracle, Postgres, Redis, SingleStore, Spanner) have been simplified. They no longer perform direct source validation or store direct references to database connection pools/clients, deferring these operations to the Invoke method.
  • Updated RequiresClientAuthorization Signature: The RequiresClientAuthorization method across all modified tools now returns a (bool, error) tuple, aligning with a more robust error handling pattern.
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.

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

Copy link
Copy Markdown
Contributor

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 multiple tools to decouple them from their source configurations. The changes correctly move the source retrieval logic from the Initialize method to the Invoke method, which is a good improvement for lazy initialization and consistency. The signature of RequiresClientAuthorization has also been updated across all tools. I've found one area for improvement in spannerlistgraphs to make it fully consistent with the other refactored tools.

@averikitsch averikitsch assigned averikitsch and unassigned duwenxin99 Dec 1, 2025
Base automatically changed from config-server to config-branch December 9, 2025 01:36
Comment thread internal/tools/postgres/postgreslistlocks/postgreslistlocks.go
@Yuan325

Yuan325 commented Dec 11, 2025

Copy link
Copy Markdown
Contributor Author

/gemini review

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

Copy link
Copy Markdown
Contributor

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 is a great refactoring that decouples tool implementations from their source configurations. By moving the source retrieval and validation from Initialize to Invoke, the design becomes cleaner and more consistent. The changes have been applied across a wide range of tools for different databases.

I've found one inconsistency in the spanner-list-graphs tool where the Initialize function was not fully refactored to remove the source validation logic. My review comment provides details on how to address this to make it consistent with the other tools.

Overall, this is a solid improvement to the codebase's architecture.

Comment thread internal/tools/spanner/spannerlistgraphs/spannerlistgraphs.go
@Yuan325 Yuan325 requested a review from averikitsch December 11, 2025 18:29
@Yuan325 Yuan325 merged commit 46437aa into config-branch Dec 13, 2025
6 of 11 checks passed
@Yuan325 Yuan325 deleted the config-oracle-etc branch December 13, 2025 15:12
Yuan325 added a commit that referenced this pull request Dec 17, 2025
…/spanner): decouple source config from tools (#2068)

Within Invoke() and RequiresClientAuthorization(), get Source from
ResourceManager and use it to call functions directly.
Yuan325 added a commit that referenced this pull request Dec 18, 2025
…/spanner): decouple source config from tools (#2068)

Within Invoke() and RequiresClientAuthorization(), get Source from
ResourceManager and use it to call functions directly.
Yuan325 added a commit that referenced this pull request Dec 19, 2025
…/spanner): decouple source config from tools (#2068)

Within Invoke() and RequiresClientAuthorization(), get Source from
ResourceManager and use it to call functions directly.
Yuan325 added a commit that referenced this pull request Dec 19, 2025
…/spanner): decouple source config from tools (#2068)

Within Invoke() and RequiresClientAuthorization(), get Source from
ResourceManager and use it to call functions directly.
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.

3 participants