adding partitioned table for reporting_facility_monthly_follow_ups_a…

app/models/reports/facility_monthly_follow_up_and_registration.rb

@@ -8,6 +8,10 @@ def self.materialized?
       true
     end
 
+    def self.partitioned?
+      true
+    end
+
     NON_COUNT_FIELDS = %i[
       block_region_id
       district_region_id

app/models/reports/patient_state.rb

@@ -17,12 +17,6 @@ def self.partitioned?
       true
     end
 
-    def self.partitioned_refresh(refresh_month)
-      ActiveRecord::Base.connection.exec_query(
-        "CALL simple_reporting.add_shard_to_table('#{refresh_month}', 'reporting_patient_states')"
-      )
-    end
-
     def self.by_assigned_region(region_or_source)
       region = region_or_source.region
       where("assigned_#{region.region_type}_region_id" => region.id)

app/models/reports/view.rb

@@ -53,5 +53,11 @@ def self.fetch_refresh_month_for_date(date_of_refresh)
       month_offset = (day_of_month / 2) + 1
       day_of_month.odd? ? month_of_refresh.prev_month : (month_of_refresh - month_offset.month)
     end
+
+    def self.partitioned_refresh(refresh_month)
+      ActiveRecord::Base.connection.exec_query(
+        "CALL simple_reporting.add_shard_to_table('#{refresh_month}', '#{table_name}')"
+      )
+    end
   end
 end

lib/tasks/reporting.rake

@@ -2,19 +2,16 @@
 
 namespace :reporting do
   desc "Do a full refresh of the partitioned reporting table"
-  task full_partitioned_refresh: :environment do
-    reporting_tables = %w[
-      Reports::PatientState
-    ].freeze
+  task :full_partitioned_refresh, [:table_name] => :environment do |t, args|
+    unless args[:table_name].present?
+      puts "ERROR: table_name is required."
+      puts "Usage: rake reporting:full_partitioned_refresh[reporting_patient_states]"
+      exit 1
+    end
 
-    reporting_tables.each do |reporting_table|
-      klass = reporting_table.constantize
-      ActiveRecord::Base.connection.exec_query(
-        "TRUNCATE TABLE simple_reporting.#{klass.table_name}"
-      )
-      Reports::Month.order(:month_date).select(:month_date).each_with_index do |reporting_month, index|
-        RefreshReportingPartitionedTableJob.set(wait_until: Time.now + (index * 45).minutes).perform_async(reporting_month.month_date.to_s, klass.table_name)
-      end
+    reporting_table = args[:table_name]
+    Reports::Month.order(:month_date).select(:month_date).each_with_index do |reporting_month, index|
+      RefreshReportingPartitionedTableJob.set(wait_until: Time.now + (index * 45).minutes).perform_async(reporting_month.month_date.to_s, reporting_table)
     end
   end
 

spec/models/reports/facility_monthly_follow_up_and_registration_spec.rb

@@ -225,4 +225,10 @@
     expect(total.monthly_follow_ups_htn_only + total.monthly_follow_ups_htn_and_dm).to eq(dashboard_htn_follow_ups)
     expect(total.monthly_follow_ups_dm_only + total.monthly_follow_ups_htn_and_dm).to eq(dashboard_dm_follow_ups)
   end
+
+  describe "#partitioned?" do
+    it "returns true" do
+      expect(described_class.partitioned?).to be(true)
+    end
+  end
 end

spec/models/reports/patient_state_spec.rb

@@ -863,18 +863,6 @@ def patient_states(patient, from: nil, to: nil)
     end
   end
 
-  describe "#partitioned_refresh" do
-    let(:refresh_month) { Date.new(2024, 6, 1) }
-
-    it "executes the correct SQL to refresh the partition" do
-      expect(ActiveRecord::Base.connection).to receive(:exec_query).with(
-        "CALL simple_reporting.add_shard_to_table('#{refresh_month}', 'reporting_patient_states')"
-      )
-
-      described_class.partitioned_refresh(refresh_month)
-    end
-  end
-
   context "screening" do
     it "doesn't consider the patient if it's only screened" do
       diagnosed_patient = create(:patient, recorded_at: Date.new(2024, 5, 1), diagnosed_confirmed_at: Date.new(2024, 6, 1))

spec/models/reports/view_spec.rb

@@ -80,4 +80,18 @@ def table_descriptions_sql(table_name)
       end
     end
   end
+
+  describe "#partitioned_refresh" do
+    let(:refresh_month) { Date.new(2024, 6, 1) }
+
+    it "executes the correct SQL to refresh the partition" do
+      [Reports::PatientState, Reports::FacilityMonthlyFollowUpAndRegistration].each do |klass|
+        expect(ActiveRecord::Base.connection).to receive(:exec_query).with(
+          "CALL simple_reporting.add_shard_to_table('#{refresh_month}', '#{klass.table_name}')"
+        )
+
+        klass.partitioned_refresh(refresh_month)
+      end
+    end
+  end
 end