upfluence-thrift 1.0.11 → 2.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: 41dea1fac87e62e08f31f9fca894dde1a3b33247
-  data.tar.gz: 03136b94c08b9a1eaeb03f51ee5f5b9797da7eac
+SHA256:
+  metadata.gz: ea7054176b376c94f565f2a9404599502b35853c1fcba1735d7153a16ccb1fa7
+  data.tar.gz: d90c18cf190256af2d4a0240bb4867bcf7b6e53b2f29f021e90e9d8eb8976a4e
 SHA512:
-  metadata.gz: 3a33f22c457a6c8606d33d6e3e27da1099cf3463502433132016219e8266576daaca0b21155e866871a5399c4269f2b84c201e31dfcf31a95519107386fc8ffb
-  data.tar.gz: 82d36e40a917bfe445beac495dcb944ccded0cdb7ced3f414ce90993888e14deaff0c7d50693e41b338ce9e911a841e7ce81879a305e419952b72f179b2db3b8
+  metadata.gz: 41c69f0f716a5f5be7d9ee0a8e06c5d66463dc5ba5a7714c3f6e8db51248933d06d78c0b32b7217112358a0dcfc9e973195efd412e6b2cac526134074af228d7
+  data.tar.gz: 0c6b50c4e0b91776991f82f4b80d053685a1f9c09e16ef5bb1efea0dbec866ebde4dc1c2a0dac26cd1c05d9f1d7c7afb3131078d47f2380e963767c4c4c7e78e

data/lib/thrift.rb

@@ -1,4 +1,4 @@
-# 
+#
 # Licensed to the Apache Software Foundation (ASF) under one
 # or more contributor license agreements. See the NOTICE file
 # distributed with this work for additional information
@@ -6,28 +6,30 @@
 # to you under the Apache License, Version 2.0 (the
 # "License"); you may not use this file except in compliance
 # with the License. You may obtain a copy of the License at
-# 
+#
 #   http://www.apache.org/licenses/LICENSE-2.0
-# 
+#
 # Unless required by applicable law or agreed to in writing,
 # software distributed under the License is distributed on an
 # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 # KIND, either express or implied. See the License for the
 # specific language governing permissions and limitations
 # under the License.
-# 
+#
 # Contains some contributions under the Thrift Software License.
 # Please see doc/old-thrift-license.txt in the Thrift distribution for
 # details.
 
 $:.unshift File.dirname(__FILE__)
 
+require 'thrift/definition'
 require 'thrift/bytes'
 require 'thrift/core_ext'
 require 'thrift/exceptions'
 require 'thrift/types'
 require 'thrift/processor'
 require 'thrift/multiplexed_processor'
+require 'thrift/middleware'
 require 'thrift/client'
 require 'thrift/struct'
 require 'thrift/union'

data/lib/thrift/client.rb

@@ -16,10 +16,11 @@
 # specific language governing permissions and limitations
 # under the License.
 #
+#
 
 module Thrift
   module Client
-    def initialize(iprot, oprot=nil)
+    def initialize(iprot, oprot = nil)
       @iprot = iprot
       @oprot = oprot || iprot
       @seqid = 0
@@ -39,9 +40,15 @@ module Thrift
 
     def send_message_args(args_class, args)
       data = args_class.new
+
       args.each do |k, v|
         data.send("#{k.to_s}=", v)
       end
+
+      send_message_instance(data)
+    end
+
+    def send_message_instance(data)
       begin
         data.write(@oprot)
       rescue StandardError => e
@@ -86,4 +93,27 @@ module Thrift
       raise x
     end
   end
+
+  class BaseClient
+    include Client
+
+    def call_unary(name, req)
+      @oprot.write_message_begin(name, Thrift::MessageTypes::ONEWAY, @seqid)
+      send_message_instance(req)
+    end
+
+    def call_binary(name, req, resp_klass)
+      @oprot.write_message_begin(name, Thrift::MessageTypes::CALL, @seqid)
+      send_message_instance(req)
+      receive_message(resp_klass)
+    end
+  end
+
+  class << self
+    def build_client(input)
+      return BaseClient.new(input) if input.is_a? BaseProtocol
+
+      input
+    end
+  end
 end

data/lib/thrift/definition.rb

@@ -0,0 +1,64 @@
+module Thrift
+  class StructDefinition
+    attr_reader :klass
+
+    def initialize(klass)
+      @klass = klass
+    end
+
+    def namespace
+      @klass::NAMESPACE
+    end
+
+    def name
+      @klass::NAME
+    end
+
+    def struct_type
+      "#{namespace}.#{name}"
+    end
+  end
+
+  class ServiceDefinition < StructDefinition
+    attr_reader :klass
+
+    def initialize(klass)
+      @klass = klass
+    end
+
+    def client_class
+      @klass::Client
+    end
+
+    def processor_class
+      @klass::Processor
+    end
+
+    def namespace
+      @klass::NAMESPACE
+    end
+
+    def service
+      @klass::SERVICE
+    end
+
+    def service_type
+      "#{namespace}.#{service}"
+    end
+  end
+
+  STRUCT_DEFINITIONS = {}
+  SERVICE_DEFINITIONS = {}
+
+  class << self
+    def register_struct_type(klass)
+      definition = StructDefinition.new(klass)
+      STRUCT_DEFINITIONS[definition.struct_type] = definition
+    end
+
+    def register_service_type(klass)
+      definition = ServiceDefinition.new(klass)
+      SERVICE_DEFINITIONS[definition.service_type] = definition
+    end
+  end
+end

data/lib/thrift/exceptions.rb

@@ -1,4 +1,4 @@
-# 
+#
 # Licensed to the Apache Software Foundation (ASF) under one
 # or more contributor license agreements. See the NOTICE file
 # distributed with this work for additional information
@@ -6,21 +6,21 @@
 # to you under the Apache License, Version 2.0 (the
 # "License"); you may not use this file except in compliance
 # with the License. You may obtain a copy of the License at
-# 
+#
 #   http://www.apache.org/licenses/LICENSE-2.0
-# 
+#
 # Unless required by applicable law or agreed to in writing,
 # software distributed under the License is distributed on an
 # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 # KIND, either express or implied. See the License for the
 # specific language governing permissions and limitations
 # under the License.
-# 
+#
 
 module Thrift
   class Exception < StandardError
     def initialize(message)
-      super
+      super(message)
       @message = message
     end
 

data/lib/thrift/middleware.rb

@@ -0,0 +1,50 @@
+module Thrift
+  module Middleware
+    class NopMiddleware
+      def handle_binary(_mth, args = {}, &block)
+        block.call(args)
+      end
+
+      def handle_unary(_mth, args = {},  &block)
+        block.call(args)
+      end
+    end
+
+    class MultiMiddleware
+      def initialize(middlewares)
+        @middlewares = middlewares
+      end
+
+      def handle_unary(mth, args = {}, &block)
+        @middlewares.reverse.reduce(block) do |acc, m|
+          Proc.new do |args|
+            m.handle_unary(mth, args) { |args| acc.call(args) }
+          end
+        end.call(args)
+      end
+
+      def handle_binary(mth, args = {}, &block)
+        @middlewares.reverse.reduce(block) do |acc, m|
+          Proc.new do |args|
+            m.handle_binary(mth, args) { |args| acc.call(args) }
+          end
+        end.call(args)
+      end
+    end
+
+    NOP_MIDDLEWARE = NopMiddleware.new
+
+    class << self
+      def wrap(middlewares)
+        case middlewares.length
+        when 0
+          NOP_MIDDLEWARE
+        when 1
+          middlewares.first
+        else
+          MultiMiddleware.new(middlewares)
+        end
+      end
+    end
+  end
+end

data/lib/thrift/processor.rb

@@ -19,8 +19,9 @@
 
 module Thrift
   module Processor
-    def initialize(handler)
+    def initialize(handler, middlewares = [])
       @handler = handler
+      @middleware = Middleware.wrap(middlewares)
     end
 
     def process(iprot, oprot)
@@ -31,7 +32,6 @@ module Thrift
         rescue => e
           write_exception(e, oprot, name, seqid)
         end
-
         true
       else
         iprot.skip(Types::STRUCT)
@@ -39,7 +39,7 @@ module Thrift
         write_exception(
           ApplicationException.new(
             ApplicationException::UNKNOWN_METHOD,
-            'Unknown function '+name,
+            'Unknown function ' + name,
           ),
           oprot,
           name,

data/lib/thrift/protocol/json_protocol.rb

@@ -1,5 +1,5 @@
 # encoding: UTF-8
-# 
+#
 # Licensed to the Apache Software Foundation (ASF) under one
 # or more contributor license agreements. See the NOTICE file
 # distributed with this work for additional information
@@ -7,16 +7,16 @@
 # to you under the Apache License, Version 2.0 (the
 # "License"); you may not use this file except in compliance
 # with the License. You may obtain a copy of the License at
-# 
+#
 #   http://www.apache.org/licenses/LICENSE-2.0
-# 
+#
 # Unless required by applicable law or agreed to in writing,
 # software distributed under the License is distributed on an
 # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 # KIND, either express or implied. See the License for the
 # specific language governing permissions and limitations
 # under the License.
-# 
+#
 
 
 module Thrift
@@ -469,7 +469,7 @@ module Thrift
     end
 
     def write_string(str)
-      write_json_string(str)
+      write_json_string(Bytes.convert_to_utf8_byte_buffer(str))
     end
 
     def write_binary(str)
@@ -753,7 +753,7 @@ module Thrift
     end
 
     def read_string
-      read_json_string
+      Bytes.convert_to_string(read_json_string)
     end
 
     def read_binary

data/lib/thrift/server/rack_application.rb

@@ -21,37 +21,48 @@ require 'rack'
 
 module Thrift
   class RackApplication
-    THRIFT_HEADER = "application/x-thrift"
+    THRIFT_HEADER = 'application/x-thrift'.freeze
 
     def self.for(path, processor, protocol_factory)
       Rack::Builder.new do
         map path do
-          run lambda { |env|
-            request = Rack::Request.new(env)
-            if RackApplication.valid_thrift_request?(request)
-              RackApplication.successful_request(request, processor, protocol_factory)
-            else
-              RackApplication.failed_request
-            end
-          }
+          run Thrift::RackApplication.new(processor, protocol_factory)
         end
       end
     end
 
-    def self.successful_request(rack_request, processor, protocol_factory)
-      response = Rack::Response.new([], 200, {'Content-Type' => THRIFT_HEADER})
-      transport = IOStreamTransport.new rack_request.body, response
-      protocol = protocol_factory.get_protocol transport
-      processor.process protocol, protocol
-      response
+    def initialize(processor, protocol_factory)
+      @processor = processor
+      @protocol_factory = protocol_factory
+      @headers = { 'Content-Type' => THRIFT_HEADER }
     end
 
-    def self.failed_request
-      Rack::Response.new(['Not Found'], 404, {'Content-Type' => THRIFT_HEADER})
+    def call(env)
+      req = Rack::Request.new(env)
+
+      if valid_thrift_request?(req)
+        successful_request(req)
+      else
+        failed_request
+      end
+    end
+
+    def successful_request(req)
+      resp = Rack::Response.new([], 200, @headers)
+      transport = IOStreamTransport.new req.body, resp
+      protocol = @protocol_factory.get_protocol transport
+
+      @processor.process protocol, protocol
+
+      [resp.status, resp.headers, resp.body]
+    end
+
+    def failed_request
+      [404, @headers, 'Not found']
     end
 
-    def self.valid_thrift_request?(rack_request)
-      rack_request.post? && rack_request.env["CONTENT_TYPE"] == THRIFT_HEADER
+    def valid_thrift_request?(req)
+      req.post? && req.env['CONTENT_TYPE'] == THRIFT_HEADER
     end
   end
 end

data/lib/thrift/struct.rb

@@ -1,4 +1,4 @@
-# 
+#
 # Licensed to the Apache Software Foundation (ASF) under one
 # or more contributor license agreements. See the NOTICE file
 # distributed with this work for additional information
@@ -6,16 +6,16 @@
 # to you under the Apache License, Version 2.0 (the
 # "License"); you may not use this file except in compliance
 # with the License. You may obtain a copy of the License at
-# 
+#
 #   http://www.apache.org/licenses/LICENSE-2.0
-# 
+#
 # Unless required by applicable law or agreed to in writing,
 # software distributed under the License is distributed on an
 # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 # KIND, either express or implied. See the License for the
 # specific language governing permissions and limitations
 # under the License.
-# 
+#
 
 require 'set'
 
@@ -24,15 +24,15 @@ module Thrift
     def initialize(d={}, &block)
       # get a copy of the default values to work on, removing defaults in favor of arguments
       fields_with_defaults = fields_with_default_values.dup
-      
-      # check if the defaults is empty, or if there are no parameters for this 
+
+      # check if the defaults is empty, or if there are no parameters for this
       # instantiation, and if so, don't bother overriding defaults.
       unless fields_with_defaults.empty? || d.empty?
         d.each_key do |name|
           fields_with_defaults.delete(name.to_s)
         end
       end
-      
+
       # assign all the user-specified arguments
       unless d.empty?
         d.each do |name, value|
@@ -43,14 +43,14 @@ module Thrift
           instance_variable_set("@#{name}", value)
         end
       end
-      
+
       # assign all the default values
       unless fields_with_defaults.empty?
         fields_with_defaults.each do |name, default_value|
           instance_variable_set("@#{name}", (default_value.dup rescue default_value))
         end
       end
-      
+
       yield self if block_given?
     end
 
@@ -67,7 +67,7 @@ module Thrift
       end
       fields_with_default_values
     end
-    
+
     def inspect(skip_optional_nulls = true)
       fields = []
       each_field do |fid, field_info|