class EtcdDiscovery::Service

Attributes

attributes[RW]

Public Class Methods

get(service) click to toggle source
# File lib/etcd-discovery/service.rb, line 18
def self.get(service)
  raise TypeError, "service should be a String, is a #{service.class}" unless service.is_a? String

  client = EtcdDiscovery.config.client
  begin
    service = client.get("/services_infos/#{service}")
    new service.node
  rescue Etcd::KeyNotFound
    new("name" => service)
  end
end
new(arg) click to toggle source
# File lib/etcd-discovery/service.rb, line 8
def initialize(arg)
  if arg.is_a? Etcd::Node
    @attributes = JSON.parse arg.value
  elsif arg.is_a? Hash
    @attributes = arg
  else
    raise TypeError, "requires a Etcd::Node or a Hash, not a #{arg.class}"
  end
end

Public Instance Methods

all() click to toggle source
# File lib/etcd-discovery/service.rb, line 30
def all
  client = EtcdDiscovery.config.client
  begin
    node = client.get("/services/#{attributes["name"]}", recursive: true).node
  rescue Etcd::KeyNotFound
    raise ServiceNotFound, attributes["name"]
  end
  raise ServiceNotFound, attributes["name"] if node.children.empty?
  node.children.map { |c| Host.new(c) }
end
one() click to toggle source
# File lib/etcd-discovery/service.rb, line 41
def one
  all.sample
end
set_credentials(user, password) click to toggle source
# File lib/etcd-discovery/service.rb, line 68
def set_credentials(user, password)
  @attributes["user"] = user
  @attributes["password"] = password
end
to_json() click to toggle source
# File lib/etcd-discovery/service.rb, line 64
def to_json
  attributes.to_json
end
to_s() click to toggle source
# File lib/etcd-discovery/service.rb, line 60
def to_s
  to_uri.to_s
end
to_uri(schemes = ["https", "http"]) click to toggle source
# File lib/etcd-discovery/service.rb, line 45
def to_uri(schemes = ["https", "http"])
  a = attributes
  return one.to_uri(schemes) unless a["public"]

  schemes = [schemes] if !schemes.is_a?(Array)
  scheme = schemes.reject do |s|
    a["ports"][s].nil?
  end.first
  if a["user"].nil? || a["user"] == ""
    URI("#{scheme}://#{a["hostname"]}:#{a["ports"][scheme]}")
  else
    URI("#{scheme}://#{a["user"]}:#{a["password"]}@#{a["hostname"]}:#{a["ports"][scheme]}")
  end
end