json-streamer 1.1.1 → 1.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9fa1fe76772ac9b1bf8e8b64da2362b13f3d4d5d
-  data.tar.gz: 723a1e77a3fbed2ff1a387409633d53f852070c9
+  metadata.gz: 928cc40a89ddc3be04f541abaed833e0fa89757a
+  data.tar.gz: ed5bc1ab4c5b00a6369b76a0e008d86ff7000bb1
 SHA512:
-  metadata.gz: '06830de8e7a7bbadd1e3e0db3cfd77cbad559807ae50c2588c3aeb1f1bbd6554cd04abaf590c40eb93862f519d035eade7dc90934eadb200e8a0afdb37ecacb0'
-  data.tar.gz: 0e443f59c654125a9cba390b9ea12575f437e37f280a303b134a403774d5421a7517f379fba7dd7f92124f38e566f16f289b98e181e35f508ac85d2375a3b20a
+  metadata.gz: afaba7db14dea5fe606b04711adcd3c9d04521f4ae4a7c9dc65f1d91debecc7b41c76a733f10c6d3e4bcb95f4690c5379d4ab9ac9d028f0d9ebf835353268298
+  data.tar.gz: a31a925947007fba4c4367e3869276f3d818294dc1afaee66f877e7153d034848bb2a87d45d151025570c7576b958e45efc49511d32def4fbd39510b40303bcf

data/lib/json/streamer/json_streamer.rb

@@ -13,10 +13,9 @@ module Json
         @file_io = file_io
         @chunk_size = chunk_size
 
-        @current_nesting_level = -1
-        @current_key = nil
+        @current_level = -1
         @aggregator = {}
-        @temp_aggregator_keys = {}
+        @aggregator_keys = {}
 
         @parser.start_object {start_object}
         @parser.start_array {start_array}
@@ -25,87 +24,115 @@ module Json
 
       # Callbacks containing `yield` have to be defined in the method called via block otherwise yield won't work
       def get(nesting_level:-1, key:nil, yield_values:true)
-        yield_nesting_level = nesting_level
-        wanted_key = key
+        @yield_level = nesting_level
+        @yield_key = key
+        @yield_values = yield_values
 
         @parser.value do |v|
-          if @aggregator[@current_nesting_level].kind_of? Array
-            if yield_values and yield_value?(yield_nesting_level)
-              yield v
-            else
-              @aggregator[@current_nesting_level] << v
-            end
-          else
-            @aggregator[@current_nesting_level][@current_key] = v
-            if yield_values and yield_value?(yield_nesting_level, wanted_key)
-              yield v
-            end
-          end
+          value(v) { |desired_object| yield desired_object }
         end
 
         @parser.end_object do
-          if yield_object?(yield_nesting_level, wanted_key)
-            yield @aggregator[@current_nesting_level].clone
-            @aggregator[@current_nesting_level] = {}
-          else
-            merge_up
-          end
-
-          @current_nesting_level -= 1
+          end_level(Hash.new) { |desired_object| yield desired_object }
         end
 
         @parser.end_array do
-          if yield_object?(yield_nesting_level, wanted_key)
-            yield @aggregator[@current_nesting_level].clone
-            @aggregator[@current_nesting_level] = []
-          else
-            merge_up
-          end
-
-          @current_nesting_level -= 1
+          end_level(Array.new) { |desired_object| yield desired_object }
         end
 
-        if @file_io
-          @file_io.each(@chunk_size) do |chunk|
-            @parser << chunk
-          end
+        @file_io.each(@chunk_size) { |chunk| @parser << chunk } if @file_io
+      end
+
+      def start_object
+        new_level(Hash.new)
+      end
+
+      def start_array
+        new_level(Array.new)
+      end
+
+      def key(k)
+        @current_key = k
+      end
+
+      def value(value)
+        reset_current_key if array_level?(@current_level)
+        yield value if yield_value?
+        add_value(value)
+      end
+
+      def add_value(value)
+        if array_level?(@current_level)
+          @aggregator[@current_level] << value
+        else
+          @aggregator[@current_level][@current_key] = value
         end
       end
 
-      def yield_object?(yield_nesting_level, wanted_key)
-        @current_nesting_level.eql? yield_nesting_level or (not wanted_key.nil? and wanted_key == @temp_aggregator_keys[@current_nesting_level-1])
+      def end_level(type)
+        if yield_object?
+          yield @aggregator[@current_level].clone
+          reset_current_level(type)
+        else
+          merge_up
+        end
+
+        @current_level -= 1
       end
 
-      def yield_value?(yield_nesting_level, wanted_key = nil)
-        (@current_nesting_level + 1).eql? yield_nesting_level or (not wanted_key.nil? and wanted_key == @current_key)
+      def yield_object?
+        @current_level.eql?(@yield_level) or (not @yield_key.nil? and @yield_key == previous_key)
       end
 
-      def start_object
-        @temp_aggregator_keys[@current_nesting_level] = @current_key
-        @current_nesting_level += 1
-        @aggregator[@current_nesting_level] = {}
+      def yield_value?
+        @yield_values and ((next_level).eql?(@yield_level) or (not @yield_key.nil? and @yield_key == @current_key))
       end
 
-      def start_array
-        @current_nesting_level += 1
-        @aggregator[@current_nesting_level] = []
+      def new_level(type)
+        reset_current_key if array_level?(@current_level)
+        set_aggregator_key
+        @current_level += 1
+        reset_current_level(type)
       end
 
-      def key(k)
-        @current_key = k
+      def reset_current_level(type)
+        @aggregator[@current_level] = type
+      end
+
+      def set_aggregator_key
+        @aggregator_keys[@current_level] = @current_key
+      end
+
+      def reset_current_key
+        @current_key = nil
+      end
+
+      def array_level?(nesting_level)
+        @aggregator[nesting_level].is_a?(Array)
       end
 
       def merge_up
-        return if @current_nesting_level == 0
-        previous_nesting_level = @current_nesting_level - 1
-        if @aggregator[previous_nesting_level].kind_of? Array
-          @aggregator[previous_nesting_level] << @aggregator[@current_nesting_level]
+        return if @current_level.zero?
+
+        if array_level?(previous_level)
+          @aggregator[previous_level] << @aggregator[@current_level]
         else
-          @aggregator[previous_nesting_level][@temp_aggregator_keys[previous_nesting_level]] = @aggregator[@current_nesting_level]
+          @aggregator[previous_level][previous_key] = @aggregator[@current_level]
         end
 
-        @aggregator.delete(@current_nesting_level)
-        @aggregator
+        @aggregator.delete(@current_level)
+      end
+
+      def previous_level
+        @current_level - 1
+      end
+
+      def next_level
+        @current_level + 1
+      end
+
+      def previous_key
+        @aggregator_keys[previous_level]
       end
     end
   end

data/lib/json/streamer/version.rb

@@ -1,5 +1,5 @@
 module Json
   module Streamer
-    VERSION = "1.1.1"
+    VERSION = "1.1.2"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: json-streamer
 version: !ruby/object:Gem::Version
-  version: 1.1.1
+  version: 1.1.2
 platform: ruby
 authors:
 - thisismydesign