thrift-client 0.0.7

data/test/foobar_types.rb

@@ -0,0 +1,54 @@
+#
+# Autogenerated by Thrift Compiler (0.9.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+
+module Foo
+  class Foo
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+    FIRST = 1
+    SECOND = 2
+    THIRD = 3
+    FOURTH = 4
+
+    FIELDS = {
+      FIRST => {:type => ::Thrift::Types::I32, :name => 'first'},
+      SECOND => {:type => ::Thrift::Types::I64, :name => 'second'},
+      THIRD => {:type => ::Thrift::Types::DOUBLE, :name => 'third'},
+      FOURTH => {:type => ::Thrift::Types::STRING, :name => 'fourth'}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class Bar
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+    FOURTH = 1
+    THIRD = 2
+    SECOND = 3
+    FIRST = 4
+
+    FIELDS = {
+      FOURTH => {:type => ::Thrift::Types::STRING, :name => 'fourth'},
+      THIRD => {:type => ::Thrift::Types::DOUBLE, :name => 'third'},
+      SECOND => {:type => ::Thrift::Types::I64, :name => 'second'},
+      FIRST => {:type => ::Thrift::Types::I32, :name => 'first'}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+end

data/test/multiplexed_protocol_test.rb

@@ -0,0 +1,53 @@
+$:.unshift File.expand_path("../../lib", __FILE__)
+$:.unshift File.expand_path("../../test", __FILE__)
+
+require "test/unit"
+require "thrift_client"
+require "foobar_service"
+require 'foobar_types'
+require 'thrift'
+# require 'yaml'
+
+class ClientTest < Test::Unit::TestCase
+  include Test::Unit::Assertions
+  def setup
+    super
+    transport = Thrift::Socket.new('127.0.0.1', 9999)
+    transport = Thrift::FramedTransport.new(transport)
+    protocol = Thrift::BinaryProtocol.new(transport)
+    protocol = Thrift::MultiplexedProtocol.new(protocol,"foo")
+    @client = Foo::FoobarService::Client.new(protocol)
+    transport.open
+    puts "setup"
+  end
+ 
+  def test_multiprotocol
+    puts "test start"
+
+    @client.ping
+    
+
+    assert_nothing_raised do @client.ping
+    end
+
+    bar = @client.getBar
+    assert_not_nil(bar)
+
+    foo = Foo::Foo.new
+    foo.first = 1
+    foo.second = 2
+    foo.third = 3.0
+    foo.fourth = "4"
+    assert(@client.putFoo(foo))
+    
+    assert_nothing_raised do @client.ping
+    end
+
+    puts "test stop"
+  end
+
+  def teardown
+    super
+    puts "teardown"
+  end
+end

data/test/multiplexed_server.rb

@@ -0,0 +1,102 @@
+$:.unshift File.expand_path("../../test", __FILE__)
+$:.unshift File.expand_path("../../lib", __FILE__)
+require 'thrift'
+require 'foobar_service'
+require 'foobar_types'
+require 'thrift_client'
+
+module Server
+  include Thrift
+
+  class FooHandler
+
+    def ping
+      puts "ping"
+    end
+
+    def getBar
+      bar = Foo::Bar.new
+      bar.fourth = "4"
+      bar.third = 3.0
+      bar.second = 2
+      bar.first = 1
+      return bar
+    end
+
+    def putFoo(foo)
+      puts foo.inspect
+      return true
+    end
+
+    def getTimeout
+      sleep(2)
+      return false
+    end
+
+    def getError()
+      puts "error"
+      raise ApplicationException.new("Exception")
+      return true
+    end
+
+  end
+
+  def self.start_server(host, port, serverClass)
+    handler = FooHandler.new
+    foo_processor = Foo::FoobarService::Processor.new(handler)
+    puts foo_processor.class.name.split("::")[-2].downcase.gsub("service","")
+    multi_processor = Thrift::MultiplexedProcessor.new
+    multi_processor.register(foo_processor.class.name.split("::")[-2], foo_processor)
+    transport = ServerSocket.new(host, port)
+    # transport = FramedTransport.new(transport)
+    transport_factory = FramedTransportFactory.new
+    args = [multi_processor, transport, transport_factory]
+    if serverClass == NonblockingServer
+      # logger = Logger.new(STDOUT)
+      # logger.level = Logger::WARN
+      # args << logger
+    end
+    server = serverClass.new(*args)
+    @server_thread = Thread.new do
+      server.serve
+    end
+    @server = server
+  end
+
+  def self.shutdown
+    return if @server.nil?
+    if @server.respond_to? :shutdown
+      @server.shutdown
+    else
+      @server_thread.kill
+    end
+  end
+end
+
+def resolve_const(const)
+  const and const.split('::').inject(Object) { |k,c| k.const_get(c) }
+end
+
+host = '127.0.0.1'
+port = ARGV[0] || 9999
+
+#serverklass = 'Thrift::SimpleServer'
+#serverklass = 'Thrift::ThreadedServer'
+serverklass = 'Thrift::ThreadPoolServer'
+# serverklass = 'Thrift::NonblockingServer'
+
+Server.start_server(host, port.to_i, resolve_const(serverklass))
+# let our host know that the interpreter has started
+# ideally we'd wait until the server was serving, but we don't have a hook for that
+
+puts "multiplexed server start on #{host}:#{port}"
+puts ""
+
+# STDOUT.flush
+
+Marshal.load(STDIN) # wait until we're instructed to shut down
+
+Server.shutdown
+
+puts "server shutdown"
+

data/test/options_config_test.rb

@@ -0,0 +1,86 @@
+$:.unshift File.expand_path("../../lib", __FILE__)
+$:.unshift File.expand_path("../../test", __FILE__)
+
+require "test/unit"
+require "thrift_client"
+require "thrift"
+
+
+DISCONNECT_ERRORS = [
+    IOError,
+    Thrift::ProtocolException,
+    Thrift::TransportException
+  ]
+APPLICATION_ERRORS =[
+    Thrift::ApplicationException
+  ]
+DEFAULTS = {
+    :protocol => Thrift::CompactProtocol,
+    :transport => Thrift::Socket,
+    :transport_wrapper => Thrift::FramedTransport,
+    :disconnect_exception_classes => nil,
+    :application_exception_classes => nil,
+    :size => 1,
+    :timeout => nil,
+    :client_class => nil,
+    :test_on_borrow => true,
+    :pool_timeout => 5,
+    :before_method => [],
+    :after_method => [],
+    :on_exception => []
+  }
+config = {
+    "protocol" => "Thrift::CompactProtocol",
+    "transport" => "Thrift::Socket",
+    "transport_wrapper" => "Thrift::FramedTransport",
+    "disconnect_exception_classes" => "NameError,ArgumentError,StandardError,IOError",
+    "application_exception_classes" => "IOError,,",
+    "size" => nil,
+    "timeout" => nil,
+    "client_class" => nil,
+    "test_on_borrow" => "true",
+    "pool_timeout" => "5",
+    "before_method" => [],
+    "after_method" => [],
+    "on_exception" => []
+  }
+
+user_options = {}
+user_options.store(:protocol,eval(config["protocol"]))
+user_options.store(:transport,eval(config["transport"]))
+user_options.store(:transport_wrapper,eval(config["transport_wrapper"]))
+
+user_define_disconnect_exceptions = config["disconnect_exception_classes"].split(",")
+# puts user_define_exceptions
+user_define_disconnect_exceptions.each do |exception|
+    DISCONNECT_ERRORS.push(eval(exception))
+end
+DISCONNECT_ERRORS.uniq!
+DISCONNECT_ERRORS.each do |error|
+    # puts error.class
+    # puts error
+end
+user_options.store(:disconnect_exception_classes, DISCONNECT_ERRORS)
+
+user_define_application_exceptions = config["application_exception_classes"].split(",")
+puts user_define_application_exceptions
+user_define_application_exceptions.each do |exception|
+    # puts exception
+    APPLICATION_ERRORS.push(eval(exception))
+end
+APPLICATION_ERRORS.uniq!
+APPLICATION_ERRORS.each do |error|
+    # puts error.class
+    # puts error
+end
+user_options.store(:application_exception_classes, APPLICATION_ERRORS)
+
+if config["size"] != nil
+    user_options.store(:size, config["size"].to_i)
+end
+
+servers = "192.168.3.111:8888,192.168.2.111:8888".split(",")
+servers.each do |server|
+    puts server
+    puts server.class
+end

data/test/server.rb

@@ -0,0 +1,97 @@
+$:.unshift File.expand_path("../../test", __FILE__)
+require 'thrift'
+require 'foobar_service'
+require 'foobar_types'
+
+module Server
+  include Thrift
+
+  class FooHandler
+
+    def ping
+      puts "ping"
+    end
+
+    def getBar
+      bar = Foo::Bar.new
+      bar.fourth = "4"
+      bar.third = 3.0
+      bar.second = 2
+      bar.first = 1
+      return bar
+    end
+
+    def putFoo(foo)
+      puts foo.inspect
+      return true
+    end
+
+    def getTimeout
+      sleep(2)
+      return false
+    end
+
+    def getError()
+      puts "error"
+      raise ApplicationException.new("Exception")
+      return true
+    end
+
+  end
+
+  def self.start_server(host, port, serverClass)
+    handler = FooHandler.new
+    processor = Foo::FoobarService::Processor.new(handler)
+    transport = ServerSocket.new(host, port)
+    # transport = FramedTransport.new(transport)
+    transport_factory = FramedTransportFactory.new
+    args = [processor, transport, transport_factory]
+    if serverClass == NonblockingServer
+      # logger = Logger.new(STDOUT)
+      # logger.level = Logger::WARN
+      # args << logger
+    end
+    server = serverClass.new(*args)
+    @server_thread = Thread.new do
+      server.serve
+    end
+    @server = server
+  end
+
+  def self.shutdown
+    return if @server.nil?
+    if @server.respond_to? :shutdown
+      @server.shutdown
+    else
+      @server_thread.kill
+    end
+  end
+end
+
+def resolve_const(const)
+  const and const.split('::').inject(Object) { |k,c| k.const_get(c) }
+end
+
+host = '127.0.0.1'
+port = ARGV[0] || 9999
+
+#serverklass = 'Thrift::SimpleServer'
+#serverklass = 'Thrift::ThreadedServer'
+serverklass = 'Thrift::ThreadPoolServer'
+# serverklass = 'Thrift::NonblockingServer'
+
+Server.start_server(host, port.to_i, resolve_const(serverklass))
+# let our host know that the interpreter has started
+# ideally we'd wait until the server was serving, but we don't have a hook for that
+
+puts "server start on #{host}:#{port}"
+puts ""
+
+# STDOUT.flush
+
+Marshal.load(STDIN) # wait until we're instructed to shut down
+
+Server.shutdown
+
+puts "server shutdown"
+

metadata

@@ -0,0 +1,79 @@
+--- !ruby/object:Gem::Specification
+name: thrift-client
+version: !ruby/object:Gem::Version
+  version: 0.0.7
+platform: ruby
+authors:
+- Ted Wang
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-09-24 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thrift
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.0
+description: Thrift client
+email: ted@ximalaya.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/thrift_client/abstract_server.rb
+- lib/thrift_client/abstract_thrift_client.rb
+- lib/thrift_client/multi_client_server.rb
+- lib/thrift_client/single_client_server.rb
+- lib/thrift_client/thrift.rb
+- lib/thrift_client.rb
+- test/client_test.rb
+- test/foobar_constants.rb
+- test/foobar_service.rb
+- test/foobar_types.rb
+- test/multiplexed_protocol_test.rb
+- test/multiplexed_server.rb
+- test/options_config_test.rb
+- test/server.rb
+homepage: http://www.ximalaya.com
+licenses:
+- MIT2.0
+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.0.3
+signing_key: 
+specification_version: 4
+summary: A Thrift client wrapper that encapsulates some common behavior.
+test_files:
+- test/client_test.rb
+- test/foobar_constants.rb
+- test/foobar_service.rb
+- test/foobar_types.rb
+- test/multiplexed_protocol_test.rb
+- test/multiplexed_server.rb
+- test/options_config_test.rb
+- test/server.rb