lib/mongo/message/query_message.rb in mongo-0.0.3 vs lib/mongo/message/query_message.rb in mongo-0.0.4

@@ -6,11 +6,14 @@
   module Mongo
     module Driver
 
       class QueryMessage < Message
 
+        attr_reader :query
+
         def initialize(db_name, collection_name, query)
           super(OP_QUERY)
+          @query = query
           write_int(0)
           write_string("#{db_name}.#{collection_name}")
           write_int(query.number_to_skip)
           write_int(query.number_to_return)
           sel = query.selector