Contents

module Sinatra
  module CometIO

    def self.sessions
      @@sessions ||= Hash.new{|h,session_id|
        h[session_id] = {
          :queue => [{:type => :__session_id, :data => session_id}],
          :stream => nil,
          :last => nil
        }
      }
    end

    def self.gc
      self.sessions.each do |id, s|
        next unless s[:last] and s[:last] < Time.now-CometIO.options[:timeout]*2-10
        self.sessions.delete id rescue next
        self.emit :disconnect, id
      end
    end

    EM::defer do
      loop do
        self.gc
        sleep CometIO.options[:timeout]+5
      end
    end

    def self.push(type, data, opt={})
      session_ids = opt[:to].to_s.empty? ? self.sessions.keys : [opt[:to]]
      session_ids.each do |id|
        next unless self.sessions.include? id
        s = self.sessions[id]
        if s[:queue].empty? and s[:stream] != nil
          begin
            s[:stream].write([{:type => type, :data => data}].to_json)
            s[:stream].flush
            s[:stream].close
          rescue
            s[:stream].close
            s[:queue].push :type => type, :data => data
          end
        else
          s[:queue].push :type => type, :data => data
        end
      end
    end

    def self.create_session(ip_addr)
      Digest::MD5.hexdigest "#{Time.now.to_i}_#{Time.now.usec}_#{ip_addr}"
    end

  end
end
EventEmitter.apply Sinatra::CometIO

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
sinatra-cometio-0.4.1	lib/sinatra-cometio/cometio.rb
sinatra-cometio-0.4.0	lib/sinatra-cometio/cometio.rb
sinatra-cometio-0.3.9	lib/sinatra-cometio/cometio.rb
sinatra-cometio-0.3.8	lib/sinatra-cometio/cometio.rb
sinatra-cometio-0.3.7	lib/sinatra-cometio/cometio.rb