wrapbox 0.4.0 → 0.4.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 67a6e2d03ab3ecd91eb20638f7ea5faa3b42e6e9
-  data.tar.gz: d64a6d52580d830f3c48dad77885fb1d344baaa9
+  metadata.gz: ec530a0b68981cc7a90bcd451c4a54d371d275dd
+  data.tar.gz: 17e6212341e23126993ebde697b9ddccf4ecc397
 SHA512:
-  metadata.gz: dc8c3a0b68c1d636a501cf205fa41741c526b1c432d2e2e09875b181a4bf3e3b3d8e2ff754ef341f7fcf6d98426a1a187eafdcca8d6c03241ecc7f531625d982
-  data.tar.gz: 800d8fedf7f5d0aea0850b3d5158699d0107ccfda0eb2f00319e331af2e38d48ba0e18a274bca0c08f6f008cda1761d2cfb70e018bd40342c912ce70043cb9bd
+  metadata.gz: eccc15faf35450e79202e8b134b9fd304ca66bb3a8179e976b6e181f83cc69d2e4a613326ab2fd872eb268c9485dbc2eb7feb0bf2147985d4b913feaabda0523
+  data.tar.gz: 2f3623dbb0aa16b3b37b22e0e06e0eacf985f21be5c12b8d1bf217234c3c9e36773bf53cc53429f76b9ae4ff3a5401da89548369183750e1a1eedb8b7ab702af

data/lib/wrapbox/runner/docker.rb

@@ -101,6 +101,8 @@ module Wrapbox
         desc "run_cmd [shell command]", "Run shell on docker"
         method_option :config, aliases: "-f", required: true, banner: "YAML_FILE", desc: "yaml file path"
         method_option :config_name, aliases: "-n", required: true, default: "default"
+        method_option :cpu, type: :numeric
+        method_option :memory, type: :numeric
         method_option :environments, aliases: "-e"
         def run_cmd(*args)
           repo = Wrapbox::ConfigRepository.new.tap { |r| r.load_yaml(options[:config]) }
@@ -110,7 +112,12 @@ module Wrapbox
           environments = options[:environments].to_s.split(/,\s*/).map { |kv| kv.split("=") }.map do |k, v|
             {name: k, value: v}
           end
-          runner.run_cmd(args, environments: environments)
+          if options[:cpu] || options[:memory]
+            container_definition_overrides = {cpu: options[:cpu], memory: options[:memory]}.reject { |_, v| v.nil? }
+          else
+            container_definition_overrides = {}
+          end
+          runner.run_cmd(args, environments: environments, container_definition_overrides: container_definition_overrides)
         end
       end
     end

data/lib/wrapbox/runner/ecs.rb

@@ -263,6 +263,7 @@ module Wrapbox
           end
         end
 
+        @logger.debug("Container Definitions: #{container_definitions}")
         register_retry_count = 0
         begin
           client.register_task_definition({
@@ -361,6 +362,8 @@ module Wrapbox
         method_option :config, aliases: "-f", required: true, banner: "YAML_FILE", desc: "yaml file path"
         method_option :config_name, aliases: "-n", required: true, default: "default"
         method_option :cluster, aliases: "-c"
+        method_option :cpu, type: :numeric
+        method_option :memory, type: :numeric
         method_option :environments, aliases: "-e"
         method_option :task_role_arn
         method_option :timeout, type: :numeric
@@ -385,7 +388,12 @@ module Wrapbox
             execution_retry: options[:execution_retry],
             max_retry_interval: options[:max_retry_interval]
           }.reject { |_, v| v.nil? }
-          runner.run_cmd(args, environments: environments, **run_options)
+          if options[:cpu] || options[:memory]
+            container_definition_overrides = {cpu: options[:cpu], memory: options[:memory]}.reject { |_, v| v.nil? }
+          else
+            container_definition_overrides = {}
+          end
+          runner.run_cmd(args, environments: environments, container_definition_overrides: container_definition_overrides, **run_options)
         end
       end
     end

data/lib/wrapbox/version.rb

@@ -1,3 +1,3 @@
 module Wrapbox
-  VERSION = "0.4.0"
+  VERSION = "0.4.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: wrapbox
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.4.1
 platform: ruby
 authors:
 - joker1007