fix(unique-jobs): Fix on retry in unique jobs (#2862)

## Context

`retry_on` does not work with `until_executed` strategy.
veeqo/activejob-uniqueness#75

## Description

This PR updates all the jobs that are affected with this issue - a
combination of `retry_on` and `unique :until_executed`

Author: ivannovosad

app/jobs/integrations/aggregator/invoices/crm/create_customer_association_job.rb

@@ -7,8 +7,6 @@ module Crm
         class CreateCustomerAssociationJob < ApplicationJob
           queue_as 'integrations'
 
-          unique :until_executed, on_conflict: :log
-
           retry_on LagoHttpClient::HttpError, wait: :polynomially_longer, attempts: 10
           retry_on RequestLimitError, wait: :polynomially_longer, attempts: 100
 

app/jobs/integrations/aggregator/invoices/crm/create_job.rb

@@ -7,8 +7,6 @@ module Crm
         class CreateJob < ApplicationJob
           queue_as 'integrations'
 
-          unique :until_executed, on_conflict: :log
-
           retry_on LagoHttpClient::HttpError, wait: :polynomially_longer, attempts: 10
           retry_on Integrations::Aggregator::BasePayload::Failure, wait: :polynomially_longer, attempts: 10
           retry_on RequestLimitError, wait: :polynomially_longer, attempts: 100

app/jobs/integrations/aggregator/invoices/crm/update_job.rb

@@ -7,8 +7,6 @@ module Crm
         class UpdateJob < ApplicationJob
           queue_as 'integrations'
 
-          unique :until_executed, on_conflict: :log
-
           retry_on LagoHttpClient::HttpError, wait: :polynomially_longer, attempts: 10
           retry_on Integrations::Aggregator::BasePayload::Failure, wait: :polynomially_longer, attempts: 10
           retry_on RequestLimitError, wait: :polynomially_longer, attempts: 100

app/jobs/integrations/aggregator/subscriptions/crm/create_customer_association_job.rb

@@ -7,8 +7,6 @@ module Crm
         class CreateCustomerAssociationJob < ApplicationJob
           queue_as 'integrations'
 
-          unique :until_executed, on_conflict: :log
-
           retry_on LagoHttpClient::HttpError, wait: :polynomially_longer, attempts: 10
           retry_on RequestLimitError, wait: :polynomially_longer, attempts: 100
 

app/jobs/integrations/aggregator/subscriptions/crm/create_job.rb

@@ -7,8 +7,6 @@ module Crm
         class CreateJob < ApplicationJob
           queue_as 'integrations'
 
-          unique :until_executed, on_conflict: :log
-
           retry_on LagoHttpClient::HttpError, wait: :polynomially_longer, attempts: 10
           retry_on Integrations::Aggregator::BasePayload::Failure, wait: :polynomially_longer, attempts: 10
           retry_on RequestLimitError, wait: :polynomially_longer, attempts: 100

app/jobs/integrations/aggregator/subscriptions/crm/update_job.rb

@@ -7,8 +7,6 @@ module Crm
         class UpdateJob < ApplicationJob
           queue_as 'integrations'
 
-          unique :until_executed, on_conflict: :log
-
           retry_on LagoHttpClient::HttpError, wait: :polynomially_longer, attempts: 10
           retry_on Integrations::Aggregator::BasePayload::Failure, wait: :polynomially_longer, attempts: 10
           retry_on RequestLimitError, wait: :polynomially_longer, attempts: 100

config/application.rb

@@ -42,3 +42,5 @@ class Application < Rails::Application
     config.active_support.cache_format_version = 7.1
   end
 end
+
+require_relative "../lib/active_job/uniqueness/strategies/until_executed_patch"

lib/active_job/uniqueness/strategies/until_executed_patch.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require "active_job/uniqueness/strategies/until_executed"
+
+# https://github.com/veeqo/activejob-uniqueness/issues/75
+# retry_on does not work with until_executed strategy
+
+module ActiveJob
+  module Uniqueness
+    module Strategies
+      module UntilExecutedPatch
+        def before_enqueue
+          return if lock(resource: lock_key, ttl: lock_ttl)
+          # We're retrying the job, so we don't need to lock again
+          return if job.executions > 0
+
+          handle_conflict(resource: lock_key, on_conflict: on_conflict)
+          abort_job
+        end
+      end
+    end
+  end
+end
+
+ActiveJob::Uniqueness::Strategies::UntilExecuted.prepend(
+  ActiveJob::Uniqueness::Strategies::UntilExecutedPatch
+)