hako 0.14.1 → 0.15.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b73b66a504c37b90e3234870e6333263e7abfc8c
-  data.tar.gz: 1de7e857c4c8f40c275f72bd1822e211521d447b
+  metadata.gz: 824d56a1bf34b619cd7251d92817f5c080a71b63
+  data.tar.gz: 80aa695a436bd7af6f6d43d2ac5196d791fb8efc
 SHA512:
-  metadata.gz: 805980ec90386912e4e69423820f6a224203c52cf06210f67c677c41ecfe181db3285597a6749703e8d73fdff18235352d868d416f182e8e01ca2b85db5d1888
-  data.tar.gz: 01adc0732edc57ebab0304b4f2e95aa4f753038a6ac162def04688f7f8fb57c905948450ac50847b88f12abc190a19452e3b8e68a13a7603610fbb05320bd4fa
+  metadata.gz: 7942006910edb928222008730977d4842d9afe03c76699833e37acd2d5a60f43e6a04a0b2228c7aeeabdb123f7d98a891ab81400862ba2be59a0d4e140d206a7
+  data.tar.gz: 83f44968892aaaf5f8699b6962c842d520ae4e22727ee2d41d5be7e1cd152843fd0412aaffc86c18cf9256643073b31af20dc2612d9f50a81fdf1c5d2b779ab7

data/README.md

@@ -47,6 +47,10 @@ Events:
   2015-10-05 13:35:53 +0900: (service hello) has reached a steady state.
   2015-10-05 13:35:14 +0900: (service hello) stopped 1 running tasks.
 
+% hako rollback examples/hello.yml  # [EXPERIMENTAL]
+I, [2016-05-02T13:07:12.679926 #10961]  INFO -- : Current task defintion is hello:29. Rolling back to arn:aws:ecs:ap-northeast-1:XXXXXXXXXXXX:task-definition/hello:28
+I, [2016-05-02T13:07:12.959116 #10961]  INFO -- : Updated service: arn:aws:ecs:ap-northeast-1:XXXXXXXXXXXX:service/hello
+I, [2016-05-02T13:08:27.280686 #10961]  INFO -- : Deployment completed
 ```
 
 ## Front image

data/lib/hako/cli.rb

@@ -6,6 +6,7 @@ module Hako
   class CLI
     SUB_COMMANDS = %w[
       deploy
+      rollback
       oneshot
       show-yaml
       status
@@ -91,6 +92,41 @@ module Hako
       end
     end
 
+    class Rollback
+      def run(argv)
+        parse!(argv)
+        require 'hako/application'
+        require 'hako/commander'
+
+        if @verbose
+          Hako.logger.level = Logger::DEBUG
+        end
+
+        Commander.new(Application.new(@yaml_path)).rollback(dry_run: @dry_run)
+      end
+
+      def parse!(argv)
+        @dry_run = false
+        @verbose = false
+        parser.parse!(argv)
+        @yaml_path = argv.first
+
+        if @yaml_path.nil?
+          puts parser.help
+          exit 1
+        end
+      end
+
+      def parser
+        @parser ||= OptionParser.new do |opts|
+          opts.banner = 'hako rollback [OPTIONS] FILE'
+          opts.version = VERSION
+          opts.on('-n', '--dry-run', 'Enable dry-run mode') { @dry_run = true }
+          opts.on('-v', '--verbose', 'Enable verbose logging') { @verbose = true }
+        end
+      end
+    end
+
     class Oneshot
       def run(argv)
         parse!(argv)

data/lib/hako/commander.rb

@@ -29,6 +29,15 @@ module Hako
       nil
     end
 
+    # @param [Boolean] dry_run
+    # @return [nil]
+    def rollback(dry_run: false)
+      scripts = @app.yaml.fetch('scripts', []).map { |config| load_script(config, dry_run: dry_run) }
+      scheduler = load_scheduler(@app.yaml['scheduler'], scripts, dry_run: dry_run)
+
+      scheduler.rollback
+    end
+
     # @param [Array<String>] commands
     # @param [String] tag
     # @param [Hash<String, String>] env

data/lib/hako/scheduler.rb

@@ -30,6 +30,10 @@ module Hako
       raise NotImplementedError
     end
 
+    def rollback
+      raise NotImplementedError
+    end
+
     def status
       raise NotImplementedError
     end

data/lib/hako/schedulers/ecs.rb

@@ -57,6 +57,32 @@ module Hako
         end
       end
 
+      def rollback
+        current_service = describe_service
+        unless current_service
+          Hako.logger.error 'Unable to find service'
+          exit 1
+        end
+
+        task_definition = ecs_client.describe_task_definition(task_definition: current_service.task_definition).task_definition
+        current_definition = "#{task_definition.family}:#{task_definition.revision}"
+        target_definition = find_rollback_target(task_definition)
+        Hako.logger.info "Current task defintion is #{current_definition}. Rolling back to #{target_definition}"
+
+        if @dry_run
+          Hako.logger.info 'Deployment completed (dry-run)'
+        else
+          service = ecs_client.update_service(cluster: current_service.cluster_arn, service: current_service.service_arn, task_definition: target_definition).service
+          Hako.logger.info "Updated service: #{service.service_arn}"
+
+          deregistered_definition = ecs_client.deregister_task_definition(task_definition: current_definition).task_definition
+          Hako.logger.debug "Deregistered #{deregistered_definition.task_definition_arn}"
+
+          wait_for_ready(service)
+          Hako.logger.info 'Deployment completed'
+        end
+      end
+
       # @param [Hash<String, Container>] containers
       # @param [Array<String>] commands
       # @param [Hash<String, String>] env
@@ -543,6 +569,27 @@ module Hako
           events[0].id
         end
       end
+
+      # @param [Aws::ECS::Types::TaskDefinition]
+      # @return [String]
+      def find_rollback_target(task_definition)
+        if task_definition.status != 'ACTIVE'
+          raise 'Cannot find rollback target from INACTIVE task_definition!'
+        end
+
+        arn_found = false
+        ecs_client.list_task_definitions(family_prefix: task_definition.family, status: 'ACTIVE', sort: 'DESC').each do |page|
+          page.task_definition_arns.each do |arn|
+            if arn_found
+              return arn
+            elsif arn == task_definition.task_definition_arn
+              arn_found = true
+            end
+          end
+        end
+
+        raise "Unable to find rollback target. #{task_definition.task_definition_arn} is INACTIVE?"
+      end
     end
   end
 end

data/lib/hako/version.rb

@@ -1,4 +1,4 @@
 # frozen_string_literal: true
 module Hako
-  VERSION = '0.14.1'
+  VERSION = '0.15.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: hako
 version: !ruby/object:Gem::Version
-  version: 0.14.1
+  version: 0.15.0
 platform: ruby
 authors:
 - Kohei Suzuki
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-04-26 00:00:00.000000000 Z
+date: 2016-05-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk