drip-drop.rb

DripDrop::Node.new do |node|
  ###
  ### ZMQ Forwarder
  ###
  fwd_pub = node.zmq_publish(FORWARDER_OUT)
  node.zmq_subscribe(FORWARDER_IN,:socket_ctype => :bind).on_recv_raw do |message|
    fwd_pub.send_message(message)
    print 'f'
  end

  ###
  ### Persist all data to a MongoDB Instance
  ###
  db = EM::Mongo::Connection.new.db('dripdrop')
  collection = db.collection('stats')
  node.zmq_subscribe(FORWARDER_OUT).on_recv do |message|
    hash = collection.insert(Hash.new)
    print 'm'
  end
  
  ###
  ###  Broadcast a ZMQ sub socket out to web sockets
  ###
  node.websocket(WEBSOCKET_ADDR).on_open {|ws|
    #This actually isn't the most efficient way to do this, normally
    #you'd do one sub socket per process + node.recv_internal,
    #but this is here for fun
    node.zmq_subscribe(FORWARDER_OUT).on_recv {|message|
      print 'w'
      ws.send(message.to_hash.to_json)
    }
  }
end