json-emitter 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 82b399ea5929c610e5f113a35c72b9df9f6cf9957292dc8e2dde5837472484c3
-  data.tar.gz: 9925436be580fa8b772a18e4bad918ffcb3938f638cf0e633db4788c4e1df52c
+  metadata.gz: 8fb1f58b4cbaa70be670062103c939cad985a21b709862771989851795a425c8
+  data.tar.gz: 58aa111fa70c53090f33ccf8103086a0615ea0e4ac5cfca1b5d42f6047c43107
 SHA512:
-  metadata.gz: 85048433a36d8d985d57099c2d0dacc0f5ddb31eb128ef8f030c8136224bd37d101db41ed14aff3f8a59b9c5958aa4e391b4b28598242ff1978d6c65d114f206
-  data.tar.gz: e058dd052d8604acbf8c8516a887dd1672a0018faf4d25c874feacf273655fbd83f10c587982fa028c98b16dd67bccd32d51421ba5a8bb49395096672df74026
+  metadata.gz: 309af9a8b1e04ba28ee14f5a583d70852fc23db51ec6eaa5d14d0590f42381ef1fe502909b5449880fedff08bc4e6920112651ff7ce0f6480ff3c71de5ca7e8c
+  data.tar.gz: 15276de4e33f4d8506736aef13c090ba4c7dcc42a301cecbbf216be58f3e9e514a339f1acfb34ed389f8dbd7c18d92c420477ab47cd90448e461b380cb1c6433

data/lib/json-emitter/emitter.rb

@@ -3,6 +3,13 @@ module JsonEmitter
   # Builds Enumerators that yield JSON from Ruby Arrays or Hashes.
   #
   class Emitter
+    def initialize
+      @wrappers = JsonEmitter.wrappers.map(&:call).compact
+      @error_handlers = JsonEmitter.error_handlers
+      @pass_through_errors = []
+      @pass_through_errors << Puma::ConnectionError if defined? Puma::ConnectionError
+    end
+
     #
     # Generates an Enumerator that will stream out a JSON array.
     #
@@ -11,6 +18,48 @@ module JsonEmitter
     # @return [Enumerator]
     #
     def array(enum, &mapper)
+      Enumerator.new { |y|
+        wrapped {
+          array_generator(enum, &mapper).each { |json_val|
+            y << json_val
+          }
+        }
+      }
+    end
+
+    #
+    # Generates an Enumerator that will stream out a JSON object.
+    #
+    # @param hash [Hash] Keys should be Strings or Symbols and values should be any JSON-compatible value like a number, string, boolean, Array, or Hash.
+    # @return [Enumerator]
+    #
+    def object(hash)
+      Enumerator.new { |y|
+        wrapped {
+          object_generator(hash).each { |json_val|
+            y << json_val
+          }
+        }
+      }
+    end
+
+    # Wrap the enumeration in a block. It will be passed a callback which it must call to continue.
+    # TODO better docs and examples.
+    def wrap(&block)
+      if (wrapper = block.call)
+        @wrappers.unshift wrapper
+      end
+    end
+
+    # Add an error handler.
+    # TODO better docs and examples.
+    def error(&handler)
+      @error_handlers += [handler]
+    end
+
+    private
+
+    def array_generator(enum, &mapper)
       Enumerator.new { |y|
         y << "[".freeze
 
@@ -29,13 +78,7 @@ module JsonEmitter
       }
     end
 
-    #
-    # Generates an Enumerator that will stream out a JSON object.
-    #
-    # @param hash [Hash] Keys should be Strings or Symbols and values should be any JSON-compatible value like a number, string, boolean, Array, or Hash.
-    # @return [Enumerator]
-    #
-    def object(hash)
+    def object_generator(hash)
       Enumerator.new { |y|
         y << "{".freeze
 
@@ -56,17 +99,30 @@ module JsonEmitter
       }
     end
 
-    private
-
     def json_values(x)
       case x
-      when Hash then object x
-      when Enumerable then array x
+      when Hash
+        object_generator x
+      when Enumerable
+        array_generator x
       when Proc
         y = x.call
         json_values y
-      else [MultiJson.dump(x)]
+      else
+        [MultiJson.dump(x)]
       end
     end
+
+    def wrapped(&final)
+      @wrappers.reduce(final) { |f, outer_wrapper|
+        ->() { outer_wrapper.call(f) }
+      }.call
+
+    rescue *@pass_through_errors => e
+      raise e
+    rescue => e
+      @error_handlers.each { |h| h.call(e) }
+      raise e
+    end
   end
 end

data/lib/json-emitter/version.rb

@@ -1,4 +1,4 @@
 module JsonEmitter
   # Library version
-  VERSION = "0.0.1".freeze
+  VERSION = "0.0.2".freeze
 end

data/lib/json-emitter.rb

@@ -16,6 +16,13 @@ require 'json-emitter/buffered_stream'
 # buffered. This works very well with so-called "HTTP chunked responses" in Rack/Rails/Sinatra/Grape/etc.
 #
 module JsonEmitter
+  class << self
+    attr_reader :wrappers
+    attr_reader :error_handlers
+  end
+  @wrappers = []
+  @error_handlers = []
+
   #
   # Generates an stream that will output a JSON array. The input can be any Enumerable, such as an Array or an Enumerator.
   #
@@ -97,4 +104,16 @@ module JsonEmitter
     emitter = Emitter.new.object(hash)
     Stream.new(emitter)
   end
+
+  # Wrap the enumeration in a Proc. It will be passed a callback which it must call to continue.
+  # TODO better docs and examples.
+  def self.wrap(&wrapper)
+    @wrappers.unshift wrapper
+  end
+
+  # Add an error handler.
+  # TODO better docs and examples.
+  def self.error(&handler)
+    @error_handlers << handler
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: json-emitter
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Jordan Hollinger
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-01-29 00:00:00.000000000 Z
+date: 2019-01-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: multi_json