/
mongo.rb
53 lines (42 loc) · 1.24 KB
/
mongo.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
require 'mongo'
class Eventwire::Adapters::Mongo
DB_NAME = 'broker'
def initialize
@handlers = []
end
def publish(event_name, event_data = nil)
collection = db.collection('event_handlers')
collection.find(:event_name => event_name).each do |handler|
queue = db.collection(handler['handler'])
queue.save({:event_data => event_data})
end
end
def subscribe(event_name, handler_id, &handler)
@handlers << [handler_id, handler]
collection = db.collection('event_handlers')
collection.find_and_modify :query => {:handler => handler_id},
:update => {:handler => handler_id, :event_name => event_name},
:upsert => true
end
def start
@started = true
loop do
@handlers.each do |queue_name, handler|
break unless @started
queue = db.collection(queue_name)
if event_data = queue.find_and_modify({:remove => true})
handler.call event_data['event_data']
end
end
end
end
def stop
@started = false
end
def db
@db ||= Mongo::Connection.new('localhost', 27017, :safe => true).db(DB_NAME)
end
def purge
Mongo::Connection.new.drop_database(DB_NAME)
end
end