frosty 0.0.1.alpha1

checksums.yaml

@@ -0,0 +1,15 @@
+---
+!binary "U0hBMQ==":
+  metadata.gz: !binary |-
+    N2Y3YjU1MjM3YWM2MmM3ZDdiZGE0NjUzMzg4MzEyOWVhNzRlNTcyZA==
+  data.tar.gz: !binary |-
+    MzgxNWM0YjM5MjEzNjkyMjEwNWUxYzQ5N2U1YTMwNzcxYTZiNDY1Mg==
+!binary "U0hBNTEy":
+  metadata.gz: !binary |-
+    OWEyNzhlNzJmNDZlYzM0OWEwNzQwMGYwYmIwMWQ2N2U3Y2FmZTZhZWYxZTQ4
+    ZjMwMzRiODFkMDBmMDQ3ODY5NTZlODFmMmExM2VhZDNiZDFjM2QwNWJkYzZl
+    NDA4NWY5YTYwZmQ3MmMyMDkwMzZmNWE0NDRlN2FjNmYyMGY3NTY=
+  data.tar.gz: !binary |-
+    MDgwYTkzMmI3NmI1NzhmY2M5YTJjN2I0NWVlYWU0NjI3MjhhZTQwMTY3Mzkz
+    YzViYmZhOWE3ZTBiMjUxYTk1ZWYxMzI5ODQ1MzFiNzZjNmMyYjg5NjhjNDVk
+    YWNkMzQwNzMxZTNmNTM5YzZhNjU4MTJjMzRkZGIzYmViNjIwYmY=

data/lib/frosty.rb

@@ -0,0 +1,61 @@
+require 'frosty/service'
+require 'frosty/node'
+require 'frosty/etcd'
+require 'etcd'
+
+class Frosty
+  include Frosty::Etcd
+
+  attr_accessor :environment, :server, :port
+  def initialize(server = '127.0.0.1', port = '4001', options ={})
+    @server = server
+    @port = port
+    @environment = options['environment'] || 'default'
+  end
+
+  def can_connect?
+    connect
+    ret = true
+    begin
+      @connection.get('/')
+    rescue
+      ret = false
+    end
+    ret
+  end
+
+  def connect
+    @connection = ::Etcd.client(:host => @server,
+                               :port => @port)
+  end
+
+  def services
+    array_of_keys("/#{@environment}/services/")
+  end
+
+  def nodes
+    array_of_keys("/#{@environment}/nodes/")
+  end
+
+  # This probably needs to be re-worked
+  def array_of_keys(key)
+    n = query(@connection, key)
+    if n.nil?
+      []
+    else
+      a = []
+      n.each do |value|
+        a.push(value.key.gsub(key, ''))
+      end
+      a
+    end
+  end
+
+  def service(name)
+    Frosty::Service.new(name, @environment, @connection)
+  end
+
+  def node(name)
+    Frosty::Node.new(name, @environment, @connection)
+  end
+end

data/lib/frosty/etcd.rb

@@ -0,0 +1,29 @@
+class Frosty
+  module Etcd
+    def query(connection, key)
+      begin
+        connection.get(key)
+      rescue
+        nil
+      end
+    end
+
+    def write(connection, key, value)
+      begin
+        connection.set(key, value)
+      rescue
+        nil
+      end
+    end
+
+    def hashify(responses)
+      h = {}
+      responses.each do |response|
+        unless response.value.nil?
+          h[response.key.split('/').pop] = response.value
+        end
+      end
+      h
+    end
+  end
+end

data/lib/frosty/node.rb

@@ -0,0 +1,50 @@
+require 'frosty/etcd'
+
+class Frosty
+  class Node
+    include Frosty::Etcd
+
+    def initialize(ip, environment, connection)
+      @ip = ip
+      @connection = connection
+      @base_path = "/#{environment}/nodes/#{ip}/"
+      refresh
+    end
+
+    def exists?
+      if query(@connection, @base_path).nil?
+        false
+      else
+        true
+      end
+    end
+
+    def fetch_services
+      ret = []
+      if exists?
+        services = query(@connection, @base_path + 'services')
+        unless services.nil?
+          ret = services.value.split(',')
+        end
+      end
+      ret
+    end
+
+    def refresh
+      @services = fetch_services
+    end
+
+    def add_service(name)
+      @services.push name
+    end
+
+    def services
+      @services
+    end
+
+    def save
+      services = @services.join(',')
+      write(@connection, @base_path + 'services', services)
+    end
+  end
+end

data/lib/frosty/service.rb

@@ -0,0 +1,105 @@
+require_relative 'service/endpoint'
+require 'frosty/etcd'
+
+class Frosty
+  class Service
+    include Frosty::Etcd
+
+    attr_accessor :role, :status
+    def initialize(name, environment, connection)
+      @name = name
+      @connection = connection
+      @base_path = "/#{environment}/services/#{name}/"
+      @endpoints = fetch_endpoints
+    end
+
+    def exists?
+      if query(@connection, @base_path).nil?
+        false
+      else
+        true
+      end
+    end
+
+    def has_master?
+      ret = false
+      if self.exists?
+        @endpoints.each do |ep|
+          if ep.role == 'master'
+            ret = true
+            break
+          end
+        end
+      end
+      ret
+    end
+
+    # return ip of master
+    def masters
+      m = []
+      if self.has_master?
+        @endpoints.each do |ep|
+          if ep.role == 'master'
+            m.push ep.ip
+          end
+        end
+      end
+      m
+    end
+
+    def refresh
+      @endpoints = fetch_endpoints
+    end
+
+    def fetch_endpoints
+      ret = []
+      endpoints = query(@connection, @base_path + 'endpoints')
+      unless endpoints.nil?
+        endpoints.each do |e|
+          ret.push endpoint(e.key.split('/').pop)
+        end
+      end
+      ret
+    end
+
+    def endpoints
+      ret = []
+      @endpoints.each do |endpoint|
+        ret.push endpoint.ip
+      end
+      ret
+    end
+
+    def endpoint(ip)
+      resp = query(@connection, @base_path + "endpoints/#{ip}")
+
+      ep = Frosty::Service::Endpoint.new(ip)
+      if resp.nil?
+        if has_master?
+          ep.role = 'slave'
+        else
+          ep.role = 'master'
+        end
+      else
+        info = hashify(resp)
+        ep.port = info['port']
+        ep.role = info['role']
+        ep.status = info['status']
+      end
+      ep
+    end
+
+    def save_endpoint(ep)
+      if ep.port.nil?
+        raise TypeError, "Endpoint port must be either a string or an integer"
+      end
+      path = @base_path + "endpoints/#{ep.ip}/"
+      write(@connection, path + "port", ep.port)
+      write(@connection, path + "role", ep.role)
+      unless ep.status.nil?
+        write(@connection, path + "status", ep.status)
+      end
+      refresh
+    end
+  end
+end

data/lib/frosty/service/endpoint.rb

@@ -0,0 +1,11 @@
+class Frosty
+  class Service
+    class Endpoint
+      attr_accessor :status, :port, :role
+      attr_reader :ip
+      def initialize(ip)
+        @ip = ip
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,62 @@
+--- !ruby/object:Gem::Specification
+name: frosty
+version: !ruby/object:Gem::Version
+  version: 0.0.1.alpha1
+platform: ruby
+authors:
+- Jim Rosser
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-09-30 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: etcd
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Provides Libraries to query and save to etcd based on the enigma project
+email: jarosser06@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/frosty.rb
+- lib/frosty/node.rb
+- lib/frosty/service.rb
+- lib/frosty/etcd.rb
+- lib/frosty/service/endpoint.rb
+homepage: https://github.com/enigmaproject/frosty
+licenses:
+- mit
+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: 1.3.1
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.6
+signing_key: 
+specification_version: 4
+summary: Etcd with Chef
+test_files: []