Bug 1986964 - Add new localconfig variables for the new dedicated replica to be used by the ETL cron job and possibly others

Author: dklawren

Bugzilla.pm

@@ -459,6 +459,26 @@ sub dbh_main {
   return request_cache->{dbh_main} ||= Bugzilla::DB::connect_main();
 }
 
+sub switch_to_replica1_db {
+  my $class = shift;
+
+  if (!request_cache->{dbh_replica1}) {
+    if ($class->localconfig->db_replica1_name) {
+      request_cache->{dbh_replica1} = Bugzilla::DB::connect_replica1();
+    }
+    elsif ($class->get_param_with_override('shadowdb')) {
+      request_cache->{dbh_replica1} = Bugzilla::DB::connect_shadow();
+    }
+    else {
+      request_cache->{dbh_replica1} = $class->dbh_main;
+    }
+  }
+
+  request_cache->{dbh} = request_cache->{dbh_replica1};
+
+  return $class->dbh;
+}
+
 sub languages {
   return Bugzilla::Install::Util::supported_languages();
 }
@@ -1097,6 +1117,12 @@ Switch from using the main database to using the shadow database.
 
 Change the database object to refer to the main database.
 
+=item C<switch_to_replica1_db>
+
+Switch from using the main database to using the replica1 database. This
+will fall back to the shadow database, if one is configured, or the main
+database as a last resort.
+
 =item C<params>
 
 The current Parameters of Bugzilla, as a hashref.

Bugzilla/DB.pm

@@ -182,6 +182,20 @@ sub connect_main {
   return $main_dbh;
 }
 
+sub connect_replica1 {
+  state $replica1_dbh;
+  return $replica1_dbh if $replica1_dbh;
+
+  my $connect_params = dclone(Bugzilla->localconfig);
+  $connect_params->{db_host} = $connect_params->{db_replica1_host};
+  $connect_params->{db_name} = $connect_params->{db_replica1_name};
+  $connect_params->{db_user} = $connect_params->{db_replica1_user};
+  $connect_params->{db_pass} = $connect_params->{db_replica1_pass};
+  $connect_params->{db_port} = $connect_params->{db_replica1_port};
+
+  return $replica1_dbh = _connect($connect_params);
+}
+
 sub _connect {
   my ($params) = @_;
 

Bugzilla/Install/Localconfig.pm

@@ -105,6 +105,11 @@ use constant LOCALCONFIG_VARS => (
   {name => 'shadowdb_pass',       default => '',},
   {name => 'datadog_host',        default => '',},
   {name => 'datadog_port',        default => 8125,},
+  {name => 'db_replica1_host',    default => '',},
+  {name => 'db_replica1_name',    default => '',},
+  {name => 'db_replica1_user',    default => '',},
+  {name => 'db_replica1_pass',    default => '',},
+  {name => 'db_replica1_port',    default => 0,},
 );
 
 sub _read_localconfig_from_env {

extensions/BMO/bin/export_bmo_etl.pl

@@ -68,8 +68,8 @@
 # Check to make sure another instance is not currently running
 check_and_set_lock();
 
-# Use replica if available
-my $dbh = Bugzilla->switch_to_shadow_db();
+# Use dedicated replica1
+my $dbh = Bugzilla->switch_to_replica1_db;
 
 my $ua = LWP::UserAgent::Determined->new(
   agent                 => 'Bugzilla',