lib/dripdrop/agent.rb in dripdrop-0.0.2 vs lib/dripdrop/agent.rb in dripdrop-0.0.3

@@ -1,19 +1,23 @@
 require 'dripdrop/message'
 require 'zmq'
 require 'bert'
 
 class DripDrop
+  #The Agent class is a simple ZMQ Pub client. It uses DripDrop::Message messages
   class Agent
     attr_reader :address, :context, :socket
+    
+    #address should be a string like tcp://127.0.0.1
     def initialize(address)
       @address = address
       @context = ZMQ::Context.new(1)
       @socket  = @context.socket(ZMQ::PUB)
       @socket.connect(@address)
     end
 
-    def send_message(name,content)
-      puts @socket.send(Message.new(name,content).encoded, 0)
+    #Sends a DripDrop::Message to the socket
+    def send_message(name,body,head={})
+      puts @socket.send(DripDrop::Message.new(name,:body => body, :head => head).encoded, 0)
     end
   end
 end