rollo 0.1.0

data/lib/rollo/model/host_cluster.rb

@@ -0,0 +1,128 @@
+require 'aws-sdk'
+require 'hollerback'
+
+require_relative './scaling_activity'
+require_relative './host'
+
+module Rollo
+  module Model
+    class HostCluster
+      attr_reader :last_scaling_activity
+
+      def initialize(asg_name, region, asg_resource = nil, waiter = nil)
+        @region = region
+        @asg_name = asg_name
+        @asg_resource = asg_resource ||
+            Aws::AutoScaling::Resource.new(region: region)
+        @asg = @asg_resource.group(@asg_name)
+        record_latest_scaling_activity
+
+        @waiter = waiter || Wait.new(attempts: 300, timeout: 30, delay: 5)
+      end
+
+      def reload
+        @asg.reload
+      end
+
+      def name
+        @asg_name
+      end
+
+      def desired_capacity
+        @asg.desired_capacity
+      end
+
+      def desired_capacity=(capacity)
+        @asg.set_desired_capacity({desired_capacity: capacity})
+      end
+
+      def has_desired_capacity?
+        hosts.size == desired_capacity &&
+            hosts.all? {|h| h.is_in_service? && h.is_healthy?}
+      end
+
+      def scaling_activities
+        @asg.activities.collect {|a| ScalingActivity.new(a)}
+      end
+
+      def has_started_changing_capacity?
+        scaling_activities
+            .select {|a| a.started_after_completion_of?(@last_scaling_activity)}
+            .size > 0
+      end
+
+      def has_completed_changing_capacity?
+        scaling_activities.all?(&:is_complete?)
+      end
+
+      def hosts
+        @asg.instances.collect {|h| Host.new(h)}
+      end
+
+      def increase_capacity_by(capacity_delta, &block)
+        initial = desired_capacity
+        increased = initial + capacity_delta
+
+        callbacks_for(block).try_respond_with(
+            :prepare, initial, increased)
+
+        ensure_capacity_changed_to(increased, &block)
+      end
+
+      def decrease_capacity_by(capacity_delta, &block)
+        initial = desired_capacity
+        decreased = initial - capacity_delta
+
+        callbacks_for(block).try_respond_with(
+            :prepare, initial, decreased)
+
+        ensure_capacity_changed_to(decreased, &block)
+      end
+
+      def ensure_capacity_changed_to(capacity, &block)
+        self.desired_capacity = capacity
+        wait_for_capacity_change_start(&block)
+        wait_for_capacity_change_end(&block)
+        wait_for_capacity_health(&block)
+        record_latest_scaling_activity
+      end
+
+      def wait_for_capacity_change_start(&block)
+        @waiter.until do |attempt|
+          reload
+          callbacks_for(block)
+              .try_respond_with(:waiting_for_start, attempt) if block
+          has_started_changing_capacity?
+        end
+      end
+
+      def wait_for_capacity_change_end(&block)
+        @waiter.until do |attempt|
+          reload
+          callbacks_for(block)
+              .try_respond_with(:waiting_for_end, attempt) if block
+          has_completed_changing_capacity?
+        end
+      end
+
+      def wait_for_capacity_health(&block)
+        @waiter.until do |attempt|
+          reload
+          callbacks_for(block)
+              .try_respond_with(:waiting_for_health, attempt) if block
+          has_desired_capacity?
+        end
+      end
+
+      def record_latest_scaling_activity
+        @last_scaling_activity = scaling_activities.first
+      end
+
+      private
+
+      def callbacks_for(block)
+        Hollerback::Callbacks.new(block)
+      end
+    end
+  end
+end

data/lib/rollo/model/scaling_activity.rb

@@ -0,0 +1,32 @@
+module Rollo
+  module Model
+    class ScalingActivity
+      def initialize(activity)
+        @activity = activity
+      end
+
+      def id
+        @activity.activity_id
+      end
+
+      def start_time
+        @activity.start_time
+      end
+
+      def end_time
+        @activity.end_time
+      end
+
+      def started_after_completion_of?(other)
+        self.id != other.id &&
+            !self.start_time.nil? &&
+            !other.end_time.nil? &&
+            self.start_time > other.end_time
+      end
+
+      def is_complete?
+        %w(Successful Failed Cancelled).include?(@activity.status_code)
+      end
+    end
+  end
+end

data/lib/rollo/model/service.rb

@@ -0,0 +1,104 @@
+require 'aws-sdk'
+require 'wait'
+require 'hollerback'
+
+module Rollo
+  module Model
+    class Service
+      def initialize(
+          ecs_cluster_name, ecs_service_arn, region,
+              ecs_resource = nil, waiter = nil)
+        @ecs_cluster_name = ecs_cluster_name
+        @ecs_service_arn = ecs_service_arn
+        @ecs_resource = ecs_resource || Aws::ECS::Resource.new(region: region)
+        reload
+
+        @waiter = waiter || Wait.new(attempts: 300, timeout: 30, delay: 5)
+      end
+
+      def name
+        @ecs_service.service_name
+      end
+
+      def instance
+        @ecs_service
+      end
+
+      def reload
+        @ecs_service = get_ecs_service
+      end
+
+      def is_replica?
+        @ecs_service.scheduling_strategy == 'REPLICA'
+      end
+
+      def running_count
+        @ecs_service.running_count
+      end
+
+      def desired_count
+        @ecs_service.desired_count
+      end
+
+      def desired_count=(count)
+        @ecs_resource.client
+            .update_service(
+                cluster: @ecs_cluster_name,
+                service: @ecs_service_arn,
+                desired_count: count)
+      end
+
+      def has_desired_count?
+        running_count == desired_count
+      end
+
+      def increase_instance_count_by(count_delta, &block)
+        initial = desired_count
+        increased = initial + count_delta
+
+        callbacks_for(block).try_respond_with(
+            :prepare, initial, increased)
+
+        ensure_instance_count(increased, &block)
+      end
+
+      def decrease_instance_count_by(count_delta, &block)
+        initial = desired_count
+        decreased = initial - count_delta
+
+        callbacks_for(block).try_respond_with(
+            :prepare, initial, decreased)
+
+        ensure_instance_count(decreased, &block)
+      end
+
+      def ensure_instance_count(count, &block)
+        self.desired_count = count
+        wait_for_service_health(&block)
+      end
+
+      def wait_for_service_health(&block)
+        @waiter.until do |attempt|
+          reload
+          callbacks_for(block)
+              .try_respond_with(:waiting_for_health, attempt) if block
+          has_desired_count?
+        end
+      end
+
+      private
+
+      def get_ecs_service
+        @ecs_resource.client
+            .describe_services(
+                cluster: @ecs_cluster_name,
+                services: [@ecs_service_arn])
+            .services[0]
+      end
+
+      def callbacks_for(block)
+        Hollerback::Callbacks.new(block)
+      end
+    end
+  end
+end

data/lib/rollo/model/service_cluster.rb

@@ -0,0 +1,53 @@
+require 'aws-sdk'
+require 'hollerback'
+
+require_relative './service'
+
+module Rollo
+  module Model
+    class ServiceCluster
+      def initialize(ecs_cluster_name, region, ecs_resource = nil)
+        @region = region
+        @ecs_cluster_name = ecs_cluster_name
+        @ecs_resource = ecs_resource || Aws::ECS::Resource.new(region: region)
+        @ecs_cluster = get_ecs_cluster
+      end
+
+      def name
+        @ecs_cluster_name
+      end
+
+      def replica_services
+        get_ecs_service_arns
+            .collect {|arn| Service.new(@ecs_cluster_name, arn, @region)}
+            .select(&:is_replica?)
+      end
+
+      def with_replica_services(&block)
+        all_replica_services = replica_services
+
+        callbacks = Hollerback::Callbacks.new(block)
+        callbacks.try_respond_with(
+            :start, all_replica_services)
+
+        all_replica_services.each do |service|
+          callbacks.try_respond_with(:each_service, service)
+        end
+      end
+
+      private
+
+      def get_ecs_cluster
+        @ecs_resource.client
+            .describe_clusters(clusters: [@ecs_cluster_name])
+            .clusters[0]
+      end
+
+      def get_ecs_service_arns
+        @ecs_resource.client
+            .list_services(cluster: @ecs_cluster.cluster_name)
+            .inject([]) {|arns, response| arns + response.service_arns}
+      end
+    end
+  end
+end

data/lib/rollo/model.rb

@@ -0,0 +1,2 @@
+require_relative 'model/host_cluster'
+require_relative 'model/service_cluster'

data/lib/rollo/version.rb

@@ -0,0 +1,3 @@
+module Rollo
+  VERSION = "0.1.0"
+end

data/lib/rollo.rb

@@ -0,0 +1,5 @@
+require_relative "rollo/version"
+require_relative 'rollo/commands'
+
+module Rollo
+end

data/rollo.gemspec

@@ -0,0 +1,37 @@
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'rollo/version'
+require 'date'
+
+Gem::Specification.new do |spec|
+  spec.name = 'rollo'
+  spec.version = Rollo::VERSION
+  spec.authors = ['Toby Clemson']
+  spec.email = ['tobyclemson@gmail.com']
+
+  spec.date = Date.today.to_s
+  spec.summary = 'Cluster / service roller for AWS ECS.'
+  spec.description = 'Strategies for rolling ECS container instance clusters.'
+  spec.homepage = 'https://github.com/infrablocks/rollo'
+  spec.license = "MIT"
+
+  spec.files = `git ls-files -z`.split("\x0").reject do |f|
+    f.match(%r{^(test|spec|features)/})
+  end
+  spec.bindir = 'exe'
+  spec.executables = spec.files.grep(%r{^exe/}) {|f| File.basename(f)}
+  spec.require_paths = ['lib']
+
+  spec.add_dependency 'aws-sdk', '~> 3.0'
+  spec.add_dependency 'aws-sdk-ecs', '~> 1.22'
+  spec.add_dependency 'wait', '~> 0.5'
+  spec.add_dependency 'hollerback', '~> 0.1'
+  spec.add_dependency 'thor', '~> 0.20'
+
+  spec.add_development_dependency 'bundler', '~> 1.17'
+  spec.add_development_dependency 'rake', '~> 12.3'
+  spec.add_development_dependency 'rspec', '~> 3.8'
+  spec.add_development_dependency 'aruba', '~> 0.14'
+  spec.add_development_dependency 'gem-release', '~> 2.0'
+  spec.add_development_dependency 'irbtools', '~> 2.2'
+end

metadata

@@ -0,0 +1,227 @@
+--- !ruby/object:Gem::Specification
+name: rollo
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Toby Clemson
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2018-11-28 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-ecs
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.22'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.22'
+- !ruby/object:Gem::Dependency
+  name: wait
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.5'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.5'
+- !ruby/object:Gem::Dependency
+  name: hollerback
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.20'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.20'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.17'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.17'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '12.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '12.3'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.8'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.8'
+- !ruby/object:Gem::Dependency
+  name: aruba
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.14'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.14'
+- !ruby/object:Gem::Dependency
+  name: gem-release
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: irbtools
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+description: Strategies for rolling ECS container instance clusters.
+email:
+- tobyclemson@gmail.com
+executables:
+- rollo
+extensions: []
+extra_rdoc_files: []
+files:
+- ".envrc"
+- ".gitignore"
+- ".rspec"
+- ".ruby-version"
+- ".travis.yml"
+- CODE_OF_CONDUCT.md
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- exe/rollo
+- go
+- lib/rollo.rb
+- lib/rollo/commands.rb
+- lib/rollo/commands/host_cluster.rb
+- lib/rollo/commands/main.rb
+- lib/rollo/commands/service_cluster.rb
+- lib/rollo/model.rb
+- lib/rollo/model/host.rb
+- lib/rollo/model/host_cluster.rb
+- lib/rollo/model/scaling_activity.rb
+- lib/rollo/model/service.rb
+- lib/rollo/model/service_cluster.rb
+- lib/rollo/version.rb
+- rollo.gemspec
+homepage: https://github.com/infrablocks/rollo
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.2.3
+signing_key: 
+specification_version: 4
+summary: Cluster / service roller for AWS ECS.
+test_files: []