require 'mongo/message/message' require 'mongo/message/opcodes' require 'mongo/util/ordered_hash' module XGen module Mongo module Driver class QueryMessage < Message def initialize(db_name, collection_name, query) super(OP_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 if query.order_by && query.order_by.length > 0 sel = OrderedHash.new sel['query'] = query.selector sel['orderby'] = case query.order_by when String {query.order_by => 1} when Array h = OrderedHash.new query.order_by.each { |ob| h[ob] = 1 } h when Hash # Should be an ordered hash, but this message doesn't care query.order_by else raise "illegal order_by: is a #{query.order_by.class.name}, must be String, Array, Hash, or OrderedHash" end end write_doc(sel) write_doc(query.fields) if query.fields end def first_key(key) @first_key = key end end end end end