upfluence-thrift 2.2.0 → 2.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: eb0c6edc7181a95e78d4f3d1da97bf55cff534a6c4f3784c998bbefa1fc0df1c
-  data.tar.gz: c225382fb7a4bc1a0c4896f74a970c669eaf7cac96414b6c7e1be3d1dc7e66e5
+  metadata.gz: ea7054176b376c94f565f2a9404599502b35853c1fcba1735d7153a16ccb1fa7
+  data.tar.gz: d90c18cf190256af2d4a0240bb4867bcf7b6e53b2f29f021e90e9d8eb8976a4e
 SHA512:
-  metadata.gz: 9bfbd290bda69a9ca7102791fd2291e35902691598be71d1a1429dd16329ef9dad76a38d363e76098ee2b88af7ce8df93927b0f0458f41b93b8e4ec133b227b7
-  data.tar.gz: ffbb8d16a28268f9cb6cc82fa089e9bb06f96aefb3097077923478e4547fbc5cec43b2b640e0a790434e5ed41422236405c75bb835c5e2ac7735352e0dae182b
+  metadata.gz: 41c69f0f716a5f5be7d9ee0a8e06c5d66463dc5ba5a7714c3f6e8db51248933d06d78c0b32b7217112358a0dcfc9e973195efd412e6b2cac526134074af228d7
+  data.tar.gz: 0c6b50c4e0b91776991f82f4b80d053685a1f9c09e16ef5bb1efea0dbec866ebde4dc1c2a0dac26cd1c05d9f1d7c7afb3131078d47f2380e963767c4c4c7e78e

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/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

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

@@ -0,0 +1,31 @@
+require 'thrift/types/known/timestamp_types'
+
+module Thrift
+  module Types
+    module Known
+      module Timestamp
+        class << self
+          def from_time(v)
+            Timestamp.from_time(v)
+          end
+
+          def now
+            from_time(Time.now)
+          end
+        end
+
+        class Timestamp
+          class << self
+            def from_time(v)
+              Timestamp.new(seconds: v.to_i, nanos: v.nsec)
+            end
+          end
+
+          def to_time
+            Time.at(seconds, nanos, :nsec)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/thrift/types/known/timestamp_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 'timestamp_types'
+
+module Thrift
+  module Types
+    module Known
+      module Timestamp
+      end
+    end
+  end
+end