Use token from redis in gcp project billing worker

1 files changed, 14 insertions, 1 deletions

diff --git a/app/workers/check_gcp_project_billing_worker.rb b/app/workers/check_gcp_project_billing_worker.rb
index 42aa6b39d86..784d17c4654 100644
--- a/app/workers/check_gcp_project_billing_worker.rb
+++ b/app/workers/check_gcp_project_billing_worker.rb
@@ -1,14 +1,23 @@
+require 'securerandom'
+
 class CheckGcpProjectBillingWorker
   include ApplicationWorker
   include ClusterQueue
 
   LEASE_TIMEOUT = 15.seconds.to_i
 
+  def self.generate_redis_token_key
+    SecureRandom.uuid
+  end
+
   def self.redis_shared_state_key_for(token)
     "gitlab:gcp:#{token.hash}:billing_enabled"
   end
 
-  def perform(token)
+  def perform(token_key)
+    return unless token_key
+
+    token = get_token(token_key)
     return unless token
     return unless try_obtain_lease_for(token)
 
@@ -20,6 +29,10 @@ class CheckGcpProjectBillingWorker
 
   private
 
+  def get_token(token_key)
+    Gitlab::Redis::SharedState.with { |redis| redis.get(token_key) }
+  end
+
   def try_obtain_lease_for(token)
     Gitlab::ExclusiveLease
       .new("check_gcp_project_billing_worker:#{token.hash}", timeout: LEASE_TIMEOUT)