bolt 0.16.4 → 0.17.0

data/modules/aggregate/plans/nodes.pp

@@ -0,0 +1,35 @@
+plan aggregate::nodes(
+  Optional[String[0]] $task = undef,
+  Optional[String[0]] $command = undef,
+  Optional[String[0]] $script = undef,
+  TargetSpec $nodes,
+  Hash[String, Data] $params = {}
+) {
+
+  # Validation
+  $type_count = [$task, $command, $script].reduce(0) |$acc, $v| {
+    if ($v) {
+      $acc + 1
+    } else {
+      $acc
+    }
+  }
+
+  if ($type_count == 0) {
+    fail_plan("Must specify a command, script, or task to run", 'canary/invalid-params')
+  }
+
+  if ($type_count > 1) {
+    fail_plan("Must specify only one command, script, or task to run", 'canary/invalid-params')
+  }
+
+  $res = if ($task) {
+    run_task($task, $nodes, $params)
+  } elsif ($command) {
+    run_command($command, $nodes, $params)
+  } elsif ($script) {
+    run_script($script, $nodes, $params)
+  }
+
+  aggregate::nodes($res)
+}

data/modules/canary/lib/puppet/functions/canary/merge.rb

@@ -0,0 +1,11 @@
+# Merges two ResultSets into a new ResultSet
+Puppet::Functions.create_function(:'canary::merge') do
+  dispatch :merge_results do
+    param 'ResultSet', :merger
+    param 'ResultSet', :mergee
+  end
+
+  def merge_results(merger, mergee)
+    Bolt::ResultSet.new(merger.results + mergee.results)
+  end
+end

data/modules/canary/lib/puppet/functions/canary/random_split.rb

@@ -0,0 +1,20 @@
+# Splits an array into two groups, where the 1st group is a randomly selected
+# sample of the input (of the specified size) and the 2nd group is the remainder.
+#
+# This function takes 2 parameters:
+# * The array to split (Array)
+# * The number of items to sample from the array (Integer)
+#
+# Returns an array of [<sample>, <remainder>].
+Puppet::Functions.create_function(:'canary::random_split') do
+  dispatch :rand do
+    param 'Array', :arr
+    param 'Integer', :size
+  end
+
+  def rand(arr, size)
+    canaries = arr.sample(size)
+    rest = arr.reject { |r| canaries.include?(r) }
+    [canaries, rest]
+  end
+end

data/modules/canary/lib/puppet/functions/canary/skip.rb

@@ -0,0 +1,23 @@
+# Returns a ResultSet with canary/skipped-node errors for each Target provided.
+#
+# This function takes a single parameter:
+# * List of nodes (Array[Variant[Target,String]])
+#
+# Returns a ResultSet.
+Puppet::Functions.create_function(:'canary::skip') do
+  dispatch :skip_result do
+    param 'Array[Variant[Target,String]]', :nodes
+  end
+
+  def skip_result(nodes)
+    results = nodes.map do |node|
+      node = Bolt::Target.new(node) unless node.is_a? Bolt::Target
+      Bolt::Result.new(node, value: { '_error' => {
+                         'msg' => "Skipped #{node.name} because of a previous failure",
+                         'kind' => 'canary/skipped-node',
+                         'details' => {}
+                       } })
+    end
+    Bolt::ResultSet.new(results)
+  end
+end

data/modules/canary/plans/init.pp

@@ -0,0 +1,52 @@
+plan canary(
+  Optional[String[0]] $task = undef,
+  Optional[String[0]] $command = undef,
+  Optional[String[0]] $script = undef,
+  TargetSpec $nodes,
+  Hash[String, Data] $params = {},
+  Integer $canary_size = 1
+) {
+
+  # Validation
+  $type_count = [$task, $command, $script].reduce(0) |$acc, $v| {
+    if ($v) {
+      $acc + 1
+    } else {
+      $acc
+    }
+  }
+
+  if ($type_count == 0) {
+    fail_plan("Must specify a command, script, or task to run", 'canary/invalid-params')
+  }
+
+  if ($type_count > 1) {
+    fail_plan("Must specify only one command, script, or task to run", 'canary/invalid-params')
+  }
+
+  [$canaries, $rest] = canary::random_split(get_targets($nodes), $canary_size)
+  $catch_params = $params + { '_catch_errors' => true }
+
+  if ($task) {
+    $canr = run_task($task, $canaries, $catch_params)
+    if ($canr.ok) {
+      $restr = run_task($task, $rest, $catch_params)
+    }
+  } elsif ($command) {
+    $canr = run_command($command, $canaries, $catch_params)
+    if ($canr.ok) {
+      $restr = run_command($command, $rest, $catch_params)
+    }
+  } elsif ($script) {
+    $canr = run_script($script, $canaries, $catch_params)
+    if ($canr.ok) {
+      $restr = run_script($script, $rest, $catch_params)
+    }
+  }
+
+  unless ($canr.ok) {
+    $restr = canary::skip($rest)
+  }
+
+  canary::merge($canr, $restr)
+}

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: bolt
 version: !ruby/object:Gem::Version
-  version: 0.16.4
+  version: 0.17.0
 platform: ruby
 authors:
 - Puppet
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-02-26 00:00:00.000000000 Z
+date: 2018-03-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: addressable
@@ -254,20 +254,6 @@ dependencies:
     - - '='
       - !ruby/object:Gem::Version
         version: 0.8.8
-- !ruby/object:Gem::Dependency
-  name: ffi
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "<="
-      - !ruby/object:Gem::Version
-        version: 1.9.18
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "<="
-      - !ruby/object:Gem::Version
-        version: 1.9.18
 - !ruby/object:Gem::Dependency
   name: puppetlabs_spec_helper
   requirement: !ruby/object:Gem::Requirement
@@ -343,6 +329,8 @@ files:
 - bolt-modules/boltlib/lib/puppet/functions/run_plan.rb
 - bolt-modules/boltlib/lib/puppet/functions/run_script.rb
 - bolt-modules/boltlib/lib/puppet/functions/run_task.rb
+- bolt-modules/boltlib/lib/puppet/functions/set_var.rb
+- bolt-modules/boltlib/lib/puppet/functions/vars.rb
 - bolt-modules/boltlib/types/targetspec.pp
 - exe/bolt
 - exe/bolt-inventory-pdb
@@ -365,6 +353,8 @@ files:
 - lib/bolt/result_set.rb
 - lib/bolt/target.rb
 - lib/bolt/transport/base.rb
+- lib/bolt/transport/local.rb
+- lib/bolt/transport/local/shell.rb
 - lib/bolt/transport/orch.rb
 - lib/bolt/transport/ssh.rb
 - lib/bolt/transport/ssh/connection.rb
@@ -373,6 +363,14 @@ files:
 - lib/bolt/util.rb
 - lib/bolt/version.rb
 - lib/bolt_ext/puppetdb_inventory.rb
+- modules/aggregate/lib/puppet/functions/aggregate/count.rb
+- modules/aggregate/lib/puppet/functions/aggregate/nodes.rb
+- modules/aggregate/plans/count.pp
+- modules/aggregate/plans/nodes.pp
+- modules/canary/lib/puppet/functions/canary/merge.rb
+- modules/canary/lib/puppet/functions/canary/random_split.rb
+- modules/canary/lib/puppet/functions/canary/skip.rb
+- modules/canary/plans/init.pp
 - vendored/facter/lib/facter.rb
 - vendored/facter/lib/facter/Cfkey.rb
 - vendored/facter/lib/facter/application.rb