mongo-ruby-driver/lib/mongo/util/conversions.rb

90 lines
3.0 KiB
Ruby
Raw Normal View History

# encoding: UTF-8
2009-10-06 23:20:21 +00:00
# --
2011-01-17 17:26:32 +00:00
# Copyright (C) 2008-2011 10gen Inc.
2009-10-06 23:20:21 +00:00
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ++
module Mongo #:nodoc:
# Utility module to include when needing to convert certain types of
# objects to mongo-friendly parameters.
module Conversions
2010-01-08 21:29:06 +00:00
ASCENDING_CONVERSION = ["ascending", "asc", "1"]
DESCENDING_CONVERSION = ["descending", "desc", "-1"]
2009-10-06 23:20:21 +00:00
# Converts the supplied +Array+ to a +Hash+ to pass to mongo as
# sorting parameters. The returned +Hash+ will vary depending
# on whether the passed +Array+ is one or two dimensional.
#
# Example:
#
# <tt>array_as_sort_parameters([["field1", :asc], ["field2", :desc]])</tt> =>
# <tt>{ "field1" => 1, "field2" => -1}</tt>
def array_as_sort_parameters(value)
2010-05-07 01:25:18 +00:00
order_by = BSON::OrderedHash.new
if value.first.is_a? Array
value.each do |param|
if (param.class.name == "String")
order_by[param] = 1
else
order_by[param[0]] = sort_value(param[1]) unless param[1].nil?
end
end
2009-12-29 16:31:59 +00:00
elsif !value.empty?
if order_by.size == 1
order_by[value.first] = 1
2009-10-06 23:20:21 +00:00
else
order_by[value.first] = sort_value(value[1])
2009-10-06 23:20:21 +00:00
end
end
order_by
end
# Converts the supplied +String+ or +Symbol+ to a +Hash+ to pass to mongo as
2009-10-06 23:20:21 +00:00
# a sorting parameter with ascending order. If the +String+
# is empty then an empty +Hash+ will be returned.
#
# Example:
#
# *DEPRECATED
#
# <tt>string_as_sort_parameters("field")</tt> => <tt>{ "field" => 1 }</tt>
# <tt>string_as_sort_parameters("")</tt> => <tt>{}</tt>
def string_as_sort_parameters(value)
return {} if (str = value.to_s).empty?
{ str => 1 }
2009-10-06 23:20:21 +00:00
end
# Converts the +String+, +Symbol+, or +Integer+ to the
# corresponding sort value in MongoDB.
#
# Valid conversions (case-insensitive):
#
# <tt>ascending, asc, :ascending, :asc, 1</tt> => <tt>1</tt>
# <tt>descending, desc, :descending, :desc, -1</tt> => <tt>-1</tt>
#
# If the value is invalid then an error will be raised.
def sort_value(value)
val = value.to_s.downcase
2010-01-08 21:29:06 +00:00
return 1 if ASCENDING_CONVERSION.include?(val)
return -1 if DESCENDING_CONVERSION.include?(val)
2009-10-06 23:20:21 +00:00
raise InvalidSortValueError.new(
2009-10-08 14:02:54 +00:00
"#{self} was supplied as a sort direction when acceptable values are: " +
2010-01-08 21:29:06 +00:00
"Mongo::ASCENDING, 'ascending', 'asc', :ascending, :asc, 1, Mongo::DESCENDING, " +
"'descending', 'desc', :descending, :desc, -1.")
2009-10-06 23:20:21 +00:00
end
end
2009-10-08 14:02:54 +00:00
end