Don't do type checking when polymorphism will do.

lib/moped/connection.rb

@@ -116,7 +116,7 @@ def read
     # @since 1.0.0
     def receive_replies(operations)
       operations.map do |operation|
-        read if operation.is_a?(Protocol::Query) || operation.is_a?(Protocol::GetMore)
+        operation.receive_replies(self)
       end
     end
 

lib/moped/protocol/command.rb

@@ -18,11 +18,8 @@ def initialize(database, command, options = {})
 
       def log_inspect
         type = "COMMAND"
-
         "%-12s database=%s command=%s" % [type, database, selector.inspect]
       end
-
     end
-
   end
 end

lib/moped/protocol/get_more.rb

@@ -72,9 +72,22 @@ def initialize(database, collection, cursor_id, limit, options = {})
 
       def log_inspect
         type = "GET_MORE"
-
         "%-12s database=%s collection=%s limit=%s cursor_id=%s" % [type, database, collection, limit, cursor_id]
       end
+
+      # Receive replies to the message.
+      #
+      # @example Receive replies.
+      #   message.receive_replies(connection)
+      #
+      # @param [ Connection ] connection The connection.
+      #
+      # @return [ Protocol::Reply ] The reply.
+      #
+      # @since 1.0.0
+      def receive_replies(connection)
+        connection.read
+      end
     end
   end
 end

lib/moped/protocol/message.rb

@@ -283,6 +283,19 @@ def inherited(subclass)
 
       end
 
+      # Default implementation for a message is to do nothing when receiving
+      # replies.
+      #
+      # @example Receive replies.
+      #   message.receive_replies(connection)
+      #
+      # @param [ Connection ] connection The connection.
+      #
+      # @since 1.0.0
+      #
+      # @return [ nil ] nil.
+      def receive_replies(connection); end
+
       # Serializes the message and all of its fields to a new buffer or to the
       # provided buffer.
       #

lib/moped/protocol/query.rb

@@ -111,7 +111,6 @@ def initialize(database, collection, selector, options = {})
 
       def log_inspect
         type = "QUERY"
-
         fields = []
         fields << ["%-12s", type]
         fields << ["database=%s", database]
@@ -121,12 +120,23 @@ def log_inspect
         fields << ["limit=%s", limit.inspect]
         fields << ["skip=%s", skip.inspect]
         fields << ["fields=%s", self.fields.inspect]
-
         f, v = fields.transpose
-
         f.join(" ") % v
       end
 
+      # Receive replies to the message.
+      #
+      # @example Receive replies.
+      #   message.receive_replies(connection)
+      #
+      # @param [ Connection ] connection The connection.
+      #
+      # @return [ Protocol::Reply ] The reply.
+      #
+      # @since 1.0.0
+      def receive_replies(connection)
+        connection.read
+      end
     end
   end
 end