garru-ruby_scribe_client 0.0.2

data/lib/ruby_scribe_client/reflection_limited_types.rb

@@ -0,0 +1,148 @@
+#
+# Autogenerated by Thrift
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+module TTypeTag
+  T_VOID = 1
+  T_BOOL = 2
+  T_BYTE = 3
+  T_I16 = 6
+  T_I32 = 8
+  T_I64 = 10
+  T_DOUBLE = 4
+  T_STRING = 11
+  T_STRUCT = 12
+  T_MAP = 13
+  T_SET = 14
+  T_LIST = 15
+  T_ENUM = 101
+  T_NOT_REFLECTED = 102
+  VALID_VALUES = Set.new([T_VOID, T_BOOL, T_BYTE, T_I16, T_I32, T_I64, T_DOUBLE, T_STRING, T_STRUCT, T_MAP, T_SET, T_LIST, T_ENUM, T_NOT_REFLECTED]).freeze
+end
+
+class SimpleType
+  include ::Thrift::Struct
+  TTYPE = 1
+  NAME = 2
+
+  Thrift::Struct.field_accessor self, :ttype, :name
+  FIELDS = {
+    TTYPE => {:type => Thrift::Types::I32, :name => 'ttype'},
+    NAME => {:type => Thrift::Types::STRING, :name => 'name'}
+  }
+
+  def struct_fields; FIELDS; end
+
+  def validate
+    unless @ttype.nil? || TTypeTag::VALID_VALUES.include?(@ttype)
+      raise Thrift::ProtocolException.new(Thrift::ProtocolException::UNKNOWN, 'Invalid value of field ttype!')
+    end
+  end
+
+end
+
+class ContainerType
+  include ::Thrift::Struct
+  TTYPE = 1
+  SUBTYPE1 = 2
+  SUBTYPE2 = 3
+
+  Thrift::Struct.field_accessor self, :ttype, :subtype1, :subtype2
+  FIELDS = {
+    TTYPE => {:type => Thrift::Types::I32, :name => 'ttype'},
+    SUBTYPE1 => {:type => Thrift::Types::STRUCT, :name => 'subtype1', :class => SimpleType},
+    SUBTYPE2 => {:type => Thrift::Types::STRUCT, :name => 'subtype2', :class => SimpleType, :optional => true}
+  }
+
+  def struct_fields; FIELDS; end
+
+  def validate
+    unless @ttype.nil? || TTypeTag::VALID_VALUES.include?(@ttype)
+      raise Thrift::ProtocolException.new(Thrift::ProtocolException::UNKNOWN, 'Invalid value of field ttype!')
+    end
+  end
+
+end
+
+class ThriftType
+  include ::Thrift::Struct
+  IS_CONTAINER = 1
+  SIMPLE_TYPE = 2
+  CONTAINER_TYPE = 3
+
+  Thrift::Struct.field_accessor self, :is_container, :simple_type, :container_type
+  FIELDS = {
+    IS_CONTAINER => {:type => Thrift::Types::BOOL, :name => 'is_container'},
+    SIMPLE_TYPE => {:type => Thrift::Types::STRUCT, :name => 'simple_type', :class => SimpleType, :optional => true},
+    CONTAINER_TYPE => {:type => Thrift::Types::STRUCT, :name => 'container_type', :class => ContainerType, :optional => true}
+  }
+
+  def struct_fields; FIELDS; end
+
+  def validate
+  end
+
+end
+
+class Argument
+  include ::Thrift::Struct
+  KEY = 1
+  NAME = 2
+  TYPE = 3
+
+  Thrift::Struct.field_accessor self, :key, :name, :type
+  FIELDS = {
+    KEY => {:type => Thrift::Types::I16, :name => 'key'},
+    NAME => {:type => Thrift::Types::STRING, :name => 'name'},
+    TYPE => {:type => Thrift::Types::STRUCT, :name => 'type', :class => ThriftType}
+  }
+
+  def struct_fields; FIELDS; end
+
+  def validate
+  end
+
+end
+
+class Method
+  include ::Thrift::Struct
+  NAME = 1
+  RETURN_TYPE = 2
+  ARGUMENTS = 3
+
+  Thrift::Struct.field_accessor self, :name, :return_type, :arguments
+  FIELDS = {
+    NAME => {:type => Thrift::Types::STRING, :name => 'name'},
+    RETURN_TYPE => {:type => Thrift::Types::STRUCT, :name => 'return_type', :class => ThriftType},
+    ARGUMENTS => {:type => Thrift::Types::LIST, :name => 'arguments', :element => {:type => Thrift::Types::STRUCT, :class => Argument}}
+  }
+
+  def struct_fields; FIELDS; end
+
+  def validate
+  end
+
+end
+
+class Service
+  include ::Thrift::Struct
+  NAME = 1
+  METHODS = 2
+  FULLY_REFLECTED = 3
+
+  Thrift::Struct.field_accessor self, :name, :methods, :fully_reflected
+  FIELDS = {
+    NAME => {:type => Thrift::Types::STRING, :name => 'name'},
+    METHODS => {:type => Thrift::Types::LIST, :name => 'methods', :element => {:type => Thrift::Types::STRUCT, :class => Method}},
+    FULLY_REFLECTED => {:type => Thrift::Types::BOOL, :name => 'fully_reflected'}
+  }
+
+  def struct_fields; FIELDS; end
+
+  def validate
+  end
+
+end
+

data/lib/ruby_scribe_client/scribe.rb

@@ -0,0 +1,82 @@
+#
+# Autogenerated by Thrift
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require File.join([File.dirname(__FILE__), '..', '..', 'vendor','thrift'])
+require File.join([File.dirname(__FILE__) ,'FacebookService'])
+require File.join([File.dirname(__FILE__), 'scribe_types'])
+
+module Scribe
+  class Client < FacebookService::Client 
+    include ::Thrift::Client
+
+    def Log(messages)
+      send_Log(messages)
+      return recv_Log()
+    end
+
+    def send_Log(messages)
+      send_message('Log', Log_args, :messages => messages)
+    end
+
+    def recv_Log()
+      result = receive_message(Log_result)
+      return result.success unless result.success.nil?
+      raise Thrift::ApplicationException.new(Thrift::ApplicationException::MISSING_RESULT, 'Log failed: unknown result')
+    end
+
+  end
+
+  class Processor < FacebookService::Processor 
+    include ::Thrift::Processor
+
+    def process_Log(seqid, iprot, oprot)
+      args = read_args(iprot, Log_args)
+      result = Log_result.new()
+      result.success = @handler.Log(args.messages)
+      write_result(result, oprot, 'Log', seqid)
+    end
+
+  end
+
+  # HELPER FUNCTIONS AND STRUCTURES
+
+  class Log_args
+    include ::Thrift::Struct
+    MESSAGES = 1
+
+    Thrift::Struct.field_accessor self, :messages
+    FIELDS = {
+      MESSAGES => {:type => Thrift::Types::LIST, :name => 'messages', :element => {:type => Thrift::Types::STRUCT, :class => LogEntry}}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+  end
+
+  class Log_result
+    include ::Thrift::Struct
+    SUCCESS = 0
+
+    Thrift::Struct.field_accessor self, :success
+    FIELDS = {
+      SUCCESS => {:type => Thrift::Types::I32, :name => 'success'}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+      unless @success.nil? || ResultCode::VALID_VALUES.include?(@success)
+        raise Thrift::ProtocolException.new(Thrift::ProtocolException::UNKNOWN, 'Invalid value of field success!')
+      end
+    end
+
+  end
+
+end
+

data/lib/ruby_scribe_client/scribe_types.rb

@@ -0,0 +1,32 @@
+#
+# Autogenerated by Thrift
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+require File.dirname(__FILE__) + '/fb303_types'
+
+
+module ResultCode
+  OK = 0
+  TRY_LATER = 1
+  VALID_VALUES = Set.new([OK, TRY_LATER]).freeze
+end
+
+class LogEntry
+  include ::Thrift::Struct
+  CATEGORY = 1
+  MESSAGE = 2
+
+  Thrift::Struct.field_accessor self, :category, :message
+  FIELDS = {
+    CATEGORY => {:type => Thrift::Types::STRING, :name => 'category'},
+    MESSAGE => {:type => Thrift::Types::STRING, :name => 'message'}
+  }
+
+  def struct_fields; FIELDS; end
+
+  def validate
+  end
+
+end
+

data/lib/ruby_scribe_client.rb

@@ -0,0 +1,33 @@
+#convenience wrapper for scribe
+require File.join([File.dirname(__FILE__), 'ruby_scribe_client', 'scribe'])
+
+class ScribeLogger
+  attr_accessor :host, :port, :namespace
+  
+  def initialize(host, port, namespace = nil)
+    @transport = Thrift::FramedTransport.new(Thrift::Socket.new(host, port))
+    @protocol = Thrift::BinaryProtocol.new(@transport, false, false)
+    @client = Scribe::Client.new(@protocol)
+    @namespace = namespace
+  end
+  
+  def []=(category, message)
+    log(category, message)
+  end
+  
+  def log(category, message)
+    begin
+      log_entry = LogEntry.new({:category => category_key(category), :message => message})
+      @transport.open()
+      result = @client.Log([log_entry])
+      @transport.close()
+      return result
+    rescue  Thrift::Exception => tx
+      print 'Thrift::Exception: ', tx.message, "\n"
+    end    
+  end
+  
+  def category_key(category)
+   [@namespace, category].compact.join('_')
+  end
+end

data/ruby_scribe_client.gemspec

@@ -0,0 +1,31 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{ruby_scribe_client}
+  s.version = "0.0.2"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 1.2") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Gary Tsang"]
+  s.date = %q{2009-03-12}
+  s.description = %q{Ruby Scribe Client. Package and Wrapper for generated ruby interfaces}
+  s.email = %q{}
+  s.extra_rdoc_files = ["CHANGELOG", "lib/ruby_scribe_client/FacebookService.rb", "lib/ruby_scribe_client/fb303_types.rb", "lib/ruby_scribe_client/reflection_limited_types.rb", "lib/ruby_scribe_client/scribe.rb", "lib/ruby_scribe_client/scribe_types.rb", "lib/ruby_scribe_client.rb", "LICENSE", "README"]
+  s.files = ["CHANGELOG", "etc/scribe_cat.rb", "lib/ruby_scribe_client/FacebookService.rb", "lib/ruby_scribe_client/fb303_types.rb", "lib/ruby_scribe_client/reflection_limited_types.rb", "lib/ruby_scribe_client/scribe.rb", "lib/ruby_scribe_client/scribe_types.rb", "lib/ruby_scribe_client.rb", "LICENSE", "Manifest", "Rakefile", "README", "ruby_scribe_client.gemspec", "vendor/thrift/client.rb", "vendor/thrift/deprecation.rb", "vendor/thrift/exceptions.rb", "vendor/thrift/processor.rb", "vendor/thrift/protocol/binaryprotocol.rb", "vendor/thrift/protocol/binaryprotocolaccelerated.rb", "vendor/thrift/protocol/tbinaryprotocol.rb", "vendor/thrift/protocol/tprotocol.rb", "vendor/thrift/protocol.rb", "vendor/thrift/serializer.rb", "vendor/thrift/server/httpserver.rb", "vendor/thrift/server/nonblockingserver.rb", "vendor/thrift/server/thttpserver.rb", "vendor/thrift/server/tserver.rb", "vendor/thrift/server.rb", "vendor/thrift/struct.rb", "vendor/thrift/thrift.rb", "vendor/thrift/transport/httpclient.rb", "vendor/thrift/transport/socket.rb", "vendor/thrift/transport/thttpclient.rb", "vendor/thrift/transport/tsocket.rb", "vendor/thrift/transport/ttransport.rb", "vendor/thrift/transport/unixsocket.rb", "vendor/thrift/transport.rb", "vendor/thrift/types.rb", "vendor/thrift.rb"]
+  s.has_rdoc = true
+  s.homepage = %q{http://github.com/garru/ruby_scribe_client}
+  s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "Ruby_scribe_client", "--main", "README"]
+  s.require_paths = ["lib"]
+  s.rubyforge_project = %q{ruby_scribe_client}
+  s.rubygems_version = %q{1.3.1}
+  s.summary = %q{Ruby Scribe Client. Package and Wrapper for generated ruby interfaces}
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 2
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+    else
+    end
+  else
+  end
+end

data/vendor/thrift/client.rb

@@ -0,0 +1,44 @@
+module Thrift
+  module Client
+    def initialize(iprot, oprot=nil)
+      @iprot = iprot
+      @oprot = oprot || iprot
+      @seqid = 0
+    end
+
+    def send_message(name, args_class, args = {})
+      @oprot.write_message_begin(name, MessageTypes::CALL, @seqid)
+      data = args_class.new
+      args.each do |k, v|
+        data.send("#{k.to_s}=", v)
+      end
+      begin
+        data.write(@oprot)
+      rescue StandardError => e
+        @oprot.trans.close
+        raise e
+      end
+      @oprot.write_message_end
+      @oprot.trans.flush
+    end
+
+    def receive_message(result_klass)
+      fname, mtype, rseqid = @iprot.read_message_begin
+      handle_exception(mtype)
+      result = result_klass.new
+      result.read(@iprot)
+      @iprot.read_message_end
+      result
+    end
+
+    def handle_exception(mtype)
+      if mtype == MessageTypes::EXCEPTION
+        x = ApplicationException.new
+        x.read(@iprot)
+        @iprot.read_message_end
+        raise x
+      end
+    end
+  end
+  deprecate_module! :ThriftClient => Client
+end

data/vendor/thrift/deprecation.rb

@@ -0,0 +1,155 @@
+# provide a backwards-compatible wrapper API and deprecate it
+
+module Thrift
+  DEPRECATION = true unless const_defined?(:DEPRECATION)
+end
+
+class Module
+  # Wraps the given methods to print a warning and call the real method
+  # Example:
+  #   deprecate! :readAll => :read_all
+  #--
+  # Yeah, this is ugly, passing a string to module_eval, but unfortunately
+  # using a block removes the ability to pass blocks to the defined method
+  # and breaks spec
+  def deprecate!(methods)
+    return unless Thrift::DEPRECATION
+    methods.each_pair do |old, new|
+      module_eval <<-EOF
+        def #{old}(*args, &block)
+          old, new = #{[old,new].inspect}
+          STDERR.puts "Warning: calling deprecated method \#{self.is_a?(Module) ? "\#{self}." : "\#{self.class}#"}\#{old}"
+          STDERR.puts "  from \#{caller.first}"
+          target = (self.is_a?(Module) ? (class << self;self;end) : self.class)
+          target.send :define_method, old, target.instance_method(new) # unwrap
+          target.instance_method(new).bind(self).call(*args, &block)
+        end
+      EOF
+    end
+  end
+end
+
+module Thrift::DeprecationProxy # :nodoc:
+  # there's a really weird bug in Ruby where class variables behave wrong
+  # when used in a Class.new or #class_eval rather than in a class foo block.
+  CLASS_MAPPING = {}
+  MODULE_MAPPING = {}
+  def self.new_class(obj, name)
+    klass_id = CLASS_MAPPING.size
+    CLASS_MAPPING[klass_id] = [obj, name, true]
+    klass = Class.new(obj) do
+      klass = self
+      instance_methods.sort.reject { |x| [:__id__,:__send__].include? x.to_sym }.each do |sym|
+        undef_method sym
+      end
+      define_method :__thrift_deprecation_proxy_klass_id do
+        klass_id
+      end
+      def method_missing(sym, *args, &block)
+        klass_id = __thrift_deprecation_proxy_klass_id
+        obj, name, warned = CLASS_MAPPING[klass_id]
+        obj.instance_method(sym).bind(self).call(*args, &block)
+      end
+      (class << self;self;end).class_eval do
+        instance_methods.sort.reject { |x| [:__id__,:__send__].include? x.to_sym }.each do |sym|
+          undef_method sym
+        end
+        define_method :__thrift_deprecation_proxy_klass do
+          [klass, klass_id]
+        end
+        def method_missing(sym, *args, &block)
+          klass, klass_id = __thrift_deprecation_proxy_klass
+          obj, name, warned = CLASS_MAPPING[klass_id]
+          unless warned
+            STDERR.puts "Warning: class #{name} is deprecated"
+            STDERR.puts "  from #{Thrift::DeprecationProxy.process_caller(caller)}"
+            CLASS_MAPPING[klass_id][2] = true
+          end
+          if klass.__id__ == self.__id__
+            obj.send sym, *args, &block
+          else
+            obj.method(sym).unbind.bind(self).call(*args, &block)
+          end
+        end
+      end
+    end
+    CLASS_MAPPING[klass_id][2] = false
+    klass
+  end
+  def self.new_module(obj, name)
+    mod_id = MODULE_MAPPING.size
+    MODULE_MAPPING[mod_id] = [obj, name, true]
+    mod = Module.new do
+      include obj
+      instance_methods.sort.reject { |x| [:__id__,:__send__].include? x.to_sym }.each do |sym|
+        undef_method sym
+      end
+      define_method :__thrift_deprecation_proxy_module_id do
+        mod_id
+      end
+      def method_missing(sym, *args, &block)
+        mod_id = __thrift_deprecation_proxy_module_id
+        obj, name, warned = MODULE_MAPPING[mod_id]
+        unless warned
+          STDERR.puts "Warning: module #{name} is deprecated"
+          STDERR.puts "  from #{Thrift::DeprecationProxy.process_caller(caller)}"
+          MODULE_MAPPING[mod_id][2] = true
+        end
+        obj.instance_method(sym).bind(self).call(*args, &block)
+      end
+      (class << self;self;end).class_eval do
+        instance_methods.sort.reject { |x| [:__id__,:__send__].include? x.to_sym }.each do |sym|
+          undef_method sym
+        end
+        define_method :__thrift_deprecation_proxy_module_id do
+          mod_id
+        end
+        def method_missing(sym, *args, &block)
+          mod_id = __thrift_deprecation_proxy_module_id
+          obj, name, warned = MODULE_MAPPING[mod_id]
+          unless warned
+            STDERR.puts "Warning: module #{name} is deprecated"
+            STDERR.puts "  from #{Thrift::DeprecationProxy.process_caller(caller)}"
+            MODULE_MAPPING[mod_id][2] = true
+          end
+          obj.send sym, *args, &block
+        end
+      end
+    end
+    MODULE_MAPPING[mod_id][2] = false
+    mod
+  end
+  def self.process_caller(stack)
+    dir = File.dirname(__FILE__)
+    stack.find { |frame| frame[0,dir.size] != dir }
+  end
+  def self.reset_deprecation_warnings
+    CLASS_MAPPING.each { |k,v| v[2] = false }
+    MODULE_MAPPING.each { |k,v| v[2] = false }
+  end
+end
+
+module Kernel
+  # Provides an alternate name for the class for deprecation purposes
+  # Example:
+  #   deprecate_class! :TBinaryProtocol => Thrift::BinaryProtocol
+  #--
+  # at the moment this only works for creating top-level constants
+  # if necessary, this can be extended to take something like :'Thrift::TBinaryProtocol'
+  # alternately, Module can be extended with a similar method
+  def deprecate_class!(klasses)
+    return unless Thrift::DEPRECATION
+    klasses.each_pair do |old, new|
+      raise "deprecate_class! expected Class, called with #{new}" unless new.is_a? Class
+      Object.const_set old, Thrift::DeprecationProxy.new_class(new, old)
+    end
+  end
+
+  # like deprecate_class! but for Modules
+  def deprecate_module!(modules)
+    return unless Thrift::DEPRECATION
+    modules.each_pair do |old, new|
+      Object.const_set old, Thrift::DeprecationProxy.new_module(new, old)
+    end
+  end
+end

data/vendor/thrift/exceptions.rb

@@ -0,0 +1,65 @@
+module Thrift
+  class Exception < StandardError
+    def initialize(message)
+      super
+      @message = message
+    end
+
+    attr_reader :message
+  end
+  deprecate_class! :TException => Exception
+
+  class ApplicationException < Exception
+
+    UNKNOWN = 0
+    UNKNOWN_METHOD = 1
+    INVALID_MESSAGE_TYPE = 2
+    WRONG_METHOD_NAME = 3
+    BAD_SEQUENCE_ID = 4
+    MISSING_RESULT = 5
+
+    attr_reader :type
+
+    def initialize(type=UNKNOWN, message=nil)
+      super(message)
+      @type = type
+    end
+
+    def read(iprot)
+      iprot.read_struct_begin
+      while true
+        fname, ftype, fid = iprot.read_field_begin
+        if ftype == Types::STOP
+          break
+        end
+        if fid == 1 and ftype == Types::STRING
+          @message = iprot.read_string
+        elsif fid == 2 and ftype == Types::I32
+          @type = iprot.read_i32
+        else
+          iprot.skip(ftype)
+        end
+        iprot.read_field_end
+      end
+      iprot.read_struct_end
+    end
+
+    def write(oprot)
+      oprot.write_struct_begin('Thrift::ApplicationException')
+      unless @message.nil?
+        oprot.write_field_begin('message', Types::STRING, 1)
+        oprot.write_string(@message)
+        oprot.write_field_end
+      end
+      unless @type.nil?
+        oprot.write_field_begin('type', Types::I32, 2)
+        oprot.write_i32(@type)
+        oprot.write_field_end
+      end
+      oprot.write_field_stop
+      oprot.write_struct_end
+    end
+
+  end
+  deprecate_class! :TApplicationException => ApplicationException
+end

data/vendor/thrift/processor.rb

@@ -0,0 +1,39 @@
+module Thrift
+  module Processor
+    def initialize(handler)
+      @handler = handler
+    end
+
+    def process(iprot, oprot)
+      name, type, seqid  = iprot.read_message_begin
+      if respond_to?("process_#{name}")
+        send("process_#{name}", seqid, iprot, oprot)
+        true
+      else
+        iprot.skip(Types::STRUCT)
+        iprot.read_message_end
+        x = ApplicationException.new(ApplicationException::UNKNOWN_METHOD, 'Unknown function '+name)
+        oprot.write_message_begin(name, MessageTypes::EXCEPTION, seqid)
+        x.write(oprot)
+        oprot.write_message_end
+        oprot.trans.flush
+        false
+      end
+    end
+
+    def read_args(iprot, args_class)
+      args = args_class.new
+      args.read(iprot)
+      iprot.read_message_end
+      args
+    end
+
+    def write_result(result, oprot, name, seqid)
+      oprot.write_message_begin(name, MessageTypes::REPLY, seqid)
+      result.write(oprot)
+      oprot.write_message_end
+      oprot.trans.flush
+    end
+  end
+  deprecate_module! :TProcessor => Processor
+end