68 lines
2.4 KiB
Ruby
68 lines
2.4 KiB
Ruby
require 'mongo/message/message'
|
|
require 'mongo/message/opcodes'
|
|
require 'mongo/util/ordered_hash'
|
|
|
|
module XGen
|
|
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
|
|
if query.contains_special_fields
|
|
sel = OrderedHash.new
|
|
sel['query'] = query.selector
|
|
if query.order_by && query.order_by.length > 0
|
|
sel['orderby'] = case query.order_by
|
|
when String
|
|
{query.order_by => 1}
|
|
when Array
|
|
h = OrderedHash.new
|
|
query.order_by.each { |ob|
|
|
case ob
|
|
when String
|
|
h[ob] = 1
|
|
when Hash # should have one entry; will handle all
|
|
ob.each { |k,v| h[k] = v }
|
|
else
|
|
raise "illegal query order_by value #{query.order_by.inspect}"
|
|
end
|
|
}
|
|
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
|
|
if query.hint_fields && query.hint_fields.length > 0
|
|
hints = OrderedHash.new
|
|
query.hint_fields.each { |hf| hints[hf] = 1 }
|
|
sel['$hint'] = hints
|
|
end
|
|
if query.explain
|
|
sel['$explain'] = true
|
|
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
|