add/remove service constraints

Author: Flask

tasks/service_scale.json

@@ -1,4 +1,3 @@
-
 {
   "description": "Scale one replicated service",
   "input_method": "stdin",

tasks/service_update.json

@@ -9,6 +9,14 @@
       "image": {
           "description": "The new image to use for the service",
           "type": "String[1]"
+      },
+      "constraint_add": {
+        "description": "Add or update a service constraint (selector==value, selector!=value)",
+        "type": "Optional[Array]"
+      },
+      "constraint_rm": {
+        "description": "Remove a service constraint if exists.",
+        "type": "Optional[Array]"
       }
   }
 }

tasks/service_update.rb

@@ -5,9 +5,22 @@
 require 'open3'
 require 'puppet'
 
-def service_update(image, service)
+def service_update(image, service, constraint_add, constraint_rm)
   cmd_string = 'docker service update'
   cmd_string += " --image #{image}" unless image.nil?
+
+  if constraint_add.is_a? Array
+    constraint_add.each do |param|
+      cmd_string += " --constraint-add #{param}"
+    end
+  end
+
+  if constraint_rm.is_a? Array
+    constraint_rm.each do |param|
+      cmd_string += " --constraint-rm #{param}"
+    end
+  end
+
   cmd_string += " #{service}" unless service.nil?
 
   stdout, stderr, status = Open3.capture3(cmd_string)
@@ -18,9 +31,11 @@ def service_update(image, service)
 params = JSON.parse(STDIN.read)
 image = params['image']
 service = params['service']
+constraint_add = params['constraint_add']
+constraint_rm = params['constraint_rm']
 
 begin
-  result = service_update(image, service)
+  result = service_update(image, service, constraint_add, constraint_rm)
   puts result
   exit 0
 rescue Puppet::Error => e

tasks/swarm_init.json

@@ -1,4 +1,3 @@
-
 {
   "description": "Initializes a swarm",
   "input_method": "stdin",

tasks/swarm_join.json

@@ -1,4 +1,3 @@
-
 {
   "description": "Join a swarm",
   "input_method": "stdin",

tasks/swarm_leave.json

@@ -1,4 +1,3 @@
-
 {
   "description": "Leave a swarm",
   "input_method": "stdin",

tasks/swarm_token.json

@@ -1,4 +1,3 @@
-
 {
   "description": "Gets the swarm token from the master",
   "input_method": "stdin",