Support multi-cluster routing for Google CloudBigtable.

PiperOrigin-RevId: 331019716

perfkitbenchmarker/linux_benchmarks/cloud_bigtable_ycsb_benchmark.py

@@ -420,6 +420,8 @@ def Run(benchmark_spec):
         'bigtable_replication_zone'] = FLAGS.bigtable_replication_cluster_zone
     metadata['bigtable_storage_type'] = FLAGS.bigtable_storage_type
     metadata['bigtable_node_count'] = FLAGS.bigtable_node_count
+    metadata['bigtable_multicluster_routing'] = (
+        FLAGS.bigtable_multicluster_routing)
 
   # By default YCSB uses a BufferedMutator for Puts / Deletes.
   # This leads to incorrect update latencies, since since the call returns

perfkitbenchmarker/providers/gcp/gcp_bigtable.py

@@ -31,6 +31,13 @@ def _ValidateReplicationFlags(flag_dict):
   return (not flag_dict['bigtable_replication_cluster'] or
           flag_dict['bigtable_replication_cluster_zone'])
 
+
+def _ValidateRoutingFlags(flag_dict):
+  """Verifies correct usage of the bigtable routing flags."""
+  return (not flag_dict['bigtable_multicluster_routing'] or
+          flag_dict['bigtable_replication_cluster'])
+
+
 flags.DEFINE_integer('bigtable_node_count', 3,
                      'Number of nodes to create in the bigtable cluster.')
 flags.DEFINE_enum('bigtable_storage_type', 'ssd', ['ssd', 'hdd'],
@@ -41,10 +48,16 @@ def _ValidateReplicationFlags(flag_dict):
                      'Whether to create a Bigtable replication cluster.')
 flags.DEFINE_string('bigtable_replication_cluster_zone', None,
                     'Zone in which to create a Bigtable replication cluster.')
+flags.DEFINE_boolean('bigtable_multicluster_routing', False,
+                     'Whether to use multi-cluster routing.')
 flags.register_multi_flags_validator(
     ['bigtable_replication_cluster', 'bigtable_replication_cluster_zone'],
     _ValidateReplicationFlags, message='bigtable_replication_cluster_zone must '
     'be set if bigtable_replication_cluster is True.')
+flags.register_multi_flags_validator(
+    ['bigtable_replication_cluster', 'bigtable_multicluster_routing'],
+    _ValidateRoutingFlags, message='bigtable_replication_cluster must '
+    'be set if bigtable_multicluster_routing is True.')
 
 
 class GcpBigtableInstance(resource.BaseResource):
@@ -89,6 +102,15 @@ def _Create(self):
 
     logging.info('Creating instance %s.', self.name)
 
+    if FLAGS.bigtable_multicluster_routing:
+      cmd = util.GcloudCommand(self, 'beta', 'bigtable', 'app-profiles',
+                               'update', 'default')
+      cmd.flags['instance'] = self.name
+      cmd.flags['route-any'] = True
+      cmd.flags['force'] = True
+      cmd.flags['zone'] = []
+      cmd.Issue()
+
   def _Delete(self):
     """Deletes the instance."""
     cmd = util.GcloudCommand(self, 'beta', 'bigtable', 'instances', 'delete',