cube-tapjoy 0.0.3

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in cube-ruby.gemspec
+gemspec

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Cody Krieger
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,87 @@
+# cube-ruby
+
+A Cube client for Ruby (http://square.github.com/cube). Heavily based on
+[this statsd ruby client][sdc].
+
+MIT licensed. See [the LICENSE file][license] for more details.
+for more details.
+
+## Installation
+
+Since Cube is under constant development, it's best if you specify a particular
+version of cube-ruby in your `Gemfile`. That way, you can ensure your Ruby
+application and your Cube instance will be compatible with each other.
+
+Here's a version compatibility table:
+
+    Cube   | cube-ruby
+    --------------------
+    0.2.0  | 0.0.1
+    0.2.1  | 0.0.2 (master)
+
+Add this line to your application's `Gemfile`:
+
+    gem 'cube-ruby', require: "cube"
+
+    # or specify a particular version:
+    gem 'cube-ruby', '0.0.2', require: "cube"
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install cube-ruby
+
+## Usage
+
+### Set up a global Cube client
+
+```ruby
+# use default hostname and port of localhost:1180
+$cube = Cube::Client.new
+
+# use custom hostname and port
+$cube = Cube::Client.new 'cube.example.org', 2280
+```
+
+### Send Cube some metrics!
+
+```ruby
+# send a new "foo" event to cube
+$cube.send "foo"
+
+# send a new event to cube that looks like this:
+# { type: "request", data: { value: "somevalue" } }
+$cube.send "request", value: "somevalue"
+
+# optionally specify a specific date/time (two days ago)
+$cube.send "request", DateTime.now - 2, value: "othervalue"
+
+# specify an event id (https://github.com/square/cube/wiki/Events)
+event_id = 42
+$cube.send "request", DateTime.now, event_id, duration_ms: 234
+```
+
+Yes, the method is called `send`. You can still call `Object#send` on
+`Cube::Client` objects by using the `__send__` method, [per the Ruby docs][rd].
+
+## Testing
+
+Run the specs with `rake`.
+
+To include real UDP socket testing in the specs, run `LIVE=true rake`.
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request
+
+[sdc]: https://github.com/github/statsd-ruby
+[license]: https://github.com/codykrieger/cube-ruby/blob/master/LICENSE
+[rd]: http://ruby-doc.org/core-1.9.3/Object.html#method-i-send
+

data/Rakefile

@@ -0,0 +1,12 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"
+
+require 'rake/testtask'
+Rake::TestTask.new(:spec) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.pattern = 'spec/**/*_spec.rb'
+  spec.verbose = true
+end
+
+task :default => :spec
+

data/cube-tapjoy.gemspec

@@ -0,0 +1,19 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/cube/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Cody Krieger"]
+  gem.email         = ["cody@codykrieger.com"]
+  gem.description   = %q{A Cube client for Ruby (http://square.github.com/cube).}
+  gem.summary       = %q{A Cube client for Ruby (http://square.github.com/cube).}
+  gem.homepage      = "https://github.com/Tapjoy/cube-ruby"
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "cube-tapjoy"
+  gem.require_paths = ["lib"]
+  gem.version       = Cube::VERSION
+
+  gem.add_development_dependency "minitest"
+end

data/lib/cube.rb

@@ -0,0 +1,95 @@
+require 'cube/version'
+
+module Cube
+  require 'date'
+  # We'll be sending data to Cube over a UDP socket.
+  require 'socket'
+  # Cube requires data to be in JSON format.
+  require 'json'
+
+  class Client
+    # A namespace to prepend to all Cube calls.
+    attr_accessor :namespace
+
+    # We'll use this to eliminate any unwanted/disallowed characters from our
+    # event type later on.
+    RESERVED_CHARS_REGEX = /[^\w\d]/
+
+    class << self
+      # Set to any logger instance that responds to #debug and #error (like the
+      # Rails or stdlib logger) to enable metric logging.
+      attr_accessor :logger
+    end
+
+    # Set 'er up with a host and port, defaults to `localhost:1180`.
+    #
+    # @param [String] The hostname to send metrics to.
+    # @param [Integer] The UDP port to send metrics to.
+    def initialize(host="localhost", port=1180)
+      @host, @port = host, port
+    end
+
+    # The primary endpoint for sending metrics to Cube.
+    #
+    # @param [String] The desired name of the new Cube event.
+    # @param [Array] A splat that takes an optional DateTime, an event id
+    #   (typically an integer, but can be any object), and a Hash of data.
+    def send(type, *args)
+      time = nil
+      id = nil
+      data = nil
+
+      until args.empty?
+        arg = args.shift
+
+        if arg.is_a? DateTime
+          time ||= arg
+        elsif arg.is_a? Hash
+          data ||= arg
+        else
+          id ||= arg
+        end
+      end
+
+      # Send off our parsed arguments to be further massaged and socketized.
+      actual_send type, time, id, data
+    end
+
+    private
+
+    # Actually send the given data to a socket, and potentially log it along
+    # the way.
+    #
+    # @param [String] The desired name of the new Cube event.
+    # @param [DateTime] Optional. A specific time when the Cube event occurred.
+    # @param [Object] Optional. Typically an integer, but can be any object.
+    # @param [Hash] Optional. Anything in this hash will be stored in the
+    #   `data` subdocument of the Cube event.
+    def actual_send(type, time, id, data)
+      # Namespace support!
+      prefix = "#{@namespace}_" unless @namespace.nil?
+      # Get rid of any unwanted characters, and replace each of them with an _.
+      type = type.to_s.gsub RESERVED_CHARS_REGEX, '_'
+
+      # Start constructing the message to be sent to Cube over UDP.
+      message = {
+        :type => "#{prefix}#{type}"
+      }
+      message[:time] = time.iso8601 unless time.nil?
+      message[:id] = id unless id.nil?
+      message[:data] = data unless data.nil?
+
+      # JSONify it, log it, and send it off.
+      message_str = message.to_json
+      self.class.logger.debug { "Cube: #{message_str}" } if self.class.logger
+
+      socket.send message_str, 0, @host, @port
+    rescue => err
+      self.class.logger.error { "Cube: #{err.class} #{err}" } if self.class.logger
+    end
+
+    # Helper for getting the socket. `@socket` can be set to a mock object to
+    # test without needing an actual UDP socket.
+    def socket ; @socket ||= UDPSocket.new ; end
+  end
+end

data/lib/cube/version.rb

@@ -0,0 +1,3 @@
+module Cube
+  VERSION = "0.0.3"
+end

data/realtest.rb

@@ -0,0 +1,6 @@
+$:.unshift File.join(File.dirname(__FILE__), 'lib')
+require 'cube'
+
+cube = Cube::Client.new
+cube.send "testorz"
+

data/spec/cube_spec.rb

@@ -0,0 +1,133 @@
+require 'helper'
+require 'date'
+require 'json'
+
+# a few things yanked verbatim from here: https://github.com/github/statsd-ruby/blob/master/spec/statsd_spec.rb
+# license: https://github.com/github/statsd-ruby/blob/master/LICENSE.txt
+
+describe Cube::Client do
+  before do
+    @cube = Cube::Client.new "cube.example.org", 1185
+    class << @cube
+      attr_reader :host, :port
+      def socket ; @socket ||= FakeUDPSocket.new ; end
+    end
+  end
+
+  after { @cube.socket.clear }
+
+  describe "#initialize" do
+    it "should set the host and port" do
+      @cube.host.must_equal "cube.example.org"
+      @cube.port.must_equal 1185
+    end
+
+    it "should default the host and port to localhost:1180" do
+      cube = Cube::Client.new
+      cube.instance_variable_get('@host').must_equal 'localhost'
+      cube.instance_variable_get('@port').must_equal 1180
+    end
+  end
+
+  describe "#send" do
+    it "should format the message according to the Cube spec" do
+      type = "request"
+      @cube.send type
+      recv = JSON.parse @cube.socket.recv.first
+      recv["type"].must_equal "request"
+    end
+
+    describe "with a time" do
+      it "should format the message according to the Cube spec" do
+        type = "request"
+        time = DateTime.now
+        @cube.send type, time
+        recv = JSON.parse @cube.socket.recv.first
+        recv["type"].must_equal "request"
+        recv["time"].must_equal time.iso8601
+      end
+    end
+
+    describe "with an id" do
+      it "should format the message according to the Cube spec" do
+        type = "request"
+        time = DateTime.now
+        id = 42
+        data = { :duration_ms => 234 }
+        @cube.send type, time, id, data
+        recv = JSON.parse @cube.socket.recv.first
+        recv["type"].must_equal "request"
+        recv["time"].must_equal time.iso8601
+        recv["id"].must_equal 42
+        recv["data"].must_equal({ "duration_ms" => 234 })
+      end
+    end
+
+    describe "with data" do
+      it "should format the message according to the Cube spec" do
+        type = "request"
+        time = DateTime.now
+        id = 42
+        data = { :duration_ms => 234 }
+        @cube.send type, time, id, data
+        recv = JSON.parse @cube.socket.recv.first
+        recv["type"].must_equal "request"
+        recv["time"].must_equal time.iso8601
+        recv["id"].must_equal 42
+        recv["data"].must_equal({ "duration_ms" => 234 })
+      end
+    end
+  end
+
+  describe "with namespace" do
+    before { @cube.namespace = 'someservice' }
+
+    it "should add namespace to send" do
+      @cube.send "request"
+      recv = JSON.parse @cube.socket.recv.first
+      recv["type"].must_equal 'someservice_request'
+    end
+  end
+
+  describe "with logging" do
+    require 'stringio'
+    before { Cube::Client.logger = Logger.new(@log = StringIO.new) }
+
+    it "should write to the log in debug" do
+      Cube::Client.logger.level = Logger::DEBUG
+
+      @cube.send "request"
+
+      recv = @cube.socket.recv.first
+      @log.string.must_match "Cube: #{recv}"
+    end
+
+    it "should not write to the log unless debug" do
+      Cube::Client.logger.level = Logger::INFO
+
+      @cube.send "request"
+
+      recv = @cube.socket.recv.first
+      @log.string.must_be_empty
+    end
+  end
+end
+
+describe Cube::Client do
+  describe "with a real UDP socket" do
+    it "should actually send stuff over the socket" do
+      socket = UDPSocket.new
+      host, port = 'localhost', 12345
+      socket.bind host, port
+
+      time = DateTime.now
+      cube = Cube::Client.new host, port
+      cube.send "request", time
+
+      recv = JSON.parse socket.recvfrom(64).first
+      recv["type"].must_equal "request"
+      recv["time"].must_equal time.iso8601
+    end
+  end
+end if ENV['LIVE']
+

data/spec/helper.rb

@@ -0,0 +1,37 @@
+require 'minitest/autorun'
+
+# yanked almost verbatim from here: https://github.com/github/statsd-ruby/blob/master/spec/helper.rb
+# license: https://github.com/github/statsd-ruby/blob/master/LICENSE.txt
+
+$:.unshift File.dirname(__FILE__)
+$:.unshift File.join(File.dirname(__FILE__), '..', 'lib')
+
+require 'cube'
+require 'logger'
+
+class FakeUDPSocket
+  def initialize
+    @buffer = []
+  end
+
+  def send(message, *args)
+    @buffer.push [message]
+  end
+
+  def recv
+    res = @buffer.shift
+  end
+
+  def clear
+    @buffer = []
+  end
+
+  def to_s
+    inspect
+  end
+
+  def inspect
+    "<FakeUDPSocket: #{@buffer.inspect}>"
+  end
+end
+

metadata

@@ -0,0 +1,92 @@
+--- !ruby/object:Gem::Specification 
+name: cube-tapjoy
+version: !ruby/object:Gem::Version 
+  hash: 25
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 3
+  version: 0.0.3
+platform: ruby
+authors: 
+- Cody Krieger
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-05-08 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: minitest
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id001
+description: A Cube client for Ruby (http://square.github.com/cube).
+email: 
+- cody@codykrieger.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- cube-tapjoy.gemspec
+- lib/cube.rb
+- lib/cube/version.rb
+- realtest.rb
+- spec/cube_spec.rb
+- spec/helper.rb
+has_rdoc: true
+homepage: https://github.com/Tapjoy/cube-ruby
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: A Cube client for Ruby (http://square.github.com/cube).
+test_files: 
+- spec/cube_spec.rb
+- spec/helper.rb