active_record_proxy_adapters 0.8.0 → 0.8.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8e87a47a033b2402f2fdd0dbae2aec360ff04afbe4639f83615030da82d80cfc
-  data.tar.gz: 7caec5f8d663d2b67482f28f30b669e36a00fedb8d9eea6b06404ad3df9ce594
+  metadata.gz: a5c79bbbfc75972477b105dbee115b4107a12b3a326164c8ef3562c16910609a
+  data.tar.gz: d7276cdc0c42fb138fdf662c867ff3e3e2abcb2a2b58904d8080b630f91e6ac6
 SHA512:
-  metadata.gz: 65850028f544840668b1aa172a6442665cf9142c1a52934773f433c823fc56365b0251b6188636ccd252f9b33f74c9a40c9578d44953b57e33af842393e92e54
-  data.tar.gz: eb1c1b67ba524477b2b792d8a23f003f057cd764c4ee2930e9c5613efee9262ea7ba03c18084004b406b26e9c9e770a905733a2164ce6a84d107b172589860b0
+  metadata.gz: e0cb991cc6c4f4f49fcdb330e2d4254a13fd0e8b3a10f2ce9cf38f9dac36305c9c0ad148d2660839e354d2fd9b251ce0d92df5c05fd45414b0d1b298cd03df6c
+  data.tar.gz: d98aa2d8ddf561875041c9807604e1b13fbda7600bee9b2528cb7a34beb700d0540f9750f69a7d4fa10a0f3074bce71d1ad2f51b28701d018784c68a81edc78d

data/lib/active_record_proxy_adapters/railtie.rb

@@ -7,6 +7,7 @@ module ActiveRecordProxyAdapters
   class Railtie < Rails::Railtie
     require "active_record_proxy_adapters/connection_handling"
     require "active_record_proxy_adapters/middleware"
+    require "active_record_proxy_adapters/rake"
 
     config.to_prepare do
       Rails.autoloaders.each do |autoloader|
@@ -22,5 +23,10 @@ module ActiveRecordProxyAdapters
     initializer "active_record_proxy_adapters.configure_rails_initialization" do |app|
       app.middleware.use ActiveRecordProxyAdapters::Middleware
     end
+
+    rake_tasks do
+      ActiveRecordProxyAdapters::Rake.load_tasks
+      ActiveRecordProxyAdapters::Rake.enhance_db_tasks
+    end
   end
 end

data/lib/active_record_proxy_adapters/rake.rb

@@ -0,0 +1,47 @@
+# frozen_string_literal: true
+
+require "rake"
+
+module ActiveRecordProxyAdapters
+  # Enhances all rails db rake tasks to stick to writer connection
+  module Rake
+    module_function
+
+    def load_tasks
+      Dir[rake_tasks_path].each { |rake_path| load(rake_path) }
+    end
+
+    def rake_tasks_path
+      File.join(__dir__, "tasks/**/*.rake")
+    end
+
+    def enhance_db_tasks
+      ::Rake::Task
+        .tasks
+        .select(&enhanceable_db_task?)
+        .each { |task| task.enhance([push_to_stack_rake_task.name], &pop_from_stack_and_reenable) }
+    end
+
+    def push_to_stack_rake_task
+      ::Rake::Task["arpa:push_to_stack"]
+    end
+
+    def pop_from_stack_rake_task
+      ::Rake::Task["arpa:pop_from_stack"]
+    end
+
+    def pop_from_stack_and_reenable
+      proc do
+        pop_from_stack_rake_task.invoke
+        [push_to_stack_rake_task, pop_from_stack_rake_task].each(&:reenable)
+      end
+    end
+
+    def enhanceable_db_task?
+      proc do |task|
+        task_name = task.name
+        task_name.start_with?("db:") && task_name != "db:load_config"
+      end
+    end
+  end
+end

data/lib/active_record_proxy_adapters/tasks/arpa.rake

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+namespace :arpa do
+  desc "Pushes to connected_to stack before rake task is executed"
+  task :push_to_stack do
+    writing_role = ActiveRecordProxyAdapters::ActiveRecordContext.writing_role
+
+    ActiveRecord::Base.connected_to_stack << {
+      role: writing_role,
+      shard: nil,
+      prevent_writes: false,
+      klasses: [ActiveRecord::Base]
+    }
+
+    Thread.current.thread_variable_set(:arpa_rake_pushed_to_stack, true)
+  end
+
+  desc "Pops from connected_to stack after rake task is invoked"
+  task :pop_from_stack do
+    if Thread.current.thread_variable_get(:arpa_rake_pushed_to_stack)
+      ActiveRecord::Base.connected_to_stack.pop
+      Thread.current.thread_variable_set(:arpa_rake_pushed_to_stack, nil)
+    end
+  end
+end

data/lib/active_record_proxy_adapters/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ActiveRecordProxyAdapters
-  VERSION = "0.8.0"
+  VERSION = "0.8.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: active_record_proxy_adapters
 version: !ruby/object:Gem::Version
-  version: 0.8.0
+  version: 0.8.1
 platform: ruby
 authors:
 - Matt Cruz
@@ -149,8 +149,10 @@ files:
 - lib/active_record_proxy_adapters/postgresql_proxy.rb
 - lib/active_record_proxy_adapters/primary_replica_proxy.rb
 - lib/active_record_proxy_adapters/railtie.rb
+- lib/active_record_proxy_adapters/rake.rb
 - lib/active_record_proxy_adapters/sqlite3_proxy.rb
 - lib/active_record_proxy_adapters/synchronizable_configuration.rb
+- lib/active_record_proxy_adapters/tasks/arpa.rake
 - lib/active_record_proxy_adapters/transactionable_proxy_a_r_70.rb
 - lib/active_record_proxy_adapters/trilogy_proxy.rb
 - lib/active_record_proxy_adapters/version.rb