thrifter 0.1.0

data/test/statsd_middleware_test.rb

@@ -0,0 +1,99 @@
+require_relative './test_helper'
+
+class StatsdMiddlewareTest < MiniTest::Unit::TestCase
+  attr_reader :rpc
+
+  def setup
+    super
+
+    @rpc = Thrifter::RPC.new(:foo, :args)
+  end
+
+  def test_times_each_rpc
+    app = stub
+    app.stubs(:call).with(rpc)
+
+    statsd = mock
+    statsd.expects(:time).with(rpc.name).yields.returns(:response)
+
+    middleware = Thrifter::StatsdMiddleware.new app, statsd
+    result = middleware.call rpc
+
+    assert :response == result, 'Return value incorrect'
+  end
+
+  def test_counts_transport_exceptions_and_reraises
+    app = stub
+    app.stubs(:call).with(rpc).raises(Thrift::TransportException)
+
+    statsd = mock
+    statsd.expects(:time).yields
+    statsd.expects(:increment).with('errors.transport')
+
+    middleware = Thrifter::StatsdMiddleware.new app, statsd
+
+    assert_raises Thrift::TransportException do
+      middleware.call rpc
+    end
+  end
+
+  def test_counts_protocol_exceptions
+    app = stub
+    app.stubs(:call).with(rpc).raises(Thrift::ProtocolException)
+
+    statsd = mock
+    statsd.expects(:time).yields
+    statsd.expects(:increment).with('errors.protocol')
+
+    middleware = Thrifter::StatsdMiddleware.new app, statsd
+
+    assert_raises Thrift::ProtocolException do
+      middleware.call rpc
+    end
+  end
+
+  def test_counts_application_exceptions
+    app = stub
+    app.stubs(:call).with(rpc).raises(Thrift::ApplicationException)
+
+    statsd = mock
+    statsd.expects(:time).yields
+    statsd.expects(:increment).with('errors.application')
+
+    middleware = Thrifter::StatsdMiddleware.new app, statsd
+
+    assert_raises Thrift::ApplicationException do
+      middleware.call rpc
+    end
+  end
+
+  def test_counts_timeouts
+    app = stub
+    app.stubs(:call).with(rpc).raises(Timeout::Error)
+
+    statsd = mock
+    statsd.expects(:time).yields
+    statsd.expects(:increment).with('errors.timeout')
+
+    middleware = Thrifter::StatsdMiddleware.new app, statsd
+
+    assert_raises Timeout::Error do
+      middleware.call rpc
+    end
+  end
+
+  def test_counts_other_errors
+    app = stub
+    app.stubs(:call).with(rpc).raises(StandardError)
+
+    statsd = mock
+    statsd.expects(:time).yields
+    statsd.expects(:increment).with('errors.other')
+
+    middleware = Thrifter::StatsdMiddleware.new app, statsd
+
+    assert_raises StandardError do
+      middleware.call rpc
+    end
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,45 @@
+require 'bundler/setup'
+require 'thrifter'
+require 'thrifter/queueing'
+
+root = File.expand_path '../..', __FILE__
+$LOAD_PATH << "#{root}/vendor/gen-rb"
+
+require 'test_service'
+
+require 'sidekiq'
+require 'sidekiq/testing'
+require 'minitest/autorun'
+require 'mocha/mini_test'
+
+Sidekiq::Testing.fake!
+
+class FakeTransport
+  def initialize(*)
+
+  end
+
+  def open
+
+  end
+
+  def close
+
+  end
+end
+
+class FakeProtocol
+  def initialize(*)
+
+  end
+end
+
+class NullStatsd
+  def incr(*)
+
+  end
+
+  def time(*)
+    yield
+  end
+end

data/thrifter.gemspec

@@ -0,0 +1,35 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'thrifter/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "thrifter"
+  spec.version       = Thrifter::VERSION
+  spec.authors       = ["ahawkins"]
+  spec.email         = ["adam@hawkins.io"]
+  spec.summary       = %q{Production ready Thrift client with improved semantics}
+  spec.description   = %q{}
+  spec.homepage      = "https://github.com/saltside/thrifter"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "thrift"
+  spec.add_dependency "thrift-base64"
+  spec.add_dependency "statsd-ruby"
+  spec.add_dependency "concord"
+  spec.add_dependency "middleware"
+  spec.add_dependency "connection_pool"
+  spec.add_dependency "tnt"
+
+  spec.add_development_dependency "bundler", "~> 1.7"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "mocha"
+  spec.add_development_dependency "sidekiq"
+  spec.add_development_dependency "sidekiq-thrift_arguments"
+  spec.add_development_dependency "eventmachine"
+end

data/vendor/gen-rb/test_constants.rb

@@ -0,0 +1,9 @@
+#
+# Autogenerated by Thrift Compiler (0.9.2)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+require 'test_types'
+

data/vendor/gen-rb/test_service.rb

@@ -0,0 +1,122 @@
+#
+# Autogenerated by Thrift Compiler (0.9.2)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+require 'test_types'
+
+module TestService
+  class Client
+    include ::Thrift::Client
+
+    def echo(message)
+      send_echo(message)
+      return recv_echo()
+    end
+
+    def send_echo(message)
+      send_message('echo', Echo_args, :message => message)
+    end
+
+    def recv_echo()
+      result = receive_message(Echo_result)
+      return result.success unless result.success.nil?
+      raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'echo failed: unknown result')
+    end
+
+    def onewayEcho(message)
+      send_onewayEcho(message)
+    end
+
+    def send_onewayEcho(message)
+      send_oneway_message('onewayEcho', OnewayEcho_args, :message => message)
+    end
+  end
+
+  class Processor
+    include ::Thrift::Processor
+
+    def process_echo(seqid, iprot, oprot)
+      args = read_args(iprot, Echo_args)
+      result = Echo_result.new()
+      result.success = @handler.echo(args.message)
+      write_result(result, oprot, 'echo', seqid)
+    end
+
+    def process_onewayEcho(seqid, iprot, oprot)
+      args = read_args(iprot, OnewayEcho_args)
+      @handler.onewayEcho(args.message)
+      return
+    end
+
+  end
+
+  # HELPER FUNCTIONS AND STRUCTURES
+
+  class Echo_args
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+    MESSAGE = 1
+
+    FIELDS = {
+      MESSAGE => {:type => ::Thrift::Types::STRUCT, :name => 'message', :class => ::TestMessage}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class Echo_result
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+    SUCCESS = 0
+
+    FIELDS = {
+      SUCCESS => {:type => ::Thrift::Types::STRUCT, :name => 'success', :class => ::TestMessage}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class OnewayEcho_args
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+    MESSAGE = 1
+
+    FIELDS = {
+      MESSAGE => {:type => ::Thrift::Types::STRUCT, :name => 'message', :class => ::TestMessage}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class OnewayEcho_result
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+
+    FIELDS = {
+
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+end
+

data/vendor/gen-rb/test_types.rb

@@ -0,0 +1,25 @@
+#
+# Autogenerated by Thrift Compiler (0.9.2)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+
+class TestMessage
+  include ::Thrift::Struct, ::Thrift::Struct_Union
+  MESSAGE = 1
+
+  FIELDS = {
+    MESSAGE => {:type => ::Thrift::Types::STRING, :name => 'message'}
+  }
+
+  def struct_fields; FIELDS; end
+
+  def validate
+    raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field message is unset!') unless @message
+  end
+
+  ::Thrift::Struct.generate_accessors self
+end
+

metadata

@@ -0,0 +1,264 @@
+--- !ruby/object:Gem::Specification
+name: thrifter
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- ahawkins
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-02-01 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thrift
+  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'
+- !ruby/object:Gem::Dependency
+  name: thrift-base64
+  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'
+- !ruby/object:Gem::Dependency
+  name: statsd-ruby
+  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'
+- !ruby/object:Gem::Dependency
+  name: concord
+  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'
+- !ruby/object:Gem::Dependency
+  name: middleware
+  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'
+- !ruby/object:Gem::Dependency
+  name: connection_pool
+  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'
+- !ruby/object:Gem::Dependency
+  name: tnt
+  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'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: sidekiq
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: sidekiq-thrift_arguments
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: eventmachine
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: ''
+email:
+- adam@hawkins.io
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".dockerignore"
+- ".gitignore"
+- Dockerfile
+- Gemfile
+- LICENSE.txt
+- Makefile
+- README.md
+- Rakefile
+- Vagrantfile
+- circle.yml
+- lib/thrifter.rb
+- lib/thrifter/error_wrapping_middleware.rb
+- lib/thrifter/ping.rb
+- lib/thrifter/queueing.rb
+- lib/thrifter/retry.rb
+- lib/thrifter/statsd_middleware.rb
+- lib/thrifter/version.rb
+- script/monkey-client
+- script/server
+- test.thrift
+- test/acceptance_test.rb
+- test/error_wrapping_middleware_test.rb
+- test/ping_test.rb
+- test/queuing_test.rb
+- test/retry_test.rb
+- test/statsd_middleware_test.rb
+- test/test_helper.rb
+- thrifter.gemspec
+- vendor/gen-rb/test_constants.rb
+- vendor/gen-rb/test_service.rb
+- vendor/gen-rb/test_types.rb
+homepage: https://github.com/saltside/thrifter
+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: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Production ready Thrift client with improved semantics
+test_files:
+- test/acceptance_test.rb
+- test/error_wrapping_middleware_test.rb
+- test/ping_test.rb
+- test/queuing_test.rb
+- test/retry_test.rb
+- test/statsd_middleware_test.rb
+- test/test_helper.rb