From d1575916122fcf0d90093cb5e5b5087167b14e8b Mon Sep 17 00:00:00 2001
From: Joshua Jansen <joshuajansen88@gmail.com>
Date: Fri, 15 Nov 2024 17:04:08 +0100
Subject: [PATCH 1/5] Switch to writing role if present

---
 lib/rollup/aggregator.rb | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/lib/rollup/aggregator.rb b/lib/rollup/aggregator.rb
index 99d6189..9e11007 100644
--- a/lib/rollup/aggregator.rb
+++ b/lib/rollup/aggregator.rb
@@ -19,7 +19,9 @@ def rollup(name, column: nil, interval: "day", dimension_names: nil, time_zone:
       records = prepare_result(result, name, dimension_names, interval)
 
       maybe_clear(clear, name, interval) do
-        save_records(records) if records.any?
+        with_writing_role do
+          save_records(records) if records.any?
+        end
       end
     end
 
@@ -184,6 +186,14 @@ def maybe_clear(clear, name, interval)
       end
     end
 
+    def with_writing_role(&block)
+      if Rollup.writing_role.present?
+        ActiveRecord::Base.connected_to(role: Rollup.writing_role, &block)
+      else
+        block.call
+      end
+    end
+
     def save_records(records)
       # order must match unique index
       # consider using index name instead

From 7e8a02a7f3ee0eef521c6d987e0126c3bb4a97c8 Mon Sep 17 00:00:00 2001
From: Joshua Jansen <joshuajansen88@gmail.com>
Date: Fri, 15 Nov 2024 17:05:37 +0100
Subject: [PATCH 2/5] Add note to readyme

---
 README.md | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/README.md b/README.md
index 019bcf6..f51b6cc 100644
--- a/README.md
+++ b/README.md
@@ -121,6 +121,16 @@ Time zone objects also work. To see a list of available time zones in Rails, run
 
 See [date storage](#date-storage) for how dates are stored.
 
+### Multiple Databases
+
+Specify a `writing_role` when you want to explicitly switch the database role when writing the rollup to the database.
+
+```ruby
+Rollup.writing_role = :writing
+```
+
+See [ActiveRecord::Base.connected_to](https://guides.rubyonrails.org/active_record_multiple_databases.html#connecting-to-the-database) for more information.
+
 ### Calculations
 
 Rollups use `count` by default. For other calculations, use:

From a298212bfec0d8e3fbfffd8416bd1752ef6d03a1 Mon Sep 17 00:00:00 2001
From: Joshua Jansen <joshuajansen88@gmail.com>
Date: Fri, 15 Nov 2024 17:05:48 +0100
Subject: [PATCH 3/5] Add class accessor to Rollup class

---
 lib/rollup/model.rb     |  2 ++
 test/aggregator_test.rb | 33 +++++++++++++++++++++++++++++++++
 2 files changed, 35 insertions(+)

diff --git a/lib/rollup/model.rb b/lib/rollup/model.rb
index 889fdee..79e70cf 100644
--- a/lib/rollup/model.rb
+++ b/lib/rollup/model.rb
@@ -1,4 +1,6 @@
 class Rollup
+  cattr_accessor :writing_role
+
   module Model
     attr_accessor :rollup_column
 
diff --git a/test/aggregator_test.rb b/test/aggregator_test.rb
index 3e92c54..de963f0 100644
--- a/test/aggregator_test.rb
+++ b/test/aggregator_test.rb
@@ -133,4 +133,37 @@ def test_range_current
     end
     assert_equal "Cannot use range and current together", error.message
   end
+
+  def test_writing_role_switching
+    connected_to_args = nil
+    ActiveRecord::Base.singleton_class.define_method(:connected_to) do |role:, &block|
+      connected_to_args = role
+      block.call
+    end
+
+    create_users
+    Rollup.writing_role = :writing
+    User.rollup("Test")
+
+    assert_equal :writing, connected_to_args, "Expected connected_to to be called with writing role"
+  ensure
+    ActiveRecord::Base.singleton_class.remove_method(:connected_to)
+    Rollup.writing_role = nil
+  end
+
+  def test_no_writing_role_switching
+    connected_to_called = false
+    ActiveRecord::Base.singleton_class.define_method(:connected_to) do |role:, &block|
+      connected_to_called = true
+      block.call
+    end
+
+    create_users
+    Rollup.writing_role = nil
+    User.rollup("Test")
+
+    refute connected_to_called, "Expected connected_to not to be called when writing_role is nil"
+  ensure
+    ActiveRecord::Base.singleton_class.remove_method(:connected_to)
+  end
 end

From 44d3711a3a5a251587773500252c9677a2415d84 Mon Sep 17 00:00:00 2001
From: Joshua Jansen <joshuajansen88@gmail.com>
Date: Fri, 15 Nov 2024 17:08:20 +0100
Subject: [PATCH 4/5] Move attr_write to lib

---
 lib/rollup.rb       | 1 +
 lib/rollup/model.rb | 2 --
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/lib/rollup.rb b/lib/rollup.rb
index 91fa723..b6668df 100644
--- a/lib/rollup.rb
+++ b/lib/rollup.rb
@@ -6,6 +6,7 @@ class Rollup < ActiveRecord::Base
   class << self
     attr_accessor :week_start
     attr_writer :time_zone
+    attr_writer :writing_role
   end
   self.week_start = :sunday
 
diff --git a/lib/rollup/model.rb b/lib/rollup/model.rb
index 79e70cf..889fdee 100644
--- a/lib/rollup/model.rb
+++ b/lib/rollup/model.rb
@@ -1,6 +1,4 @@
 class Rollup
-  cattr_accessor :writing_role
-
   module Model
     attr_accessor :rollup_column
 

From 4ec0778c3671e60d64bb004e157dd89df7f1173f Mon Sep 17 00:00:00 2001
From: Joshua Jansen <joshuajansen88@gmail.com>
Date: Fri, 15 Nov 2024 17:13:37 +0100
Subject: [PATCH 5/5] Make writing_role accessor

---
 lib/rollup.rb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/rollup.rb b/lib/rollup.rb
index b6668df..af30d49 100644
--- a/lib/rollup.rb
+++ b/lib/rollup.rb
@@ -5,8 +5,8 @@ class Rollup < ActiveRecord::Base
 
   class << self
     attr_accessor :week_start
+    attr_accessor :writing_role
     attr_writer :time_zone
-    attr_writer :writing_role
   end
   self.week_start = :sunday