add subscribe and unsubscribe to sub sockets

celluloid · May 22, 2012 · bab03fc · bab03fc
1 parent dff9d2b
commit bab03fc
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/lib/celluloid/zmq/sockets.rb b/lib/celluloid/zmq/sockets.rb
@@ -140,6 +140,18 @@ class SubSocket < Socket
       def initialize
         super :sub
       end
+
+      def subscribe(topic)
+        unless ::ZMQ::Util.resultcode_ok? @socket.setsockopt(::ZMQ::SUBSCRIBE, topic)
+          raise IOError, "couldn't set subscribe: #{::ZMQ::Util.error_string}"
+        end
+      end
+
+      def unsubscribe(topic)
+        unless ::ZMQ::Util.resultcode_ok? @socket.setsockopt(::ZMQ::UNSUBSCRIBE, topic)
+          raise IOError, "couldn't set unsubscribe: #{::ZMQ::Util.error_string}"
+        end
+      end
     end
   end
 end