etcd-discovery 0.0.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: f181a1f205a862c7cd79ab91c0c57fce2d783224
+  data.tar.gz: 9b9e09314de9574c470be1d642e1e0dafd32c16f
+SHA512:
+  metadata.gz: 7b4582670d24dabcdd52a1cc928247dae9588796a8eb43e7f3635cd242b4365bd877e9d903f818a94009cb0d23be3844a3cde6c938ccbb423f572306a3e13e0d
+  data.tar.gz: e21168f6a1c61168cc4e9ac45ee6ad6828571ba1ffc1bad42afc710434b1561f3228910a93f78f1dc46a35e083cd4f2ac73a73d1cf3fd8baac540c54985221ea

data/lib/etcd-discovery.rb

@@ -0,0 +1,31 @@
+## Require files from etcd-discovery
+
+require 'etcd'
+require 'json'
+
+dir = File.join File.dirname(__FILE__), "etcd-discovery"
+Dir["#{dir}/*.rb"].each do |file|
+  require file
+end
+
+module EtcdDiscovery
+  attr_writer :config
+
+  def self.config
+    @config ||= Config.new
+  end
+
+  def self.configure(&block)
+    yield config if block_given?
+    config.validate
+  end
+
+  # For a cleaner API
+  def self.get(service)
+    Service.get(service)
+  end
+
+  def self.register(service, host)
+    Thread.new { Service.register(service, host) }
+  end
+end

data/lib/etcd-discovery/client.rb

@@ -0,0 +1,16 @@
+module EtcdDiscovery
+  module Client
+    def self.create(config)
+      if config.use_ssl
+        Etcd.client host: config.host, port: config.port do |config|
+          config.use_ssl = true
+          config.ca_file = File.expand_path(ENV["ETCD_CACERT"], __FILE__)
+          config.ssl_cert = OpenSSL::X509::Certificate.new(File.read(ENV["ETCD_TLS_CERT"]))
+          config.ssl_key = OpenSSL::PKey::RSA.new(File.read(ENV["ETCD_TLS_KEY"]))
+        end
+      else
+        Etcd.client host: config.host, port: config.port
+      end
+    end
+  end
+end

data/lib/etcd-discovery/config.rb

@@ -0,0 +1,34 @@
+module EtcdDiscovery
+  class InvalidSSLConfig < RuntimeError
+  end
+
+  class Config
+    attr_accessor :use_ssl, :cacert, :ssl_key, :ssl_cert
+    attr_accessor :register_ttl, :register_renew
+    attr_accessor :host, :port
+
+    def initialize
+      @use_ssl = false
+      @host = "localhost"
+      @port = "4001"
+      @register_ttl = 10
+      @register_renew = 8
+    end
+
+    def validate
+      if use_ssl
+        if cacert.nil? or !File.exists? cacert
+          raise InvalidSSLConfig, "cacert"
+        elsif ssl_key.nil? or !File.exists? ssl_key
+          raise InvalidSSLConfig, "ssl_key"
+        elsif ssl_cert.nil? or !File.exists? ssl_cert
+          raise InvalidSSLConfig, "ssl_cert"
+        end
+      end
+    end
+
+    def client
+      @client ||= Client.create self
+    end
+  end
+end

data/lib/etcd-discovery/host.rb

@@ -0,0 +1,41 @@
+class Hash
+  def capitalize_keys
+    new_hash = {}
+    self.each do |k,v|
+      new_hash[k.to_s.capitalize] = v
+    end
+    return new_hash
+  end
+end
+
+module EtcdDiscovery
+  class Host
+    attr_accessor :hostname, :port, :user, :password
+
+    def initialize(arg)
+      if arg.is_a? Etcd::Node
+        params = JSON.parse arg.value
+      elsif arg.is_a? Hash
+        params = arg.capitalize_keys
+      else
+        raise TypeError, "requires a Etcd::Node or a Hash, not a #{arg.class}"
+      end
+      @hostname = params['Name']
+      @port = params['Port']
+      @user = params['User']
+      @password = params['Password']
+
+      if @hostname.nil? or @hostname.empty?
+        @hostname = Socket.gethostname
+      end
+    end
+
+    def to_json
+      {"Name" => @hostname, "Port" => @port, "User" => @user, "Password" => @password}.to_json
+    end
+
+    def to_uri
+      URI("http://#{user}:#{password}@#{hostname}:#{port}")
+    end
+  end
+end

data/lib/etcd-discovery/service.rb

@@ -0,0 +1,45 @@
+module EtcdDiscovery
+  class ServiceNotFound < RuntimeError
+  end
+
+  class Service
+    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
+        node = client.get("/services/#{service}", recursive: true)
+      rescue Etcd::KeyNotFound
+        raise ServiceNotFound, service
+      end
+      if node.children.length == 0
+        raise ServiceNotFound, service
+      end
+
+      hosts = []
+      node.children.each do |c|
+        hosts << Host.new(c)
+      end
+      return hosts
+    end
+
+    def self.register(service, host)
+      if host.is_a? Hash
+        h = Host.new host
+      elsif host.is_a? Etcd::Host
+        h = host
+      else
+        raise TypeError, "host should be a Hash or a Etcd::Host, is a #{host.class}"
+      end
+
+      config = EtcdDiscovery.config
+      client = config.client
+      value = h.to_json
+
+      while true
+        client.set("/services/#{service}/#{h.hostname}", value: value, ttl: config.register_ttl)
+        sleep config.register_renew
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,48 @@
+--- !ruby/object:Gem::Specification
+name: etcd-discovery
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+platform: ruby
+authors:
+- Léo Unbekandt
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-04-11 00:00:00.000000000 Z
+dependencies: []
+description: Ruby implementation of a service discovery tool based on etcd
+email: leo.unbekandt@appsdeck.eu
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/etcd-discovery.rb
+- lib/etcd-discovery/client.rb
+- lib/etcd-discovery/config.rb
+- lib/etcd-discovery/host.rb
+- lib/etcd-discovery/service.rb
+homepage: http://github.com/Appsdeck/etcd-discovery-ruby
+licenses:
+- BSD
+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.2.2
+signing_key: 
+specification_version: 4
+summary: Service discovery based on etcd
+test_files: []