[PART 1] Create a "priority only" distribution

app/models/concerns/ama_case_distribution.rb

@@ -17,6 +17,22 @@ def docket_coordinator
     @docket_coordinator ||= DocketCoordinator.new
   end
 
+  def priority_push_distribution(limit = nil)
+    @appeals = []
+    @rem = 0
+
+    if limit.nil?
+      # Distribute priority appeals that are tied to judges (not genpop) with no limit.
+      distribute_appeals(:legacy, nil, priority: true, genpop: "not_genpop")
+      distribute_appeals(:hearing, nil, priority: true, genpop: "not_genpop")
+    else
+      # Distribute <limit> number of cases, regardless of docket type, oldest first.
+      oldest_priority_appeals_by_docket(limit).each do |docket, number_of_cases|
+        distribute_appeals(docket, number_of_cases, priority: true)
+      end
+    end
+  end
+
   def ama_distribution
     @appeals = []
     @rem = batch_size
@@ -59,7 +75,7 @@ def ama_distribution
 
   def ama_statistics
     {
-      batch_size: batch_size,
+      batch_size: priority_push? ? @appeals.count : batch_size,
       total_batch_size: total_batch_size,
       priority_count: priority_count,
       direct_review_due_count: direct_review_due_count,
@@ -75,7 +91,7 @@ def ama_statistics
   end
 
   def distribute_appeals(docket, num, priority: false, genpop: "any", range: nil, bust_backlog: false)
-    return [] unless num > 0
+    return [] unless num.nil? || num > 0
 
     if range.nil? && !bust_backlog
       appeals = dockets[docket].distribute_appeals(self, priority: priority, genpop: genpop, limit: num)

app/models/distribution.rb

@@ -20,13 +20,15 @@ class Distribution < CaseflowRecord
   CASES_PER_ATTORNEY = 3
   ALTERNATIVE_BATCH_SIZE = 15
 
+  scope :priority_push, -> { where(priority_push: true) }
+
   class << self
     def pending_for_judge(judge)
       where(status: %w[pending started], judge: judge)
     end
   end
 
-  def distribute!
+  def distribute!(limit = nil)
     return unless %w[pending error].include? status
 
     if status == "error"
@@ -41,7 +43,7 @@ def distribute!
     multi_transaction do
       ActiveRecord::Base.connection.execute "SET LOCAL statement_timeout = #{transaction_time_out}"
 
-      ama_distribution
+      priority_push? ? priority_push_distribution(limit) : ama_distribution
 
       update!(status: "completed", completed_at: Time.zone.now, statistics: ama_statistics)
     end

app/models/dockets/hearing_request_docket.rb

@@ -20,6 +20,14 @@ def distribute_appeals(distribution, priority: false, genpop: "any", limit: 1)
       base_relation: base_relation, genpop: genpop, judge: distribution.judge
     ).call
 
+    # genpop 'any' returns 2 arrays of the limited base relation. This means if we only request 2 cases, appeals is a
+    # 2x2 array containing 4 cases overall and we will end up distributing 4 cases rather than 2. Instead, reinstate the
+    # limit here by filtering out the newest cases
+    if genpop == "any"
+      appeals_to_reject = appeals.flatten.sort_by(&:ready_for_distribution_at).drop(limit)
+      appeals = [appeals.first - appeals_to_reject, appeals.last - appeals_to_reject]
+    end
+
     HearingRequestCaseDistributor.new(
       appeals: appeals, genpop: genpop, distribution: distribution, priority: priority
     ).call

app/models/vacols/case_docket.rb

@@ -118,7 +118,7 @@ class DocketNumberCentennialLoop < StandardError; end
         #{JOIN_ASSOCIATED_VLJS_BY_PRIOR_DECISIONS}
       )
       where ((VLJ = ? and 1 = ?) or (VLJ is null and 1 = ?))
-        and rownum <= ?
+        and (rownum <= ? or 1 = ?)
   "
 
   SELECT_NONPRIORITY_APPEALS = "
@@ -324,7 +324,8 @@ def self.distribute_priority_appeals(judge, genpop, limit, dry_run = false)
                                       judge.vacols_attorney_id,
                                       (genpop == "any" || genpop == "not_genpop") ? 1 : 0,
                                       (genpop == "any" || genpop == "only_genpop") ? 1 : 0,
-                                      limit
+                                      limit,
+                                      limit.nil? ? 1 : 0
                                     ])
 
     distribute_appeals(fmtd_query, judge, dry_run)