/json-streamer

JSON streamer for Ruby allowing you to get data based on various criteria (key, nesting level, etc)

Primary LanguageRubyMIT LicenseMIT

Json::Streamer

Ruby utility that supports JSON streaming allowing you to get data based on various criteria (key, nesting level, etc).

Branch Status
Release Build Status Coverage Status Gem Version Total Downloads
Development Build Status Coverage Status

If you've tried JSON streaming with other Ruby libraries before (e.g. JSON::Stream, Yajl::FFI):

This gem will basically spare you the need to define your own callbacks (i.e. implement an actual JSON parser using start_object, end_object, key, value, etc.).

If you're new to this:

Streaming is useful for

  • big files that do not fit in the memory (or you'd rather avoid the risk)
  • files read in chunks (e.g. arriving over network)
  • cases where you expect some issue with the file (e.g. losing connection to source, invalid data at some point) but would like to get as much data as possible anyway

This gem is aimed at making streaming as easy and convenient as possible.

Performance:

The gem uses JSON::Stream's events in the background. It was chosen because it's a pure Ruby parser. A similar implementation can be done using the ~10 times faster Yajl::FFI gem that is dependent on the native YAJL library. I did not measure the performance of my implementation on top of these libraries.

Installation

Add this line to your application's Gemfile:

gem 'json-streamer'

And then execute:

$ bundle

Or install it yourself as:

$ gem install json-streamer

Usage

Check the unit tests for more examples (spec/streamer_spec.rb).

One streamer object handles one set of conditions. For multiple conditions create multiple streamers. For more details see this discussion.

require 'json/streamer'

v1.2 (and above) API

Passing IO upfront

file_stream = File.open('data.json', 'r')
chunk_size = 500 # defaults to 1000

streamer = Json::Streamer.parser(file_io: file_stream, chunk_size: chunk_size)

Get objects based on nesting level

# Level zero yields the full JSON, first level yields data within the JSON 1-by-1, etc.
streamer.get(nesting_level:1) do |object|
    p object
end

Input:

{
    "object1": "first_level_value",
    "object2": {}
}

Output:

"first_level_value"
{}

Get data based on key

streamer.get(key:'desired_key') do |object|
    p object
end

Input:

{
    "obj1" : {
        "desired_key" : "value1"
    },
    "desired_key" : "value2",
    "obj2" : {
        "desired_key" : {
            "desired_key" : "value3"
        }
    }
}

Output:

"value1"
"value2"
"value3"
{"desired_key" => "value3"}

Skip values

streamer.get(nesting_level:1, yield_values:false) do |object|
    p object
end

Input:

{
    "obj1" : {},
    "key" : "value"
}

Output:

{}

Symbolize keys

Since v1.3.0

streamer.get(nesting_level:0, symbolize_keys: true) do |object|
    p object
end

Input:

{
    "obj1" : {"key" : "value"}
}

Output:

{:obj1=>{:key=>"value"}}

Passing IO later (EventMachine-style)

# Get a JsonStreamer object that provides access to the parser
# but does not start processing immediately
streamer = Json::Streamer.parser

streamer.get(nesting_level:1) do |object|
    p object
end

Then later in your EventMachine handler:

def receive_data(data)
    streamer << data
end

Custom yield conditions

v2.0.0 introduces custom conditions which provide ultimate control over what to yield.

The Conditions API exposes 3 callbacks:

  • yield_value
  • yield_array
  • yield_object

Each of them may be redefined. They are called once the corresponding data (value, array or object) is available. They should return whether the data should be yielded for the outside. They receive the data and the aggregator as parameters.

The aggregator exposes data about the current state of the partly parsed JSON such as:

  • level - current level
  • key - current key
  • value - current value
  • key_for_level(level) - key for custom level
  • value_for_level(level) - value for custom level
  • get - the raw data (in a custom format)

Example usage (inspired by this issue):

conditions = Json::Streamer::Conditions.new
conditions.yield_value = ->(aggregator:, value:) { false }
conditions.yield_array = ->(aggregator:, array:) { false }
conditions.yield_object = lambda do |aggregator:, object:|
    aggregator.level.eql?(2) && aggregator.key_for_level(1).eql?('items1')
end

streamer.get_with_conditions(conditions) do |object|
    p object
end

Input:

{
  "other": "stuff",
  "items1": [
    {
      "key1": "value"
    },
    {
      "key2": "value"
    }
  ],
  "items2": [
    {
      "key3": "value"
    },
    {
      "key4": "value"
    }
  ]
}

Output:

{"key1"=>"value"}
{"key2"=>"value"}

Legacy API (pre-v1.2)

This functionality is deprecated but kept for compatibility reasons.

# Same as Json::Streamer.parser
streamer = Json::Streamer::JsonStreamer.new
# Same as streamer << data
streamer.parser << data

Feedback

Any feedback is much appreciated.

I can only tailor this project to fit use-cases I know about - which are usually my own ones. If you find that this might be the right direction to solve your problem too but you find that it's suboptimal or lacks features don't hesitate to contact me.

Please let me know if you make use of this project so that I can prioritize further efforts.

Conventions

This gem is developed using the following conventions:

Contributing

Bug reports and pull requests are welcome on GitHub at https://github.com/thisismydesign/json-streamer. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the Contributor Covenant code of conduct.

License

The gem is available as open source under the terms of the MIT License.