Use ActiveRecord configuration to read from replica for reporting

app/jobs/reports/base_report.rb

@@ -30,7 +30,7 @@ def report_timeout
     end
 
     def transaction_with_timeout
-      Db::EstablishConnection::ReadReplicaConnection.new.call do
+      ApplicationRecord.connected_to(role: :reading, shard: :read_replica) do
         ActiveRecord::Base.transaction do
           quoted_timeout = ActiveRecord::Base.connection.quote(report_timeout)
           ActiveRecord::Base.connection.execute("SET LOCAL statement_timeout = #{quoted_timeout}")

app/models/application_record.rb

@@ -1,3 +1,8 @@
 class ApplicationRecord < ActiveRecord::Base
   self.abstract_class = true
+
+  connects_to shards: {
+    default: { writing: :primary, reading: :primary },
+    read_replica: { reading: :primary_replica },
+  }
 end

config/application.rb

@@ -23,8 +23,23 @@ class Application < Rails::Application
     )
     IdentityConfig.build_store(configuration)
 
+    console do
+      if ENV['ALLOW_CONSOLE_DB_WRITE_ACCESS'] != 'true' &&
+         IdentityConfig.store.database_readonly_username.present? &&
+         IdentityConfig.store.database_readonly_password.present?
+        warn <<-EOS.squish
+          WARNING: Loading database a configuration with the readonly database user.
+          If you wish to make changes to records in the database set
+          ALLOW_CONSOLE_DB_WRITE_ACCESS to "true" in the environment
+        EOS
+
+        ActiveRecord::Base.establish_connection :primary_replica
+      end
+    end
+
     config.load_defaults '6.1'
     config.active_record.belongs_to_required_by_default = false
+    config.active_record.legacy_connection_handling = false
     config.assets.unknown_asset_fallback = true
 
     if IdentityConfig.store.ruby_workers_enabled

config/database.yml

@@ -1,5 +1,3 @@
-<% require 'production_database_configuration' %>
-
 postgresql: &postgresql
   adapter: postgresql
   encoding: utf8
@@ -24,22 +22,44 @@ defaults: &defaults
     statement_timeout: <%= IdentityConfig.store.database_statement_timeout %>
 
 development:
-  <<: *defaults
+  primary:
+    <<: *defaults
+  primary_replica:
+    <<: *defaults
+    replica: true
 
-test:
+_test: &test
   <<: *defaults
   pool: 10
   checkout_timeout: 10
   database: <%= ENV['POSTGRES_DB'] || "upaya_test#{ENV['TEST_ENV_NUMBER']}" %>
   user: <%= ENV['POSTGRES_USER'] %>
   password: <%= ENV['POSTGRES_PASSWORD'] %>
 
+test:
+  primary:
+    <<: *test
+  primary_replica:
+    <<: *test
+    replica: true
+
 production:
-  <<: *defaults
-  database: <%= IdentityConfig.store.database_name %>
-  username: <%= ProductionDatabaseConfiguration.username %>
-  host: <%= ProductionDatabaseConfiguration.host %>
-  password: <%= ProductionDatabaseConfiguration.password %>
-  pool: <%= ProductionDatabaseConfiguration.pool %>
-  sslmode: 'verify-full'
-  sslrootcert: '/usr/local/share/aws/rds-combined-ca-bundle.pem'
+  primary:
+    <<: *defaults
+    database: <%= IdentityConfig.store.database_name %>
+    username: <%= IdentityConfig.store.database_username %>
+    host: <%= IdentityConfig.store.database_host %>
+    password: <%= IdentityConfig.store.database_password %>
+    pool: <%= IdentityConfig.store.database_pool_idp %>
+    sslmode: 'verify-full'
+    sslrootcert: '/usr/local/share/aws/rds-combined-ca-bundle.pem'
+  primary_replica:
+    <<: *defaults
+    database: <%= IdentityConfig.store.database_name %>
+    username: <%= IdentityConfig.store.database_readonly_username %>
+    host: <%= IdentityConfig.store.database_read_replica_host %>
+    password: <%= IdentityConfig.store.database_readonly_password %>
+    pool: <%= IdentityConfig.store.database_pool_idp %>
+    sslmode: 'verify-full'
+    sslrootcert: '/usr/local/share/aws/rds-combined-ca-bundle.pem'
+    replica: true