Skip to content

impl(bigtable): add StubManager class#16022

Merged
scotthart merged 3 commits intogoogleapis:mainfrom
scotthart:bigtable_affinity_stub_manager
Mar 4, 2026
Merged

impl(bigtable): add StubManager class#16022
scotthart merged 3 commits intogoogleapis:mainfrom
scotthart:bigtable_affinity_stub_manager

Conversation

@scotthart
Copy link
Member

No description provided.

@scotthart scotthart requested a review from a team as a code owner March 3, 2026 19:55
@product-auto-label product-auto-label bot added the api: bigtable Issues related to the Bigtable API. label Mar 3, 2026
@gemini-code-assist
Copy link

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 introduces a foundational StubManager class designed to streamline the handling of BigtableStub objects within the Bigtable client library. This enhancement provides a flexible mechanism for managing stub lifecycle and selection, allowing for either a global stub or instance-specific stubs, which is crucial for optimizing resource utilization and request routing in various Bigtable interaction patterns.

Highlights

  • New StubManager Class: Introduced a new StubManager class to centralize the management of BigtableStub instances. This class supports two primary modes: kNoAffinity for a single shared stub and kInstanceAffinity for managing stubs specific to Bigtable instances.
  • Instance Affinity Support: The StubManager can maintain a collection of BigtableStub instances, each potentially bound to a specific Bigtable instance. It includes logic to retrieve an existing stub for a given instance or create a new one on demand using a provided creation function.
  • Build System Integration: The new StubManager source and header files, along with its unit tests, have been integrated into the CMake and Bazel build configurations for the google-cloud-cpp-bigtable library.
  • Comprehensive Unit Tests: Added a dedicated unit test file (stub_manager_test.cc) to thoroughly validate the functionality of the StubManager class, covering both its no-affinity and instance-affinity behaviors.
Changelog
  • google/cloud/bigtable/CMakeLists.txt
    • Added internal/stub_manager.cc and internal/stub_manager.h to the google_cloud_cpp_bigtable library sources.
    • Added internal/stub_manager_test.cc to the bigtable_client_unit_tests.
  • google/cloud/bigtable/bigtable_client_unit_tests.bzl
    • Included internal/stub_manager_test.cc in the list of unit tests for Bazel builds.
  • google/cloud/bigtable/google_cloud_cpp_bigtable.bzl
    • Added internal/stub_manager.h to the library headers for Bazel builds.
    • Added internal/stub_manager.cc to the library sources for Bazel builds.
  • google/cloud/bigtable/internal/stub_manager.cc
    • Implemented the StubManager class, including constructors for no-affinity and instance-affinity modes, and the GetStub method for retrieving stubs.
  • google/cloud/bigtable/internal/stub_manager.h
    • Declared the StubManager class, defining its public interface, internal state (mode, stub creation function, affinity stubs), and an enum for priming.
  • google/cloud/bigtable/internal/stub_manager_test.cc
    • Created unit tests for StubManager, verifying its behavior in both kNoAffinity and kInstanceAffinity configurations, including dynamic stub creation.
Activity
  • No human activity has been recorded on this pull request yet.
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 introduces a StubManager class to manage BigtableStub instances. This class supports two modes: kNoAffinity with a single shared stub, and kInstanceAffinity which maintains a map of stubs per instance, creating them on demand. The implementation is sound and includes corresponding unit tests. I have one suggestion to improve the implementation of GetStub for better performance and conciseness.

@codecov
Copy link

codecov bot commented Mar 3, 2026

Codecov Report

❌ Patch coverage is 98.71795% with 1 line in your changes missing coverage. Please review.
✅ Project coverage is 92.63%. Comparing base (ced8a7c) to head (ec1ada1).
⚠️ Report is 3 commits behind head on main.

Files with missing lines Patch % Lines
...oogle/cloud/bigtable/internal/stub_manager_test.cc 98.36% 1 Missing ⚠️
Additional details and impacted files
@@           Coverage Diff           @@
##             main   #16022   +/-   ##
=======================================
  Coverage   92.63%   92.63%           
=======================================
  Files        2333     2335    +2     
  Lines      214631   214709   +78     
=======================================
+ Hits       198827   198906   +79     
+ Misses      15804    15803    -1     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@scotthart scotthart merged commit 850fd1f into googleapis:main Mar 4, 2026
57 of 64 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

api: bigtable Issues related to the Bigtable API.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants