Contents

class KuberKit::ImageCompiler::BuildServerPool
  attr_reader :ssh_shells, :local_shell
  
  def initialize(local_shell:, build_servers:, ssh_shell_class:)
    @local_shell     = local_shell
    @ssh_shell_class = ssh_shell_class
    @ssh_shells      = connect_to_ssh_shells(build_servers)
  end

  def get_shell
    if @ssh_shells.any?
      @ssh_shells.sample
    else
      @local_shell
    end
  end

  def disconnect_all
    @ssh_shells.each(&:disconnect)
  end

  private
    def connect_to_ssh_shells(build_servers)
      build_servers.map do |bs|
        shell = @ssh_shell_class.new
        shell.connect(host: bs.host, user: bs.user, port: bs.port)
        shell
      end
    end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
kuber_kit-0.2.1	lib/kuber_kit/image_compiler/build_server_pool.rb
kuber_kit-0.2.0	lib/kuber_kit/image_compiler/build_server_pool.rb
kuber_kit-0.1.9	lib/kuber_kit/image_compiler/build_server_pool.rb