Module: Angelo::Stash

Includes:
Celluloid::Logger
Included in:
SSE, Websocket
Defined in:
lib/angelo/stash.rb

Overview

utility class for stashing connected websockets in arbitrary contexts

Defined Under Namespace

Modules: ClassMethods Classes: SSE, Websocket

Instance Method Summary collapse

Instance Method Details

#<<(s) ⇒ Object

add a websocket to this context’s stash, save peeraddr info, start server handle_websocket task to read from the socket and fire events as needed



39
40
41
42
 
# File 'lib/angelo/stash.rb', line 39

def << s
  peeraddrs[s] = s.peeraddr
  stashes[@context] << s
end

#[](context) ⇒ Object

utility method to create a new instance with a different context



78
79
80
 
# File 'lib/angelo/stash.rb', line 78

def [] context
  self.class.new @server, context
end

#all_each ⇒ Object

iterate on every connected websocket in all contexts, mostly used for ping_websockets task



85
86
87
88
89
90
91
92
93
94
 
# File 'lib/angelo/stash.rb', line 85

def all_each
  stashes.values.flatten.each do |s|
    begin
      yield s
    rescue Reel::SocketError, IOError, SystemCallError => e
      debug "all - #{e.message}"
      remove_socket s
    end
  end
end

#each(&block) ⇒ Object

iterate on each connected websocket in this context, handling errors as needed



53
54
55
56
57
58
59
60
61
62
63
 
# File 'lib/angelo/stash.rb', line 53

def each &block
  stash.dup.each do |s|
    begin
      yield s
    rescue Reel::SocketError, IOError, SystemCallError => e
      debug "context: #{@context} - #{e.message}"
      remove_socket s
    end
  end
  nil
end

#initialize(server, context = :default) ⇒ Object

create a new instance with a context, creating the array if needed



29
30
31
32
33
 
# File 'lib/angelo/stash.rb', line 29

def initialize server, context = :default
  raise ArgumentError.new "symbol required" unless Symbol === context
  @context, @server = context, server
  stashes[@context] ||= []
end

#length ⇒ Object

return the number of websockets in this context (some are potentially disconnected)



111
112
113
 
# File 'lib/angelo/stash.rb', line 111

def length
  stash.length
end

#peeraddr(s) ⇒ Object

access the peeraddr info for a given websocket



104
105
106
 
# File 'lib/angelo/stash.rb', line 104

def peeraddr s
  peeraddrs[s]
end

#peeraddrs ⇒ Object 



25
 
# File 'lib/angelo/stash.rb', line 25

def peeraddrs; self.class.peeraddrs; end

#reject!(&block) ⇒ Object

pass the given block to the underlying stashed array’s reject! method



98
99
100
 
# File 'lib/angelo/stash.rb', line 98

def reject! &block
  stash.reject! &block
end

#remove_socket(s) ⇒ Object

remove a socket from the stash, warn user, drop peeraddr info



67
68
69
70
71
72
73
74
 
# File 'lib/angelo/stash.rb', line 67

def remove_socket s
  s.close unless s.closed?
  if stash.include? s
    warn "removing socket from context ':#{@context}' (#{peeraddrs[s][2]})"
    stash.delete s
    peeraddrs.delete s
  end
end

#stash ⇒ Object

access the underlying array of this context



46
47
48
 
# File 'lib/angelo/stash.rb', line 46

def stash
  stashes[@context]
end

#stashes ⇒ Object 



24
 
# File 'lib/angelo/stash.rb', line 24

def stashes; self.class.stashes; end