upfluence-thrift 1.0.11 → 2.3.0

data/lib/thrift/transport/base_transport.rb

@@ -1,5 +1,5 @@
 # encoding: ascii-8bit
-# 
+#
 # 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
   class TransportException < Exception
@@ -48,7 +48,7 @@ module Thrift
 
   class BaseTransport
     def open?; end
-    
+
     def open; end
 
     def close; end
@@ -62,6 +62,8 @@ module Thrift
       raise NotImplementedError
     end
 
+    def set_context(ctx); end
+
     # Returns an unsigned byte as a Fixnum in the range (0..255).
     def read_byte
       buf = read_all(1)
@@ -86,7 +88,7 @@ module Thrift
         chunk = read(size - buf.length)
         buf << chunk
       end
-    
+
       buf
     end
 
@@ -100,7 +102,7 @@ module Thrift
 
     def flush; end
   end
-  
+
   class BaseTransportFactory
     def get_transport(trans)
       return trans

data/lib/thrift/transport/server_socket.rb

@@ -1,5 +1,5 @@
 # encoding: ascii-8bit
-# 
+#
 # 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.
-# 
+#
 
 require 'socket'
 
@@ -37,7 +37,7 @@ module Thrift
     attr_reader :handle
 
     def listen
-      @handle = TCPServer.new(@host, @port)
+      @handle ||= TCPServer.new(@host, @port)
     end
 
     def accept
@@ -60,4 +60,4 @@ module Thrift
 
     alias to_io handle
   end
-end
+end

data/lib/thrift/types.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'
 

data/lib/thrift/types/known/any.rb

@@ -0,0 +1,179 @@
+require 'thrift/types/known/any_types'
+require 'json'
+require 'yaml'
+
+module Thrift
+  module Types
+    module Known
+      module Any
+        class CustomYAML
+          class << self
+            def dump(obj)
+              YAML.dump(obj)[4..-1]
+            end
+
+            def load(val)
+              YAML.safe_load(val)
+            end
+          end
+        end
+
+        class MetaCodec
+          def initialize(klass)
+            @klass = klass
+          end
+
+          def encode(obj)
+            @klass.dump(to_hash(obj))
+          end
+
+          def decode(buf, obj)
+            from_hash(obj, @klass.load(buf))
+          end
+
+          private
+
+          def to_hash(obj)
+            to_hash_field(obj, type: Thrift::Types::STRUCT)
+          end
+
+          def to_hash_field(value, field)
+            case field[:type]
+            when Thrift::Types::STRUCT
+              value.struct_fields.values.reduce({}) do |acc, sfield|
+                name = sfield[:name]
+                vv = value.send("#{name}?") ? to_hash_field(value.send(name), sfield) : nil
+                vv.nil? ? acc : acc.merge(name => vv)
+              end
+            when Thrift::Types::LIST
+              value.map { |vv| to_hash_field(vv, field[:element]) }
+            when Thrift::Types::MAP
+              value.reduce({}) do |acc, (k, v)|
+                kv = to_hash_field(k, field[:key])
+                vv = to_hash_field(v, field[:value])
+
+                acc.merge(kv => vv)
+              end
+            else
+              value
+            end
+          end
+
+          def from_hash(obj, hash)
+            return nil if hash.nil?
+
+            obj.struct_fields.values.each do |field|
+              name = field[:name]
+              value = hash[name]
+
+              obj.send("#{name}=", from_value_field(value, field)) if value
+            end
+
+            obj
+          end
+
+          def from_value_field(value, field)
+            case field[:type]
+            when Thrift::Types::STRUCT
+              from_hash(field[:class].new, value)
+            when Thrift::Types::LIST
+              value.map { |vv| from_value_field(vv, field[:element]) }
+            when Thrift::Types::MAP
+              value.reduce({}) do |acc, (k, v)|
+                kv = from_value_field(k, field[:key])
+                vv = from_value_field(v, field[:value])
+                acc.merge(kv => vv)
+              end
+            else
+              value
+            end
+          end
+        end
+
+        class ProtocolCodec
+          def initialize(protocol_factory)
+            @protocol_factory = protocol_factory
+          end
+
+          def encode(obj)
+            Serializer.new(@protocol_factory).serialize(obj)
+          end
+
+          def decode(buf, obj)
+            obj.read(
+              @protocol_factory.get_protocol(MemoryBufferTransport.new(buf))
+            )
+          end
+        end
+
+        DEFAULT_CODEC = ProtocolCodec.new(JsonProtocolFactory.new)
+        JSON_CODEC = MetaCodec.new(JSON)
+        YAML_CODEC = MetaCodec.new(CustomYAML)
+
+        CODECS = {
+          ''     => DEFAULT_CODEC,
+          'json' => JSON_CODEC,
+          'yaml' => YAML_CODEC,
+          'yml'  => YAML_CODEC
+        }
+
+
+        class << self
+          def from_object(obj, codec_key = '')
+            Any.from_object(obj, codec_key)
+          end
+        end
+
+        class Any
+          class TypeNotHandled < StandardError; end
+          class CodecNotHandled < StandardError; end
+
+          class << self
+            def from_object(obj, codec_key = '')
+              struct_def = Thrift::STRUCT_DEFINITIONS.values.find do |v|
+                obj.is_a? v.klass
+              end
+
+              raise TypeNotHandled unless struct_def
+              raise CodecNotHandled unless CODECS[codec_key]
+
+              key = codec_key.eql?('') ? '' : "-#{codec_key}"
+
+              Any.new(
+                type:  "thrift#{key}/#{struct_def.struct_type}",
+                value: CODECS[codec_key].encode(obj)
+              )
+            end
+          end
+
+          def to_object
+            codec_key, struct_type = parse_type
+
+            raise TypeNotHandled unless Thrift::STRUCT_DEFINITIONS[struct_type]
+            raise CodecNotHandled unless CODECS[codec_key]
+
+            res = Thrift::STRUCT_DEFINITIONS[struct_type].klass.new
+            CODECS[codec_key].decode(value, res)
+
+            res
+          end
+
+          private
+
+          def parse_type
+            ts = type.split('/')
+
+            if ts.length != 2 || (ts[0] =~ /^thrift-?/) != 0
+              raise TypeNotHandled.new
+            end
+
+            codec = ts[0][6..-1]
+            codec = codec[1..-1] if codec[0].eql?('-')
+
+            [codec, ts[1]]
+          end
+        end
+      end
+    end
+  end
+end

data/lib/thrift/types/known/any_constants.rb

@@ -0,0 +1,17 @@
+#
+# Autogenerated by Thrift Compiler (2.3.0-upfluence)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+require 'any_types'
+
+module Thrift
+  module Types
+    module Known
+      module Any
+      end
+    end
+  end
+end

data/lib/thrift/types/known/any_types.rb

@@ -0,0 +1,43 @@
+#
+# Autogenerated by Thrift Compiler (2.3.0-upfluence)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+
+module Thrift
+  module Types
+    module Known
+      module Any
+        class Any; end
+
+        class Any
+          include ::Thrift::Struct, ::Thrift::Struct_Union
+
+          NAME = 'Any'.freeze
+          NAMESPACE = 'types.known.any'.freeze
+
+          THRIFT_FIELD_INDEX_TYPE = 1
+          THRIFT_FIELD_INDEX_VALUE = 2
+
+          FIELDS = {
+            THRIFT_FIELD_INDEX_TYPE => {type: ::Thrift::Types::STRING, name: 'type'},
+            THRIFT_FIELD_INDEX_VALUE => {type: ::Thrift::Types::STRING, name: 'value', binary: true}
+          }
+
+          def struct_fields; FIELDS; end
+
+          def validate
+            raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field type is unset!') unless @type
+            raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field value is unset!') unless @value
+          end
+
+          ::Thrift::Struct.generate_accessors self
+          ::Thrift.register_struct_type self
+        end
+
+      end
+    end
+  end
+end

data/lib/thrift/types/known/duration.rb

@@ -0,0 +1,27 @@
+require 'thrift/types/known/duration_types'
+
+module Thrift
+  module Types
+    module Known
+      module Duration
+        class << self
+          def from_number(v)
+            Duration.from_number(v)
+          end
+        end
+
+        class Duration
+          class << self
+            def from_number(v)
+              Duration.new(seconds: v.to_i, nanos: ((v % 1) * 1e9).to_i)
+            end
+          end
+
+          def to_number
+            seconds + nanos / 1e9
+          end
+        end
+      end
+    end
+  end
+end

data/lib/thrift/types/known/duration_constants.rb

@@ -0,0 +1,17 @@
+#
+# Autogenerated by Thrift Compiler (2.3.0-upfluence)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+require 'duration_types'
+
+module Thrift
+  module Types
+    module Known
+      module Duration
+      end
+    end
+  end
+end

data/lib/thrift/types/known/duration_types.rb

@@ -0,0 +1,43 @@
+#
+# Autogenerated by Thrift Compiler (2.3.0-upfluence)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+
+module Thrift
+  module Types
+    module Known
+      module Duration
+        class Duration; end
+
+        class Duration
+          include ::Thrift::Struct, ::Thrift::Struct_Union
+
+          NAME = 'Duration'.freeze
+          NAMESPACE = 'types.known.duration'.freeze
+
+          THRIFT_FIELD_INDEX_SECONDS = 1
+          THRIFT_FIELD_INDEX_NANOS = 2
+
+          FIELDS = {
+            THRIFT_FIELD_INDEX_SECONDS => {type: ::Thrift::Types::I64, name: 'seconds'},
+            THRIFT_FIELD_INDEX_NANOS => {type: ::Thrift::Types::I32, name: 'nanos'}
+          }
+
+          def struct_fields; FIELDS; end
+
+          def validate
+            raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field seconds is unset!') unless @seconds
+            raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field nanos is unset!') unless @nanos
+          end
+
+          ::Thrift::Struct.generate_accessors self
+          ::Thrift.register_struct_type self
+        end
+
+      end
+    end
+  end
+end