Contents

require "oj"

module Oj
  class ArrayParser
    class ParserError < StandardError
    end

    # @param [String, IO] string_or_io json to parse
    # @param [Hash] options parsing options to pass to Oj, see http://www.ohler.com/oj/#label-Options
    # @return [Enumerator] yields parsed top-level entries in the read json document
    def self.enumerator(string_or_io, options = {})
      Enumerator.new do |yielder|
        parser = new { |value| yielder << value }

        Oj.sc_parse(parser, string_or_io, options)
      end
    end

    def initialize(&block)
      @block = block
      @array_count = 0
    end

    def hash_start
      {}
    end

    def hash_set(hash, key, value)
      hash[key] = value
    end

    def array_start
      @array_count += 1

      if @array_count > 1
        []
      end
    end

    def array_end
      @array_count -= 1
    end

    def array_append(array, value)
      if @array_count == 1
        @block.call(value)
      else
        array << value
      end
    end

    def error(message, line, column)
      raise ParserError, "line: #{line}, column: #{column}, message: #{message}"
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
oj-array_parser-0.2.0	lib/oj/array_parser.rb