thrift-client 0.1.0 → 0.3.4

data/test/foobar/foo_service.rb

@@ -0,0 +1,133 @@
+#
+# Autogenerated by Thrift Compiler (0.9.1)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+require 'foobar/foobar_types'
+
+module Test
+  module FooService
+    class Client
+      include ::Thrift::Client
+
+      def getFoo()
+        send_getFoo()
+        return recv_getFoo()
+      end
+
+      def send_getFoo()
+        send_message('getFoo', GetFoo_args)
+      end
+
+      def recv_getFoo()
+        result = receive_message(GetFoo_result)
+        return result.success unless result.success.nil?
+        raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'getFoo failed: unknown result')
+      end
+
+      def putFoo(foo)
+        send_putFoo(foo)
+        return recv_putFoo()
+      end
+
+      def send_putFoo(foo)
+        send_message('putFoo', PutFoo_args, :foo => foo)
+      end
+
+      def recv_putFoo()
+        result = receive_message(PutFoo_result)
+        return result.success unless result.success.nil?
+        raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'putFoo failed: unknown result')
+      end
+
+    end
+
+    class Processor
+      include ::Thrift::Processor
+
+      def process_getFoo(seqid, iprot, oprot)
+        args = read_args(iprot, GetFoo_args)
+        result = GetFoo_result.new()
+        result.success = @handler.getFoo()
+        write_result(result, oprot, 'getFoo', seqid)
+      end
+
+      def process_putFoo(seqid, iprot, oprot)
+        args = read_args(iprot, PutFoo_args)
+        result = PutFoo_result.new()
+        result.success = @handler.putFoo(args.foo)
+        write_result(result, oprot, 'putFoo', seqid)
+      end
+
+    end
+
+    # HELPER FUNCTIONS AND STRUCTURES
+
+    class GetFoo_args
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+
+      FIELDS = {
+
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class GetFoo_result
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      SUCCESS = 0
+
+      FIELDS = {
+        SUCCESS => {:type => ::Thrift::Types::STRUCT, :name => 'success', :class => ::Test::Foo}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class PutFoo_args
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      FOO = 1
+
+      FIELDS = {
+        FOO => {:type => ::Thrift::Types::STRUCT, :name => 'foo', :class => ::Test::Foo}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class PutFoo_result
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      SUCCESS = 0
+
+      FIELDS = {
+        SUCCESS => {:type => ::Thrift::Types::BOOL, :name => 'success'}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+  end
+
+end

data/test/{foobar_constants.rb → foobar/foobar_constants.rb}

@@ -1,11 +1,11 @@
 #
-# Autogenerated by Thrift Compiler (0.9.0)
+# Autogenerated by Thrift Compiler (0.9.1)
 #
 # DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
 #
 
 require 'thrift'
-require 'foobar_types'
+require 'foobar/foobar_types'
 
-module Foo
+module Test
 end

data/test/{foobar_types.rb → foobar/foobar_types.rb}

@@ -1,12 +1,12 @@
 #
-# Autogenerated by Thrift Compiler (0.9.0)
+# Autogenerated by Thrift Compiler (0.9.1)
 #
 # DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
 #
 
 require 'thrift'
 
-module Foo
+module Test
   class Foo
     include ::Thrift::Struct, ::Thrift::Struct_Union
     FIRST = 1
@@ -31,16 +31,18 @@ module Foo
 
   class Bar
     include ::Thrift::Struct, ::Thrift::Struct_Union
-    FOURTH = 1
-    THIRD = 2
-    SECOND = 3
-    FIRST = 4
+    A = 1
+    B = 2
+    C = 3
+    D = 4
+    E = 5
 
     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'}
+      A => {:type => ::Thrift::Types::STRING, :name => 'a'},
+      B => {:type => ::Thrift::Types::I32, :name => 'b'},
+      C => {:type => ::Thrift::Types::BOOL, :name => 'c'},
+      D => {:type => ::Thrift::Types::DOUBLE, :name => 'd'},
+      E => {:type => ::Thrift::Types::I64, :name => 'e'}
     }
 
     def struct_fields; FIELDS; end

data/test/foobar/handler.rb

@@ -0,0 +1,74 @@
+module Test
+  class FooHandler
+    def getFoo
+      foo = Test::Foo.new
+      foo.fourth = "222"
+      foo.third = 2.2
+      foo.second = 222222222222222
+      foo.first = 200000
+      return foo
+    end
+
+    def putFoo(foo)
+      puts foo.inspect
+      return true
+    end
+  end
+
+  class BarHandler
+    def getBar
+      bar = Test::Bar.new
+      bar.a = "1111111111"
+      bar.b = 11111111
+      bar.c = false
+      bar.d = 1111111.232323
+      bar.e = 111111111111111
+      return bar
+    end
+
+    def putBar(bar)
+      puts bar.inspect
+      return true
+    end
+  end
+
+  class CommonHandler
+    def ping
+      puts "ping"
+    end
+
+    def wait(seconds)
+      sleep(seconds / 100.0)
+      return true
+    end
+
+    def timeout
+      sleep(200000)
+      return false
+    end
+
+    def apperror()
+      puts "error"
+      return Thrift::ApplicationException.new("apperr")
+    end
+
+    def ptoerror()
+      puts "error"
+      raise Thrift::ProtocolException.new("ptoerror")
+      return true
+    end
+
+    def tranerror()
+      puts "error"
+      raise Thrift::TransportException.new("tranerror")
+      return true
+    end
+
+    def ioerror()
+      puts "error"
+      raise IOError.new("ioerror")
+      return true
+    end
+
+  end
+end

data/test/helper.rb

@@ -0,0 +1,50 @@
+$:.unshift File.expand_path("../../lib", __FILE__)
+$:.unshift File.expand_path("../../test", __FILE__)
+
+require "thrift_client"
+require 'thrift'
+require 'foobar/common_service'
+require 'foobar/foo_service'
+require 'foobar/bar_service'
+require 'foobar/handler'
+
+module Test
+  module Helper
+    include Thrift
+
+    def self.start_server(host = "127.0.0.1", port = 9000)
+      common_handler = Test::CommonHandler.new
+      common_processor = Test::CommonService::Processor.new(common_handler)
+      foo_handler = Test::FooHandler.new
+      foo_processor = Test::FooService::Processor.new(foo_handler)
+      bar_handler = Test::BarHandler.new
+      bar_processor = Test::BarService::Processor.new(bar_handler)
+      processor = Thrift::MultiplexedProcessor.new
+      processor.register(common_processor.class.name.split("::")[-2].downcase.gsub("service",""), common_processor)
+      processor.register(foo_processor.class.name.split("::")[-2].downcase.gsub("service",""), foo_processor)
+      processor.register(bar_processor.class.name.split("::")[-2].downcase.gsub("service",""), bar_processor)
+      transport = Thrift::ServerSocket.new(host, port)
+      transport_factory = Thrift::FramedTransportFactory.new
+      protocol_factory = Thrift::CompactProtocolFactory.new
+      args = [processor, transport, transport_factory, protocol_factory]
+      server = Thrift::NonblockingServer.new(*args)
+      @server = server
+      sleep(1)
+      @server_thread = Thread.new do
+        @server.serve
+      end
+      sleep(1)
+      puts "server start listen on #{host}:#{port}"
+    end
+
+    def self.shutdown_server
+      return if @server.nil?
+      if @server.respond_to? :shutdown
+        @server.shutdown
+      else
+        @server_thread.kill
+      end
+    end
+
+  end
+end

data/test/passport.rb

@@ -0,0 +1,51 @@
+$:.unshift File.expand_path("../../lib", __FILE__)
+$:.unshift File.expand_path("../../test", __FILE__)
+
+require "test/unit"
+require 'thrift'
+require "thrift_client"
+require "passport_thrift_client"
+require "em-synchrony"
+require 'em-synchrony/fiber_iterator'
+
+class ProfileTest < Test::Unit::TestCase
+  include Test::Unit::Assertions
+  def setup
+    super
+    puts "setup"
+  end
+
+  def test_profile
+    EM.synchrony do
+    	client = ThriftClient.new({"multiplexed" => true, "timeout" => 10, "transport" => "eventmachine","size" => 100,"client_class" => "Passport::Thrift::RemoteUserProfileService::Client","servers" => "vm173:9100"})
+    	# client.queryUserBasicInfo(3033)
+      start = Time.now
+      EM::Synchrony::FiberIterator.new(0...2000, 1000).each do |i|
+        client.queryUserBasicInfo(3033)
+      end
+      stop = Time.now
+      em_use = stop - start
+      puts em_use
+      client.destroy
+      EM.stop
+    end
+
+   #  client = ThriftClient.new({"multiplexed" => true, "timeout" => 10, "transport" => "socket","size"=>1,"client_class" => "Passport::Thrift::RemoteUserProfileService::Client","servers" => "vm173:9100"})
+  	# i = 0
+  	# client.queryUserBasicInfo(3033)
+  	# start = Time.now
+  	# while i < 10000
+  	# 	client.queryUserBasicInfo(3033)
+  	# 	i+=1
+  	# end
+	 	# stop = Time.now
+   #  em_use = stop - start
+   #  puts em_use
+   #  client.destroy
+  end
+
+  def teardown
+    super
+    puts "teardown"
+  end
+end

data/test/start_server.rb

@@ -0,0 +1,8 @@
+$:.unshift File.expand_path("../../lib", __FILE__)
+$:.unshift File.expand_path("../../test", __FILE__)
+
+require "helper"
+port = ARGV[0]
+Test::Helper.start_server("127.0.0.1",port)
+Marshal.load(STDIN)
+

data/test/test.rb

@@ -0,0 +1,27 @@
+class RetryTest
+  def test
+    puts "method begin"
+    try = 0
+    begin
+      puts "try begin"
+      raise "error"
+    rescue Exception => e
+      puts "Exception"
+      if try < 1
+        try += 1
+        retry
+      end
+      puts "after retry"
+    end
+  end
+end
+
+RetryTest.new.test
+
+object = 2
+case object
+when 2
+  puts object
+when 3
+  puts object
+end

data/test/test_client.rb

@@ -0,0 +1,111 @@
+$:.unshift File.expand_path("../../lib", __FILE__)
+# $:.unshift File.expand_path("../../test", __FILE__)
+
+require "test/unit"
+require "thrift_client"
+require "yaml"
+require "feed"
+require 'em-synchrony'
+require 'em-synchrony/fiber_iterator'
+require 'stat-analysis-query'
+
+class ClientTest < Test::Unit::TestCase
+  include Test::Unit::Assertions
+  def setup
+    super
+    root = File.expand_path("../..", __FILE__)
+    @config = YAML.load_file(File.join(root, 'test/thrift_config.yaml'))
+    puts "setup"
+  end
+
+  def test_thread_client
+    client = ThriftClient.new(@config['client_feed'])
+    while 1
+      Thread.new {
+        client.getWebUnread('3428')
+      }
+      sleep(0.01)
+    end
+  end
+
+  def test_fiber_client
+    EM.synchrony do
+      client = ThriftClient.new(@config['client_em_feed'])
+      start = Time.now
+      number = 0
+      while number < 10
+        EM::Synchrony::FiberIterator.new(0...5, 5).each do |i|
+          client.getWebSoundRange('12', 0, 100)
+          puts i
+        end
+        number += 1
+      end
+      stop = Time.now
+      em_use = stop - start
+      EM.stop
+    end
+  end
+
+  def test_thread_return_nil
+    client = ThriftClient.new(@config['client_query'])
+    sleep(3)
+    begin
+      p client.selectWebHumanCategory
+    rescue Exception => e
+      backtrace = ""
+      e.backtrace.each { | s |
+        backtrace.concat(s).concat("\n")
+      }
+      # puts "#{e} #{e.backtrace}"
+      puts "#{e.class} #{e.message} #{backtrace}"
+    end
+    sleep(1)
+    begin
+      p client.selectWebAlbumCategoryAndTag
+    rescue Exception => e
+      puts "#{e} #{e.backtrace}"
+    end
+    begin
+      p client.selectWebAlbumCategoryAndTag
+    rescue Exception => e
+      puts "#{e} #{e.backtrace}"
+    end
+  end
+
+  def test_fiber_return_nil
+    EM.synchrony do
+      client = ThriftClient.new(@config['client_em_query'])
+      start = Time.now
+      EM::Synchrony::FiberIterator.new(0...1, 1).each do |i|
+        begin
+          p client.selectWebHumanCategory
+        rescue Exception => e
+          puts "#{e.class} #{e.message} #{e.backtrace}"
+        end
+        sleep(1)
+        begin
+          p client.selectWebAlbumCategoryAndTag
+        rescue Exception => e
+          backtrace = ""
+          e.backtrace.each { | s |
+            backtrace.concat(s).concat("\n")
+          }
+          puts "#{e} #{e.backtrace}"
+        end
+        begin
+          p client.selectWebAlbumCategoryAndTag
+        rescue Exception => e
+          puts "#{e} #{backtrace}"
+        end
+      end
+      stop = Time.now
+      em_use = stop - start
+      EM.stop
+    end
+  end
+
+  def teardown
+    super
+    puts "teardown"
+  end
+end