Created
October 24, 2011 16:34
-
-
Save kredmer/1309454 to your computer and use it in GitHub Desktop.
module RestfulMetrics - monkey patch
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| module RestfulMetrics | |
| class Client | |
| extend LogTools | |
| @@connection = nil | |
| @@debug, @@async, @@disabled = false | |
| @@fqdn = nil | |
| class << self | |
| def set_credentials(api_key) | |
| @@connection = Connection.new(api_key) | |
| self.debug = @@debug | |
| true | |
| end | |
| def set_app_id(fqdn) | |
| @@fqdn=fqdn | |
| end | |
| def debug=(debug_flag) | |
| @@debug = debug_flag | |
| @@connection.debug = @@debug if @@connection | |
| end | |
| def debug | |
| @@debug | |
| end | |
| def async=(async_flag) | |
| # DelayedJob integration | |
| require 'delayed_job' if async_flag | |
| @@async = async_flag && (defined?(Delayed) != nil) | |
| @@connection.async = @@async if @@connection | |
| end | |
| def async? | |
| @@async | |
| end | |
| def disabled=(disabled_flag) | |
| @@disabled = disabled_flag | |
| end | |
| def disabled? | |
| @@disabled | |
| end | |
| def add_metric(fqdn, name, value, distinct_id = nil) | |
| params = Hash.new | |
| params[:metric] = Hash.new | |
| params[:metric][:fqdn] = fqdn | |
| params[:metric][:name] = name | |
| params[:metric][:value] = value | |
| unless distinct_id.nil? | |
| params[:metric][:distinct_id] = distinct_id | |
| end | |
| post(Endpoint.metrics, params) | |
| end | |
| def add_fast_metric(name, value, distinct_id = nil) | |
| params = Hash.new | |
| params[:metric] = Hash.new | |
| params[:metric][:fqdn] = @@fqdn | |
| params[:metric][:name] = name | |
| params[:metric][:value] = value | |
| unless distinct_id.nil? | |
| params[:metric][:distinct_id] = distinct_id | |
| end | |
| post(Endpoint.metrics, params) | |
| end | |
| def add_compound_metric(fqdn, name, values, distinct_id = nil) | |
| params = Hash.new | |
| params[:compound_metric] = Hash.new | |
| params[:compound_metric][:fqdn] = fqdn | |
| params[:compound_metric][:name] = name | |
| params[:compound_metric][:values] = values | |
| unless distinct_id.nil? | |
| params[:compound_metric][:distinct_id] = distinct_id | |
| end | |
| post(Endpoint.compound_metrics, params) | |
| end | |
| def add_fast_compound_metric(name, values, distinct_id = nil) | |
| params = Hash.new | |
| params[:compound_metric] = Hash.new | |
| params[:compound_metric][:fqdn] = @@fqdn | |
| params[:compound_metric][:name] = name | |
| params[:compound_metric][:values] = values | |
| unless distinct_id.nil? | |
| params[:compound_metric][:distinct_id] = distinct_id | |
| end | |
| post(Endpoint.compound_metrics, params) | |
| end | |
| private | |
| def post(endpoint, data=nil) | |
| return false if disabled? | |
| if @@connection.nil? | |
| if ENV["RESTFUL_METRICS_API_KEY"] | |
| @@connection = Connection.new(ENV["RESTFUL_METRICS_API_KEY"]) | |
| else | |
| raise NoConnectionEstablished | |
| end | |
| end | |
| if development_env? # changed to development | |
| logger "Skipping while in development", :info | |
| logger "Data(not send): " << data.inspect, :info | |
| return false | |
| end | |
| if async? | |
| self.delay.transmit endpoint, data | |
| true | |
| else | |
| transmit endpoint, data | |
| end | |
| end | |
| def transmit(endpoint, data) | |
| begin | |
| @@connection.post(endpoint, data) | |
| true | |
| rescue | |
| logger "There was an error communicating with the server" | |
| false | |
| end | |
| end | |
| def extract_options!(args) | |
| if args.last.is_a?(Hash) | |
| return args.pop | |
| else | |
| return {} | |
| end | |
| end | |
| def development_env? # changed to development | |
| return true if defined?(RAILS_ENV) && RAILS_ENV == "development" | |
| return true if defined?(RACK_ENV) && RACK_ENV == "development" | |
| if defined?(Rails.env) && Rails.env == "development" | |
| return true | |
| end | |
| false | |
| end | |
| end | |
| end | |
| end |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment