git.cworth.org Git - sup/blob - lib/sup/poll.rb

   1 require 'thread'
   2
   3 module Redwood
   4
   5 class PollManager
   6   include Singleton
   7
   8   DELAY = 300
   9
  10   def initialize
  11     @mutex = Mutex.new
  12     @last_poll = nil
  13
  14     self.class.i_am_the_instance self
  15   end
  16
  17   def buffer
  18     BufferManager.spawn_unless_exists("<poll for new messages>", :hidden => true) { PollMode.new }
  19   end
  20
  21   def poll
  22     BufferManager.flash "Polling for new messages..."
  23     num, numi = buffer.mode.poll
  24     if num > 0
  25       BufferManager.flash "Loaded #{num} new messages, #{numi} to inbox."
  26     end
  27     [num, numi]
  28   end
  29
  30   def start_thread
  31     Redwood::reporting_thread do
  32       while true
  33         sleep DELAY / 2
  34         poll if @last_poll.nil? || (Time.now - @last_poll) >= DELAY
  35       end
  36     end
  37   end
  38
  39   def do_poll
  40     total_num = total_numi = 0
  41     @mutex.synchronize do
  42       found = {}
  43       Index.usual_sources.each do |source|
  44         yield "Loading from #{source}... " unless source.done? || source.broken?
  45         num = 0
  46         numi = 0
  47         Index.add_new_messages_from source do |m, offset, source_labels, entry|
  48           yield "Found message at #{offset} with labels #{m.labels * ', '}"
  49           num += 1
  50           numi += 1 if m.labels.include? :inbox
  51           m
  52         end
  53         yield "Found #{num} messages, #{numi} to inbox" unless num == 0
  54         total_num += num
  55         total_numi += numi
  56       end
  57
  58       yield "Done polling; loaded #{total_num} new messages total"
  59       @last_poll = Time.now
  60       @polling = false
  61     end
  62     [total_num, total_numi]
  63   end
  64 end
  65
  66 end