capistrano-nomad 0.5.0 → 0.5.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 050bf4a39838b2c0a98c02c1d1c38fcee02818cdb3936d2b00073adea0223aec
-  data.tar.gz: 96b95dd209e984447ed33bc3229562202887222641954d494bc922fd215ec050
+  metadata.gz: 0b71f2fc4375d471be5284038182715e6a4673dfb047b24f94409d9249509706
+  data.tar.gz: 04d0c5906845f755b6c265b8f3ff943a3dcff1b2d4d4df66591f3e9aacb80063
 SHA512:
-  metadata.gz: 53535889c6707debe917ceefa837c6f4310c037523ded6e101ff752ff6962db3eb02805d2d60dd3fec1b693a3d0bcbae1dd5fe2f4c3a3451aea1170bcca0b2af
-  data.tar.gz: a43459682a67ba79448969660fbd659aca92ac90a0524e0ccf613212f537a148da93e06c3efb862dcec5c7b2441a3b46ad05f2a9ae0bdf8b5d761501ffba7839
+  metadata.gz: 3417e1276e5e377db5e52051c39f5955a3ba50cd83243e51d6f1a2919d2e4eda7a602ee55bab9a0c91f4e4d85fc4505c18886e8164608515bc681b663d641b60
+  data.tar.gz: 66fbf60d498d0b312256a6788292c9c5ad9e84384c34152e0d000c2045267bae57b96483cdd894f0f5a5f138ba3b89c179c2d265b8eebb4c29c3a496918ed19c

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    capistrano-nomad (0.5.0)
+    capistrano-nomad (0.5.1)
       activesupport (<= 7.0.8)
       byebug
       capistrano (~> 3.0)

data/capistrano-nomad.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |spec|
   spec.name = "capistrano-nomad"
-  spec.version = "0.5.0"
+  spec.version = "0.5.1"
   spec.authors = ["James Hu"]
 
   spec.summary = "Capistrano plugin for deploying and managing Nomad jobs"

data/lib/capistrano/nomad/helpers/dsl.rb

@@ -90,15 +90,18 @@ def nomad_job(name, attributes = {})
 
       desc "Restart #{description_name} job"
       task :restart do
-        on roles :manager do
+        on(roles(:manager)) do
           # We can't restart the job directly so we'll need to fetch all its running allocs and restart each of one
           # individually instead
-          running_alloc_ids_output = capture(
-            "nomad job allocs " \
-            "-t '{{range .}}{{if eq .ClientStatus \"running\"}}{{ println .ID}}{{end}}{{end}}' " \
-            "#{name}",
+          running_alloc_ids_output = capistrano_nomad_capture_nomad_command(
+            :job,
+            :allocs,
+            {
+              namespace: namespace,
+              t: "'{{range .}}{{if eq .ClientStatus \"running\"}}{{ println .ID}}{{end}}{{end}}'",
+            },
+            name,
           )
-
           running_alloc_ids = running_alloc_ids_output.strip.split("\n")
 
           running_alloc_ids.each do |alloc_id|

data/lib/capistrano/nomad/helpers/nomad.rb

@@ -78,7 +78,7 @@ def capistrano_nomad_build_release_var_file_path(*args)
   "#{release_path}#{capistrano_nomad_build_base_var_file_path(*args)}"
 end
 
-def capistrano_nomad_execute_nomad_command(*args)
+def capistrano_nomad_run_nomad_command(kind, *args)
   converted_args = args.each_with_object([]) do |arg, collection|
     # If hash then convert it as options
     if arg.is_a?(Hash)
@@ -97,22 +97,60 @@ def capistrano_nomad_execute_nomad_command(*args)
     end
   end
 
+  # Ignore errors
+  public_send(kind, :nomad, *converted_args, raise_on_non_zero_exit: false)
+end
+
+def capistrano_nomad_execute_nomad_command(*args)
   on(roles(:manager)) do |host|
     run_interactively(host) do
-      # Ignore errors
-      execute(:nomad, *converted_args, raise_on_non_zero_exit: false)
+      capistrano_nomad_run_nomad_command(:execute, *args)
     end
   end
 end
 
+def capistrano_nomad_capture_nomad_command(*args)
+  output = nil
+
+  on(roles(:manager)) do |host|
+    output = capistrano_nomad_run_nomad_command(:capture, *args)
+  end
+
+  output
+end
+
 def capistrano_nomad_exec_within_job(name, command, namespace: nil, task: nil)
-  capistrano_nomad_execute_nomad_command(
-    :alloc,
-    :exec,
-    { namespace: namespace, task: task || name, job: true },
-    name,
-    command,
-  )
+  on(roles(:manager)) do
+    task = name unless task
+
+    # Find alloc id that contains task
+    output = capistrano_nomad_capture_nomad_command(
+      :job,
+      :allocs,
+      { namespace: namespace, t: "'{{range .}}{{ .ID }},{{ .TaskGroup }}|{{end}}'" },
+      name,
+    )
+    alloc_id = output.split("|").map { |s| s.split(",") }.find { |_, t| t == task.to_s }&.first
+
+    if alloc_id
+      capistrano_nomad_execute_nomad_command(
+        :alloc,
+        :exec,
+        { namespace: namespace, task: task },
+        alloc_id,
+        command,
+      )
+    else
+      # If alloc can't be determined then choose at random
+      capistrano_nomad_execute_nomad_command(
+        :alloc,
+        :exec,
+        { namespace: namespace, job: true },
+        task,
+        command,
+      )
+    end
+  end
 end
 
 def capistrano_nomad_upload_file(local_path:, remote_path:, erb_vars: {})

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: capistrano-nomad
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.5.1
 platform: ruby
 authors:
 - James Hu
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2023-10-14 00:00:00.000000000 Z
+date: 2023-10-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport