Sha256: 649c7ebce09f75dd275474d3523092951892393eef2e08d56cc2acbf0e2ee040
Contents?: true
Size: 1.87 KB
Versions: 65
Compression:
Stored size: 1.87 KB
Contents
module Elasticsearch
module Transport
module Transport
module Connections
module Selector
# @abstract Common functionality for connection selector implementations.
#
module Base
attr_reader :connections
# @option arguments [Connections::Collection] :connections Collection with connections.
#
def initialize(arguments={})
@connections = arguments[:connections]
end
# @abstract Selector strategies implement this method to
# select and return a connection from the pool.
#
# @return [Connection]
#
def select(options={})
raise NoMethodError, "Implement this method in the selector implementation."
end
end
# "Random connection" selector strategy.
#
class Random
include Base
# Returns a random connection from the collection.
#
# @return [Connections::Connection]
#
def select(options={})
connections.to_a.send( defined?(RUBY_VERSION) && RUBY_VERSION > '1.9' ? :sample : :choice)
end
end
# "Round-robin" selector strategy (default).
#
class RoundRobin
include Base
# Returns the next connection from the collection, rotating them in round-robin fashion.
#
# @return [Connections::Connection]
#
def select(options={})
# On Ruby 1.9, Array#rotate could be used instead
@current = !defined?(@current) || @current.nil? ? 0 : @current+1
@current = 0 if @current >= connections.size
connections[@current]
end
end
end
end
end
end
end
Version data entries
65 entries across 65 versions & 6 rubygems