herdst_worker 0.2.12 → 0.2.13

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 24667ba11c332e5770ab05a850ac72e72d4e50a90026e8c62285de7bd7a60390
-  data.tar.gz: 90931c824d1567e345fb3b702b98d20b5d5fe1f4b5eafd51ef41e5d40cf3d4a4
+  metadata.gz: 4eb4cf45cb45c997518f02f5830eb545f1bfab3d143c74b021adcc1380d3bfd7
+  data.tar.gz: cb9e765e9ecdd7435e67fd55fa65f4a54b8802674488f077a4f10c4532e2b528
 SHA512:
-  metadata.gz: db2a5e29b19ef8c9a0dc47ad2918f420681f2479b360bcd100588b571d89f384374fa86985ffc240575670a2af31fb47f75d98e55af7abe745209e76e93ab9fc
-  data.tar.gz: e927d63d9a6e48a8d423b01119d7f25e334753012146277decf7b059f1675d1b5f2f6d80defd6eb6ca0931829c68807fed130931d7b3304aa0a9155e9f1f216d
+  metadata.gz: '080daf4a1cdae1b0e853317a15c016d0a4ab0e2289f1443ad2a71a259a0a46001c32a7c424bd56dbb9ddbdfca11521157c8080e6a42e190d69191b3ee32669a3'
+  data.tar.gz: 2cc800ca64e958848e798c19e8323b028fedfa41ca2890462b29d6ed65f4ae59cd3453659a1ec514b8e6b000a75f0fabd713909af587793a35c57f95821aaf61

data/lib/herdst_worker/configuration/metadata.rb

@@ -1,3 +1,4 @@
+require "concurrent-ruby"
 require "aws-sdk-core"
 require "aws-sdk-secretsmanager"
 require "aws-sdk-ecs"
@@ -5,6 +6,24 @@ require "net/http"
 
 
 module HerdstWorker
+    module ExpiringCredentials
+
+        attr_accessor :expiry_timestamp
+
+        def set_expiry(expiry)
+            self.expiry_timestamp = expiry
+        end
+
+        def has_expired?
+            if self.expiry_timestamp.nil?
+                false
+            else
+                Time.now.utc.to_i > self.expiry_timestamp
+            end
+        end
+
+    end
+
     module Configuration
         class Metadata
     
@@ -22,6 +41,8 @@ module HerdstWorker
                 self.secrets["ENV"] = env
                 self.secret_json_keys = ["STRIPE_PLANS"]
 
+                Aws::Credentials.include HerdstWorker::ExpiringCredentials
+
                 begin
                     self.reload!
                 rescue Exception => ex
@@ -104,17 +125,32 @@ module HerdstWorker
                      make_request!(get_credentials_uri) :
                      self.config.config_for(:aws)
                 
+                expiry_time = (Time.parse(credentials["Expiration"]).utc - 5.minutes) rescue (Time.now.utc + 1.hour)
+
+                # Set credential info
                 self.aws_credentials = Aws::Credentials.new(
                     credentials["AccessKeyId"], 
                     credentials["SecretAccessKey"], 
                     credentials["Token"]
                 )
-                
+                self.aws_credentials.set_expiry(expiry_time.to_i)
+
+                # Set global config
                 Aws.config.update(
                     region: self.get_aws_region(credentials),
                     credentials: self.aws_credentials
                 )
-                
+
+                if self.is_prod?
+                    # Refresh the credentials just before the expiry time
+                    expiry_in_seconds = (expiry_time - Time.now.utc).to_i
+                    expiry_in_seconds = 60 if expiry_in_seconds < 60
+
+                    Concurrent::ScheduledTask.execute(expiry_in_seconds) do
+                        self.get_aws_credentials
+                    end
+                end
+
                 self.aws_credentials
             end
             

data/lib/herdst_worker/queue/processor.rb

@@ -107,7 +107,11 @@ module HerdstWorker
                     self.app.logger.queue_stats.info "STATS (#{self.processor_status}): #{stats.inspect}"
                 end
                 
-                # After 1 hour of running terminate application. 
+                if self.poller.client.config.credentials.has_expired?
+                    self.poller.client.config.credentials = self.app.config.metadata.get_aws_credentials
+                end
+                
+                # After hours of running terminate application. 
                 # The app will automatically restart in production
                 current_time = Time.now.utc.to_i
                 if (self.processor_status == "working") && (current_time >= self.restart_time)

data/lib/herdst_worker/version.rb

@@ -1,3 +1,3 @@
 module HerdstWorker
-  VERSION = '0.2.12'
+  VERSION = '0.2.13'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: herdst_worker
 version: !ruby/object:Gem::Version
-  version: 0.2.12
+  version: 0.2.13
 platform: ruby
 authors:
 - Herd.St
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2025-04-01 00:00:00.000000000 Z
+date: 2025-11-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport