Normalizes API

thisismydesign · Sep 17, 2017 · 1ff80b9 · 1ff80b9
1 parent 45fe248
commit 1ff80b9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/lib/json/streamer/callbacks.rb b/lib/json/streamer/callbacks.rb
@@ -30,6 +30,16 @@ def value(value)
         end
       end
 
+      def end_object
+        end_level { |obj| yield obj }
+      end
+
+      def end_array
+        end_level { |obj| yield obj }
+      end
+
+      private
+
       def end_level
         data = @aggregator.last[:data].clone
 
@@ -43,8 +53,6 @@ def end_level
         end
       end
 
-      private
-
       def add_value(value)
         if array_level?(@current_level)
           @aggregator[@current_level][:data] << value

diff --git a/lib/json/streamer/json_streamer.rb b/lib/json/streamer/json_streamer.rb
@@ -33,11 +33,11 @@ def get(nesting_level: -1, key: nil, yield_values: true, symbolize_keys: false)
         end
 
         @parser.end_object do
-          @callbacks.end_level { |desired_object| yield desired_object }
+          @callbacks.end_object { |desired_object| yield desired_object }
         end
 
         @parser.end_array do
-          @callbacks.end_level { |desired_object| yield desired_object }
+          @callbacks.end_array { |desired_object| yield desired_object }
         end
 
         @file_io.each(@chunk_size) { |chunk| @parser << chunk } if @file_io