Contents

module BirdGrinder
  class Tweeter
    class StreamProcessor
      is :loggable
      
      def initialize(parent, stream_name)
        @parent = parent
        @stream_name = stream_name.to_sym
        setup_parser
      end
      
      def receive_chunk(chunk)
        @parser << chunk
      rescue Yajl::ParseError => e
        logger.error "Couldn't parse json: #{e.message}"
      end
      
      def process_stream_item(json)
        return if !json.is_a?(Hash)
        processed = json.to_nash.normalized
        stream_type = lookup_type_for_steam_response(processed)
        case stream_type
        when :delete
          processed = processed[:delete].status
        when :limit
          processed = processed.limit
        end
        processed.stream_type = stream_type
        processed.streaming_source = @stream_name
        @parent.delegate.receive_message(:incoming_stream, processed)
      end
   
      protected
      
      def lookup_type_for_steam_response(response)
        if response.delete?
          :delete
        elsif response.limit?
          :limit
        else
          :tweet
        end
      end
      
      def setup_parser
        @parser = Yajl::Parser.new
        @parser.on_parse_complete = method(:process_stream_item)
      end
      
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
birdgrinder-0.1.1.1	lib/bird_grinder/tweeter/stream_processor.rb
birdgrinder-0.1.1	lib/bird_grinder/tweeter/stream_processor.rb