Implement Premium Enterprise Sharding

Author: etewiah

app/controllers/pwb/application_controller.rb

@@ -9,6 +9,21 @@ class ApplicationController < ActionController::Base
     # Themes allowed to be switched via ?theme= URL parameter
     ALLOWED_THEMES = %w[default brisbane bologna].freeze
 
+    around_action :switch_tenant_shard
+
+    def switch_tenant_shard
+      # Ensure website is loaded to determine shard
+      website = current_website
+      shard = website&.database_shard || :default
+      
+      # Use the shard for all PwbTenant models
+      # Note: We use writing role for both reading/writing in this simple example
+      # to avoid connection switching complexity during the demo
+      PwbTenant::ApplicationRecord.connected_to(shard: shard, role: :writing) do
+        yield
+      end
+    end
+
     def set_theme_path
       theme_name = current_website&.theme_name
       if params[:theme].present?

app/models/ahoy/event.rb

@@ -3,6 +3,7 @@
 # == Schema Information
 #
 # Table name: ahoy_events
+# Database name: primary
 #
 #  id         :bigint           not null, primary key
 #  name       :string           not null

app/models/ahoy/visit.rb

@@ -3,6 +3,7 @@
 # == Schema Information
 #
 # Table name: ahoy_visits
+# Database name: primary
 #
 #  id               :bigint           not null, primary key
 #  browser          :string

app/models/i18n/backend/active_record/translation.rb

@@ -1,6 +1,7 @@
 # == Schema Information
 #
 # Table name: translations
+# Database name: primary
 #
 #  id             :integer          not null, primary key
 #  interpolations :text

app/models/pwb/address.rb

@@ -1,6 +1,7 @@
 # == Schema Information
 #
 # Table name: pwb_addresses
+# Database name: primary
 #
 #  id             :integer          not null, primary key
 #  city           :string

app/models/pwb/agency.rb

@@ -9,6 +9,7 @@ module Pwb
 # == Schema Information
 #
 # Table name: pwb_agencies
+# Database name: primary
 #
 #  id                              :integer          not null, primary key
 #  analytics_id_type               :integer
@@ -50,7 +51,6 @@ module Pwb
 #
 #  index_pwb_agencies_on_website_id  (website_id)
 #
-  #
   class Agency < ApplicationRecord
     self.table_name = 'pwb_agencies'
 

app/models/pwb/auth_audit_log.rb

@@ -18,6 +18,7 @@ module Pwb
 # == Schema Information
 #
 # Table name: pwb_auth_audit_logs
+# Database name: primary
 #
 #  id             :bigint           not null, primary key
 #  email          :string
@@ -49,7 +50,6 @@ module Pwb
 #  fk_rails_...  (user_id => pwb_users.id)
 #  fk_rails_...  (website_id => pwb_websites.id)
 #
-  #
   class AuthAuditLog < ApplicationRecord
     self.table_name = 'pwb_auth_audit_logs'
 

app/models/pwb/authorization.rb

@@ -1,6 +1,7 @@
 # == Schema Information
 #
 # Table name: pwb_authorizations
+# Database name: primary
 #
 #  id         :bigint           not null, primary key
 #  provider   :string

app/models/pwb/contact.rb

@@ -9,6 +9,7 @@ module Pwb
 # == Schema Information
 #
 # Table name: pwb_contacts
+# Database name: primary
 #
 #  id                   :bigint           not null, primary key
 #  details              :json
@@ -52,7 +53,6 @@ module Pwb
 #
 #  fk_rails_...  (website_id => pwb_websites.id)
 #
-  #
   class Contact < ApplicationRecord
     self.table_name = 'pwb_contacts'
 

app/models/pwb/content.rb

@@ -3,6 +3,7 @@
 # == Schema Information
 #
 # Table name: pwb_contents
+# Database name: primary
 #
 #  id                      :integer          not null, primary key
 #  input_type              :string