Add lease to CheckGcpProjectBillingWorker

1 files changed, 11 insertions, 0 deletions

diff --git a/app/workers/check_gcp_project_billing_worker.rb b/app/workers/check_gcp_project_billing_worker.rb
index 97638f65e8d..254b0959063 100644
--- a/app/workers/check_gcp_project_billing_worker.rb
+++ b/app/workers/check_gcp_project_billing_worker.rb
@@ -1,16 +1,27 @@
 class CheckGcpProjectBillingWorker
   include ApplicationWorker
 
+  LEASE_TIMEOUT = 1.minute.to_i
+
   def self.redis_shared_state_key_for(token)
     "gitlab:gcp:#{token}:billing_enabled"
   end
 
   def perform(token)
     return unless token
+    return unless try_obtain_lease_for(token)
 
     billing_enabled = CheckGcpProjectBillingService.new.execute(token)
     Gitlab::Redis::SharedState.with do |redis|
       redis.set(self.class.redis_shared_state_key_for(token), billing_enabled)
     end
   end
+
+  private
+
+  def try_obtain_lease_for(token)
+    Gitlab::ExclusiveLease
+      .new("check_gcp_project_billing_worker:#{token}", timeout: LEASE_TIMEOUT)
+      .try_obtain
+  end
 end