Contents

# frozen_string_literal: true

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:)
        @ecs_cluster = ecs_cluster
      end

      def name
        @ecs_cluster_name
      end

      def replica_services
        ecs_service_arns
          .collect { |arn| Service.new(@ecs_cluster_name, arn, @region) }
          .select(&: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 ecs_cluster
        @ecs_resource
          .client
          .describe_clusters(clusters: [@ecs_cluster_name])
          .clusters[0]
      end

      def ecs_service_arns
        @ecs_resource
          .client
          .list_services(cluster: @ecs_cluster.cluster_name)
          .inject([]) { |arns, response| arns + response.service_arns }
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
rollo-0.9.0.pre.7	lib/rollo/model/service_cluster.rb