herdst_worker 0.2.12 → 0.2.14

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 24667ba11c332e5770ab05a850ac72e72d4e50a90026e8c62285de7bd7a60390
-  data.tar.gz: 90931c824d1567e345fb3b702b98d20b5d5fe1f4b5eafd51ef41e5d40cf3d4a4
+  metadata.gz: abe8b9c33c17e39eb8f3bc453f59867a813bad52d273ade952cb416ae9e6a8bb
+  data.tar.gz: 779373f610f0220af5b4abe3b0dc78d982d56ee9c8424bc6b9f1ae7c53fd76dd
 SHA512:
-  metadata.gz: db2a5e29b19ef8c9a0dc47ad2918f420681f2479b360bcd100588b571d89f384374fa86985ffc240575670a2af31fb47f75d98e55af7abe745209e76e93ab9fc
-  data.tar.gz: e927d63d9a6e48a8d423b01119d7f25e334753012146277decf7b059f1675d1b5f2f6d80defd6eb6ca0931829c68807fed130931d7b3304aa0a9155e9f1f216d
+  metadata.gz: ee12d5784b12a07a25f3b0f322faf547e0bd6158a01c1e33ba26419df97ace4e17bd4287eed4c68fbf8a28bc7e8dbd5bb47402f47fb9b52250e8ca9b58a37ea7
+  data.tar.gz: 85c9804583269bb39ca005273e2385dd4cf1204863bdd89100eb35699745d774389aca0ddbe570051c79a929857827f14a18498ff3f3fb1acb6165f6448d41e8

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,19 +107,23 @@ 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)
                     runtime = current_time - self.start_time
-                    self.app.logger.queue.info "Stopping after #{runtime} seconds of work"
+                    self.app.logger.queue.warn "Stopping after #{runtime} seconds of work"
                     set_status "stopping"
                 
                 # On finishing wait for jobs to complete and then set status
                 # to idle
                 elsif self.processor_status == "finishing"
                     if self.job_count == 0
-                        self.app.logger.queue.info "Setting processor status to idle"
+                        self.app.logger.queue.warn "Setting processor status to idle"
                         set_status "idle"
                     end
                 
@@ -127,14 +131,14 @@ module HerdstWorker
                 # to stopped. Once stopped the polling will terminate.
                 elsif self.processor_status == "stopping"
                     if self.job_count == 0
-                        self.app.logger.queue.info "Setting processor status to stopped"
+                        self.app.logger.queue.warn "Setting processor status to stopped"
                         set_status "stopped"
                     end
                 
                 end
                 
                 if self.processor_status == "stopped"
-                    self.app.logger.queue.info "Exiting program, Service requested to stop"
+                    self.app.logger.queue.warn "Exiting program, Service requested to stop"
                     throw :stop_polling
                 end
             end

data/lib/herdst_worker/version.rb

@@ -1,3 +1,3 @@
 module HerdstWorker
-  VERSION = '0.2.12'
+  VERSION = '0.2.14'
 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.14
 platform: ruby
 authors:
 - Herd.St
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2025-04-01 00:00:00.000000000 Z
+date: 2026-01-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport