ecs_compose 0.1.0.pre13 → 0.1.0.pre14

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a2272a942dcbab9f8fcac63a3dbc542c68fd86ac
-  data.tar.gz: 3f001cc3617347d0f5481a30dc75562763ae0990
+  metadata.gz: 258ad2a5c72320cfcddc857c2ef3bb65b514191b
+  data.tar.gz: ddde11787a6c3b444715af2f5139c2e27a00e631
 SHA512:
-  metadata.gz: 7f4b853b2e3260adb594529a9bdd21c7999358beac5c0828ab1ecc09c70e8301bae844459ec9d24adf3764832895d55f95b0a9b0689eb45a07a6ad86e7f2406a
-  data.tar.gz: 68d28497bb0ee507e6749a22ccad039eb1e737c2d77e489ef798c710e516747676279d31b7e700bdfaa6678b24cbc8a0d2a9da12837b34fc855bc8fb8b40120d
+  metadata.gz: ddf8db17be5df23a2afa7cd5b968efbc8cdcada1e7321f3d345c528509e431ce06cca31fd140c0b2ff7cd3edacc5b4bf6b8e255a13d83b4dbc8902a100e27b7c
+  data.tar.gz: aa1d015f363ed9590c42b75decd8d479b25cd22eb8d5379bd62b5cdc2dfc9c43328e501445ba78f407d4e0daf2457613509c5deba635772bdfd290444afdb61e

data/exe/ecs-compose

@@ -12,6 +12,7 @@ Usage:
   ecs-compose [options] register [<task_def>...]
   ecs-compose [options] up [<service>...]
   ecs-compose [options] run [-e <name>=<value>]... [--entrypoint <entrypoint>] <task> [-- [<arg>...]]
+  ecs-compose [options] scale (all=<n> | <service>=<m>)...
   ecs-compose [options] json [<task_def>]
 
 Options:
@@ -34,6 +35,7 @@ Commands:
   register  Registers the specified ECS task definitions (defaults to all)
   up        Updates the specified ECS services (defaults to all)
   run       Runs the specified task
+  scale     Specify how many copies of a service's task should be run
   json      Generate JSON for a specific task definition
 DOCOPT
 
@@ -58,14 +60,14 @@ class App
 
   # Figure out which subcommand was chosen, and run it.
   def run
-    for command in %w{register up run json}
+    for command in %w{register up run scale json}
       if options.fetch(command)
         send("command_#{command}")
         return
       end
     end
     # We shouldn't ever get here.
-    raise "Unknown command: #{command}"
+    raise "Unknown command"
   end
 
   protected
@@ -111,6 +113,30 @@ class App
     EcsCompose::TaskDefinition.wait_for_tasks([arn])
   end
 
+  def command_scale
+    # Technically we could make this work without a manifest.
+    available = manifest.task_definitions.select {|td| td.type == :service }
+    all_names = available.map {|td| td.name }
+    settings = options.fetch("<service>=<m>")
+    scales = {}
+    for setting in settings
+      setting =~ /\A([-_A-Za-z0-9]+)=(\d+)\z/ or
+        fatal_err("Can't parse '#{setting}'")
+      name, count = $1, $2.to_i
+      if name == 'all'
+        all_names.each {|n| scales[n] = count }
+      else
+        scales[name] = count
+      end
+    end
+    service_names = scales.map do |name, count|
+      service = available.find {|td| td.name == name } or
+        raise "Unknown service: #{name}"
+      service.scale(count)
+    end
+    EcsCompose::TaskDefinition.wait_for_services(service_names)
+  end
+
   def command_json
     task_definition = options.fetch('<task_def>')[0]
     if task_definition.nil?

data/lib/ecs_compose/ecs.rb

@@ -53,6 +53,13 @@ module EcsCompose
           "--task-definition", task_definition)
     end
 
+    # Update the specified service.  Sample args: `"frontend"`, `3`.
+    def self.update_service_desired_count(service, desired_count)
+      run("update-service",
+          "--service", service,
+          "--desired-count", desired_count.to_s)
+    end
+
     # Run a one-off task.  Sample args: `"migrator:1"`.  The overrides may
     # be specified in the JSON format used by `aws ecs run-task`.
     def self.run_task(task_definition, overrides_json: nil)

data/lib/ecs_compose/task_definition.rb

@@ -33,6 +33,12 @@ module EcsCompose
       name
     end
 
+    # Set the number of running copies of a service we want to have.
+    def scale(count)
+      Ecs.update_service_desired_count(name, count)
+      name
+    end
+
     # Wait for a set of services to reach a steady state.
     def self.wait_for_services(service_names)
       Ecs.wait_services_stable(service_names)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ecs_compose
 version: !ruby/object:Gem::Version
-  version: 0.1.0.pre13
+  version: 0.1.0.pre14
 platform: ruby
 authors:
 - Eric Kidd
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2015-08-10 00:00:00.000000000 Z
+date: 2015-08-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: docopt