expri 0.1.0

data/lib/expri.rb

@@ -0,0 +1,10 @@
+require_relative "expri/dumper"
+require_relative "expri/metric"
+require_relative "expri/metric_set"
+
+module Expri
+  def self.metrics(options={}, &block)
+    options[:verbose] ||= true
+    MetricSet.new(options).run(&block)
+  end
+end

data/lib/expri/dumper.rb

@@ -0,0 +1,52 @@
+module Expri
+  class Dumper
+    def initialize(options={})
+      @options = options
+    end
+
+    def run
+      str = ""
+      str << header
+      metrics = Metric.list
+      metrics.each do |metric|
+        metric = Metric.new(metric, {}, @options)
+        metric.get
+        str << dump_metric(metric)
+      end
+      str << footer
+    end
+
+    private
+
+    NUMERICAL_ATTRIBUTES = %w{divisor frequency precentile window}.
+      map { |x| x.to_sym }.freeze
+
+    def dump_metric(metric)
+      attributes = metric.attributes_without_defaults.dup
+
+      type           = attributes.delete(:type)
+      predicate      = attributes.delete(:predicate)
+
+      # at bare minimum, window is required, so attributes size should always
+      # be > 0 even after type and predicate are removed
+      attributes_str = attributes.to_a.
+        sort_by { |a| a[0] }.
+        map { |a| "#{a[0]}: #{NUMERICAL_ATTRIBUTES.include?(a[0]) ? a[1] : %{"#{a[1]}"}}" }.
+        join(",\n    ")
+
+      # predicate is important, but optional
+      str =  %{  metric "#{metric.name}",\n}
+      str << %{    type: :#{type},\n}
+      str << %{    predicate: #{predicate},\n} if predicate
+      str << %{    #{attributes_str}\n\n}
+    end
+
+    def footer
+      "end"
+    end
+
+    def header
+      "Expri.metrics do\n\n"
+    end
+  end
+end

data/lib/expri/metric.rb

@@ -0,0 +1,91 @@
+module Expri
+  class Metric
+    ATTRIBUTE_DEFAULTS = { :frequency => 3 }.freeze
+    VALID_ATTRIBUTES = %w{divisor frequency key percentile predicate predicate2
+      type window value}.map { |x| x.to_sym }.freeze
+
+    attr_accessor :attributes, :name, :options
+
+    def self.list(options={})
+      metric = Metric.new(nil, {}, {})
+      MultiJson.decode(metric.api.get(path: "/metrics", expects: 200).body)
+    end
+
+    def initialize(name, attributes={}, options={})
+      @name           = name
+      @options        = options
+      self.attributes = attributes
+    end
+
+    def ==(other)
+      attributes_without_defaults == other.attributes_without_defaults
+    end
+
+    def api
+      @api ||= Excon.new(api_url)
+    end
+
+    def attributes=(attributes)
+      # symbolize keys
+      attributes.keys.each do |key|
+        attributes[(key.to_sym rescue key) || key] = attributes.delete(key)
+      end
+
+      attributes.delete_if { |k, v| !VALID_ATTRIBUTES.include?(k) }
+      attributes.delete_if { |k, v| v == nil }
+
+      # manual normalization
+      attributes[:type] = attributes[:type].to_sym if attributes[:type]
+
+      @attributes = attributes
+    end
+
+    # only the attributes that are not at their default value
+    def attributes_without_defaults
+      attributes = @attributes.dup
+      attributes.delete_if { |k, v| ATTRIBUTE_DEFAULTS[k] == v }
+    end
+
+    def destroy
+      puts "destroy name=#{name}" if @options[:verbose]
+      api.delete(path: "/metrics/#{name}", body: MultiJson.encode(@attributes),
+        expects: 200)
+    end
+
+    def get
+      puts "show name=#{name}" if @options[:verbose]
+      self.attributes =
+        MultiJson.decode(api.get(path: "/metrics/#{name}", expects: 200).body)
+    end
+
+    def post
+      puts "create name=#{name}" if @options[:verbose]
+      # exprd's put is like most api's post
+      api.put(path: "/metrics/#{name}", body: MultiJson.encode(@attributes),
+        expects: 200)
+    rescue Excon::Errors::BadRequest => e
+      message = MultiJson.decode(e.response.body)["message"]
+      raise(message)
+    end
+
+    def put
+      begin
+        existing = Metric.new(name, {}, @options)
+        existing.get
+        if self != existing
+          existing.destroy
+          post
+        end
+      rescue Excon::Errors::NotFound
+        post
+      end
+    end
+
+    private
+
+    def api_url
+      @options[:exprd_api_url] || ENV["EXPRD_API_URL"] ||
+        raise("need EXPRD_API")
+    end
+  end
+end

data/lib/expri/metric_set.rb

@@ -0,0 +1,39 @@
+module Expri
+  class MetricSet
+    def metric(name, attributes={}, options={})
+      @metrics << Metric.new(name, attributes, @options.merge(options))
+    end
+
+    def initialize(options={})
+      @options = { :destroy_extras => true }.merge(options)
+    end
+
+    def run(&block)
+      puts "run options=#{@options}" if @options[:verbose]
+      cache_current if @options[:destroy_extras]
+
+      @metrics = []
+      instance_eval(&block)
+      @metrics.each do |metric|
+        metric.put
+      end
+
+      destroy_extras if @options[:destroy_extras]
+    end
+
+    private
+
+    def cache_current
+      @current_metrics = Metric.list
+    end
+
+    def destroy_extras
+      puts "check_extras" if @options[:verbose]
+      extra_metrics = @current_metrics - @metrics.map(&:name)
+      extra_metrics.each do |metric|
+        metric = Metric.new(metric, {}, @options)
+        metric.destroy
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,81 @@
+--- !ruby/object:Gem::Specification
+name: expri
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Brandur
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-10-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: excon
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.16.4
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.16.4
+- !ruby/object:Gem::Dependency
+  name: multi_json
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.3.6
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.3.6
+description: 
+email: brandur@mutelight.org
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/expri/dumper.rb
+- lib/expri/metric.rb
+- lib/expri/metric_set.rb
+- lib/expri.rb
+homepage: https://github.com/heroku/expri
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Versionable metrics recipes for exprd.
+test_files: []