mongoid 5.4.1 → 6.1.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (260) hide show
  1. checksums.yaml +5 -5
  2. data/README.md +3 -3
  3. data/lib/config/locales/en.yml +19 -0
  4. data/lib/mongoid.rb +4 -4
  5. data/lib/mongoid/atomic.rb +2 -2
  6. data/lib/mongoid/atomic/modifiers.rb +8 -12
  7. data/lib/mongoid/attributes.rb +22 -21
  8. data/lib/mongoid/attributes/readonly.rb +22 -0
  9. data/lib/mongoid/cacheable.rb +36 -0
  10. data/lib/mongoid/changeable.rb +36 -0
  11. data/lib/mongoid/clients.rb +8 -63
  12. data/lib/mongoid/clients/options.rb +55 -250
  13. data/lib/mongoid/clients/storage_options.rb +1 -69
  14. data/lib/mongoid/composable.rb +29 -3
  15. data/lib/mongoid/config.rb +1 -0
  16. data/lib/mongoid/contextual/atomic.rb +5 -8
  17. data/lib/mongoid/contextual/map_reduce.rb +0 -4
  18. data/lib/mongoid/contextual/memory.rb +2 -2
  19. data/lib/mongoid/contextual/mongo.rb +40 -22
  20. data/lib/mongoid/contextual/none.rb +12 -0
  21. data/lib/mongoid/copyable.rb +13 -6
  22. data/lib/mongoid/criteria.rb +5 -2
  23. data/lib/mongoid/criteria/marshalable.rb +2 -2
  24. data/lib/mongoid/criteria/modifiable.rb +17 -1
  25. data/lib/mongoid/criteria/options.rb +25 -0
  26. data/lib/mongoid/criteria/queryable.rb +87 -0
  27. data/lib/mongoid/criteria/queryable/aggregable.rb +120 -0
  28. data/lib/mongoid/criteria/queryable/extensions.rb +28 -0
  29. data/lib/mongoid/criteria/queryable/extensions/array.rb +185 -0
  30. data/lib/mongoid/criteria/queryable/extensions/big_decimal.rb +37 -0
  31. data/lib/mongoid/criteria/queryable/extensions/boolean.rb +34 -0
  32. data/lib/mongoid/criteria/queryable/extensions/date.rb +63 -0
  33. data/lib/mongoid/criteria/queryable/extensions/date_time.rb +53 -0
  34. data/lib/mongoid/criteria/queryable/extensions/hash.rb +200 -0
  35. data/lib/mongoid/criteria/queryable/extensions/nil_class.rb +86 -0
  36. data/lib/mongoid/criteria/queryable/extensions/numeric.rb +90 -0
  37. data/lib/mongoid/criteria/queryable/extensions/object.rb +206 -0
  38. data/lib/mongoid/criteria/queryable/extensions/range.rb +70 -0
  39. data/lib/mongoid/criteria/queryable/extensions/regexp.rb +79 -0
  40. data/lib/mongoid/criteria/queryable/extensions/set.rb +34 -0
  41. data/lib/mongoid/criteria/queryable/extensions/string.rb +137 -0
  42. data/lib/mongoid/criteria/queryable/extensions/symbol.rb +79 -0
  43. data/lib/mongoid/criteria/queryable/extensions/time.rb +60 -0
  44. data/lib/mongoid/criteria/queryable/extensions/time_with_zone.rb +54 -0
  45. data/lib/mongoid/criteria/queryable/forwardable.rb +65 -0
  46. data/lib/mongoid/criteria/queryable/key.rb +103 -0
  47. data/lib/mongoid/criteria/queryable/macroable.rb +27 -0
  48. data/lib/mongoid/criteria/queryable/mergeable.rb +271 -0
  49. data/lib/mongoid/criteria/queryable/optional.rb +429 -0
  50. data/lib/mongoid/criteria/queryable/options.rb +153 -0
  51. data/lib/mongoid/criteria/queryable/pipeline.rb +111 -0
  52. data/lib/mongoid/criteria/queryable/selectable.rb +662 -0
  53. data/lib/mongoid/criteria/queryable/selector.rb +212 -0
  54. data/lib/mongoid/criteria/queryable/smash.rb +104 -0
  55. data/lib/mongoid/document.rb +30 -37
  56. data/lib/mongoid/errors.rb +2 -0
  57. data/lib/mongoid/errors/ambiguous_relationship.rb +1 -1
  58. data/lib/mongoid/errors/in_memory_collation_not_supported.rb +1 -1
  59. data/lib/mongoid/errors/invalid_field.rb +2 -2
  60. data/lib/mongoid/errors/invalid_persistence_option.rb +29 -0
  61. data/lib/mongoid/errors/invalid_relation.rb +66 -0
  62. data/lib/mongoid/evolvable.rb +1 -1
  63. data/lib/mongoid/extensions.rb +0 -4
  64. data/lib/mongoid/extensions/big_decimal.rb +17 -8
  65. data/lib/mongoid/extensions/date.rb +4 -1
  66. data/lib/mongoid/extensions/decimal128.rb +3 -3
  67. data/lib/mongoid/extensions/hash.rb +1 -0
  68. data/lib/mongoid/extensions/string.rb +4 -3
  69. data/lib/mongoid/extensions/time.rb +4 -1
  70. data/lib/mongoid/fields/validators/macro.rb +18 -0
  71. data/lib/mongoid/findable.rb +2 -2
  72. data/lib/mongoid/indexable.rb +15 -13
  73. data/lib/mongoid/interceptable.rb +5 -22
  74. data/lib/mongoid/matchable.rb +13 -7
  75. data/lib/mongoid/matchable/all.rb +2 -2
  76. data/lib/mongoid/matchable/and.rb +3 -3
  77. data/lib/mongoid/matchable/default.rb +2 -2
  78. data/lib/mongoid/matchable/elem_match.rb +28 -0
  79. data/lib/mongoid/matchable/exists.rb +2 -2
  80. data/lib/mongoid/matchable/gt.rb +4 -2
  81. data/lib/mongoid/matchable/gte.rb +4 -2
  82. data/lib/mongoid/matchable/in.rb +2 -2
  83. data/lib/mongoid/matchable/lt.rb +4 -2
  84. data/lib/mongoid/matchable/lte.rb +4 -2
  85. data/lib/mongoid/matchable/ne.rb +2 -2
  86. data/lib/mongoid/matchable/nin.rb +2 -2
  87. data/lib/mongoid/matchable/or.rb +3 -3
  88. data/lib/mongoid/matchable/regexp.rb +3 -3
  89. data/lib/mongoid/matchable/size.rb +2 -2
  90. data/lib/mongoid/persistable.rb +3 -5
  91. data/lib/mongoid/persistable/creatable.rb +2 -2
  92. data/lib/mongoid/persistable/deletable.rb +1 -1
  93. data/lib/mongoid/persistable/settable.rb +1 -1
  94. data/lib/mongoid/persistable/updatable.rb +5 -12
  95. data/lib/mongoid/persistable/upsertable.rb +1 -1
  96. data/lib/mongoid/persistence_context.rb +215 -0
  97. data/lib/mongoid/query_cache.rb +3 -6
  98. data/lib/mongoid/relations/accessors.rb +3 -0
  99. data/lib/mongoid/relations/auto_save.rb +12 -4
  100. data/lib/mongoid/relations/bindings/referenced/many_to_many.rb +4 -4
  101. data/lib/mongoid/relations/counter_cache.rb +15 -5
  102. data/lib/mongoid/relations/eager.rb +6 -11
  103. data/lib/mongoid/relations/eager/base.rb +3 -3
  104. data/lib/mongoid/relations/eager/has_and_belongs_to_many.rb +2 -2
  105. data/lib/mongoid/relations/eager/has_many.rb +1 -1
  106. data/lib/mongoid/relations/embedded/batchable.rb +12 -36
  107. data/lib/mongoid/relations/embedded/in.rb +13 -1
  108. data/lib/mongoid/relations/embedded/many.rb +28 -10
  109. data/lib/mongoid/relations/embedded/one.rb +14 -1
  110. data/lib/mongoid/relations/macros.rb +9 -1
  111. data/lib/mongoid/relations/metadata.rb +3 -3
  112. data/lib/mongoid/relations/options.rb +2 -2
  113. data/lib/mongoid/relations/proxy.rb +1 -31
  114. data/lib/mongoid/relations/referenced/in.rb +19 -10
  115. data/lib/mongoid/relations/referenced/many.rb +23 -17
  116. data/lib/mongoid/relations/referenced/many_to_many.rb +20 -13
  117. data/lib/mongoid/relations/referenced/one.rb +15 -1
  118. data/lib/mongoid/relations/synchronization.rb +11 -11
  119. data/lib/mongoid/relations/touchable.rb +6 -3
  120. data/lib/mongoid/reloadable.rb +1 -1
  121. data/lib/mongoid/serializable.rb +1 -1
  122. data/lib/mongoid/traversable.rb +1 -1
  123. data/lib/mongoid/validatable/uniqueness.rb +1 -2
  124. data/lib/mongoid/version.rb +1 -1
  125. data/lib/rails/generators/mongoid/config/templates/mongoid.yml +14 -3
  126. data/spec/app/models/album.rb +5 -1
  127. data/spec/app/models/artist.rb +21 -0
  128. data/spec/app/models/book.rb +2 -1
  129. data/spec/app/models/dokument.rb +1 -0
  130. data/spec/app/models/ordered_post.rb +5 -0
  131. data/spec/app/models/oscar.rb +1 -2
  132. data/spec/app/models/page.rb +1 -1
  133. data/spec/app/models/person.rb +3 -3
  134. data/spec/app/models/princess.rb +2 -0
  135. data/spec/app/models/record.rb +1 -0
  136. data/spec/app/models/subscription.rb +1 -0
  137. data/spec/app/models/thing.rb +1 -1
  138. data/spec/config/mongoid.yml +15 -0
  139. data/spec/mongoid/atomic/modifiers_spec.rb +17 -17
  140. data/spec/mongoid/atomic_spec.rb +17 -17
  141. data/spec/mongoid/attributes/nested_spec.rb +14 -14
  142. data/spec/mongoid/attributes/readonly_spec.rb +87 -44
  143. data/spec/mongoid/attributes_spec.rb +90 -5
  144. data/spec/mongoid/cacheable_spec.rb +112 -0
  145. data/spec/mongoid/changeable_spec.rb +58 -0
  146. data/spec/mongoid/clients/factory_spec.rb +31 -3
  147. data/spec/mongoid/clients/options_spec.rb +382 -96
  148. data/spec/mongoid/clients_spec.rb +243 -101
  149. data/spec/mongoid/composable_spec.rb +7 -0
  150. data/spec/mongoid/config_spec.rb +67 -11
  151. data/spec/mongoid/contextual/atomic_spec.rb +3 -3
  152. data/spec/mongoid/contextual/mongo_spec.rb +165 -20
  153. data/spec/mongoid/contextual/none_spec.rb +15 -0
  154. data/spec/mongoid/copyable_spec.rb +13 -4
  155. data/spec/mongoid/criteria/modifiable_spec.rb +239 -7
  156. data/spec/mongoid/criteria/options_spec.rb +29 -0
  157. data/spec/mongoid/criteria/queryable/aggregable_spec.rb +370 -0
  158. data/spec/mongoid/criteria/queryable/extensions/array_spec.rb +523 -0
  159. data/spec/mongoid/criteria/queryable/extensions/big_decimal_spec.rb +59 -0
  160. data/spec/mongoid/criteria/queryable/extensions/bignum_spec.rb +58 -0
  161. data/spec/mongoid/criteria/queryable/extensions/boolean_spec.rb +213 -0
  162. data/spec/mongoid/criteria/queryable/extensions/date_spec.rb +330 -0
  163. data/spec/mongoid/criteria/queryable/extensions/date_time_spec.rb +405 -0
  164. data/spec/mongoid/criteria/queryable/extensions/fixnum_spec.rb +58 -0
  165. data/spec/mongoid/criteria/queryable/extensions/float_spec.rb +65 -0
  166. data/spec/mongoid/criteria/queryable/extensions/hash_spec.rb +327 -0
  167. data/spec/mongoid/criteria/queryable/extensions/integer_spec.rb +65 -0
  168. data/spec/mongoid/criteria/queryable/extensions/nil_class_spec.rb +77 -0
  169. data/spec/mongoid/criteria/queryable/extensions/object_spec.rb +108 -0
  170. data/spec/mongoid/criteria/queryable/extensions/range_spec.rb +309 -0
  171. data/spec/mongoid/{extensions/origin → criteria/queryable/extensions}/regexp_raw_spec.rb +2 -2
  172. data/spec/mongoid/criteria/queryable/extensions/regexp_spec.rb +90 -0
  173. data/spec/mongoid/criteria/queryable/extensions/set_spec.rb +39 -0
  174. data/spec/mongoid/criteria/queryable/extensions/string_spec.rb +302 -0
  175. data/spec/mongoid/criteria/queryable/extensions/symbol_spec.rb +167 -0
  176. data/spec/mongoid/criteria/queryable/extensions/time_spec.rb +376 -0
  177. data/spec/mongoid/criteria/queryable/extensions/time_with_zone_spec.rb +347 -0
  178. data/spec/mongoid/criteria/queryable/forwardable_spec.rb +87 -0
  179. data/spec/mongoid/criteria/queryable/key_spec.rb +52 -0
  180. data/spec/mongoid/criteria/queryable/mergeable_spec.rb +49 -0
  181. data/spec/mongoid/criteria/queryable/optional_spec.rb +1799 -0
  182. data/spec/mongoid/criteria/queryable/options_spec.rb +360 -0
  183. data/spec/mongoid/criteria/queryable/pipeline_spec.rb +200 -0
  184. data/spec/mongoid/criteria/queryable/queryable_spec.rb +137 -0
  185. data/spec/mongoid/criteria/queryable/selectable_spec.rb +4174 -0
  186. data/spec/mongoid/criteria/queryable/selector_spec.rb +844 -0
  187. data/spec/mongoid/criteria/queryable/smash_spec.rb +30 -0
  188. data/spec/mongoid/criteria_spec.rb +152 -21
  189. data/spec/mongoid/document_spec.rb +37 -88
  190. data/spec/mongoid/errors/invalid_relation_spec.rb +37 -0
  191. data/spec/mongoid/errors/mongoid_error_spec.rb +6 -3
  192. data/spec/mongoid/extensions/big_decimal_spec.rb +320 -18
  193. data/spec/mongoid/extensions/boolean_spec.rb +14 -0
  194. data/spec/mongoid/extensions/date_spec.rb +2 -6
  195. data/spec/mongoid/extensions/date_time_spec.rb +2 -6
  196. data/spec/mongoid/extensions/decimal128_spec.rb +1 -1
  197. data/spec/mongoid/extensions/float_spec.rb +8 -1
  198. data/spec/mongoid/extensions/hash_spec.rb +15 -0
  199. data/spec/mongoid/extensions/integer_spec.rb +8 -1
  200. data/spec/mongoid/extensions/object_spec.rb +11 -0
  201. data/spec/mongoid/extensions/string_spec.rb +21 -0
  202. data/spec/mongoid/extensions/time_spec.rb +2 -6
  203. data/spec/mongoid/extensions/time_with_zone_spec.rb +2 -6
  204. data/spec/mongoid/findable_spec.rb +46 -1
  205. data/spec/mongoid/indexable_spec.rb +15 -3
  206. data/spec/mongoid/interceptable_spec.rb +68 -10
  207. data/spec/mongoid/matchable/all_spec.rb +4 -4
  208. data/spec/mongoid/matchable/and_spec.rb +10 -10
  209. data/spec/mongoid/matchable/default_spec.rb +12 -12
  210. data/spec/mongoid/matchable/elem_match_spec.rb +86 -0
  211. data/spec/mongoid/matchable/exists_spec.rb +5 -5
  212. data/spec/mongoid/matchable/gt_spec.rb +18 -7
  213. data/spec/mongoid/matchable/gte_spec.rb +17 -7
  214. data/spec/mongoid/matchable/in_spec.rb +5 -5
  215. data/spec/mongoid/matchable/lt_spec.rb +18 -7
  216. data/spec/mongoid/matchable/lte_spec.rb +18 -7
  217. data/spec/mongoid/matchable/ne_spec.rb +5 -5
  218. data/spec/mongoid/matchable/nin_spec.rb +5 -5
  219. data/spec/mongoid/matchable/or_spec.rb +7 -7
  220. data/spec/mongoid/matchable/regexp_spec.rb +5 -5
  221. data/spec/mongoid/matchable/size_spec.rb +3 -3
  222. data/spec/mongoid/matchable_spec.rb +173 -53
  223. data/spec/mongoid/persistable/creatable_spec.rb +7 -2
  224. data/spec/mongoid/persistable/deletable_spec.rb +16 -1
  225. data/spec/mongoid/persistable/destroyable_spec.rb +6 -2
  226. data/spec/mongoid/persistable/savable_spec.rb +35 -30
  227. data/spec/mongoid/persistable/settable_spec.rb +45 -29
  228. data/spec/mongoid/persistable/updatable_spec.rb +184 -5
  229. data/spec/mongoid/persistence_context_spec.rb +680 -0
  230. data/spec/mongoid/positional_spec.rb +10 -10
  231. data/spec/mongoid/query_cache_spec.rb +89 -0
  232. data/spec/mongoid/relations/accessors_spec.rb +1 -1
  233. data/spec/mongoid/relations/auto_save_spec.rb +39 -6
  234. data/spec/mongoid/relations/bindings/referenced/many_to_many_spec.rb +4 -4
  235. data/spec/mongoid/relations/builders_spec.rb +37 -10
  236. data/spec/mongoid/relations/counter_cache_spec.rb +64 -3
  237. data/spec/mongoid/relations/eager/has_and_belongs_to_many_spec.rb +16 -0
  238. data/spec/mongoid/relations/eager_spec.rb +40 -0
  239. data/spec/mongoid/relations/embedded/many_spec.rb +63 -47
  240. data/spec/mongoid/relations/embedded/one_spec.rb +2 -1
  241. data/spec/mongoid/relations/macros_spec.rb +395 -7
  242. data/spec/mongoid/relations/metadata_spec.rb +15 -1
  243. data/spec/mongoid/relations/proxy_spec.rb +27 -1
  244. data/spec/mongoid/relations/referenced/in_spec.rb +41 -1
  245. data/spec/mongoid/relations/referenced/many_spec.rb +13 -25
  246. data/spec/mongoid/relations/referenced/many_to_many_spec.rb +14 -26
  247. data/spec/mongoid/relations/synchronization_spec.rb +48 -2
  248. data/spec/mongoid/relations/touchable_spec.rb +40 -0
  249. data/spec/mongoid/reloadable_spec.rb +51 -0
  250. data/spec/mongoid/serializable_spec.rb +0 -50
  251. data/spec/mongoid/validatable/presence_spec.rb +1 -1
  252. data/spec/mongoid/validatable/uniqueness_spec.rb +18 -9
  253. data/spec/mongoid/validatable_spec.rb +16 -0
  254. data/spec/spec_helper.rb +20 -11
  255. metadata +524 -469
  256. checksums.yaml.gz.sig +0 -0
  257. data.tar.gz.sig +0 -0
  258. data/lib/mongoid/clients/thread_options.rb +0 -19
  259. data/lib/mongoid/extensions/origin/regexp_raw.rb +0 -43
  260. metadata.gz.sig +0 -0
@@ -0,0 +1,429 @@
1
+ # encoding: utf-8
2
+ module Mongoid
3
+ class Criteria
4
+ module Queryable
5
+
6
+ # The optional module includes all behaviour that has to do with extra
7
+ # options surrounding queries, like skip, limit, sorting, etc.
8
+ module Optional
9
+ extend Macroable
10
+
11
+ # @attribute [rw] options The query options.
12
+ attr_accessor :options
13
+
14
+ # Add ascending sorting options for all the provided fields.
15
+ #
16
+ # @example Add ascending sorting.
17
+ # optional.ascending(:first_name, :last_name)
18
+ #
19
+ # @param [ Array<Symbol> ] fields The fields to sort.
20
+ #
21
+ # @return [ Optional ] The cloned optional.
22
+ #
23
+ # @since 1.0.0
24
+ def ascending(*fields)
25
+ sort_with_list(*fields, 1)
26
+ end
27
+ alias :asc :ascending
28
+ key :asc, :override, 1
29
+ key :ascending, :override, 1
30
+
31
+ # Adds the option for telling MongoDB how many documents to retrieve in
32
+ # it's batching.
33
+ #
34
+ # @example Apply the batch size options.
35
+ # optional.batch_size(500)
36
+ #
37
+ # @param [ Integer ] value The batch size.
38
+ #
39
+ # @return [ Optional ] The cloned optional.
40
+ #
41
+ # @since 1.0.0
42
+ def batch_size(value = nil)
43
+ option(value) { |options| options.store(:batch_size, value) }
44
+ end
45
+
46
+ # Add descending sorting options for all the provided fields.
47
+ #
48
+ # @example Add descending sorting.
49
+ # optional.descending(:first_name, :last_name)
50
+ #
51
+ # @param [ Array<Symbol> ] fields The fields to sort.
52
+ #
53
+ # @return [ Optional ] The cloned optional.
54
+ #
55
+ # @since 1.0.0
56
+ def descending(*fields)
57
+ sort_with_list(*fields, -1)
58
+ end
59
+ alias :desc :descending
60
+ key :desc, :override, -1
61
+ key :descending, :override, -1
62
+
63
+ # Add an index hint to the query options.
64
+ #
65
+ # @example Add an index hint.
66
+ # optional.hint("$natural" => 1)
67
+ #
68
+ # @param [ Hash ] value The index hint.
69
+ #
70
+ # @return [ Optional ] The cloned optional.
71
+ #
72
+ # @since 1.0.0
73
+ def hint(value = nil)
74
+ option(value) { |options| options.store(:hint, value) }
75
+ end
76
+
77
+ # Add the number of documents to limit in the returned results.
78
+ #
79
+ # @example Limit the number of returned documents.
80
+ # optional.limit(20)
81
+ #
82
+ # @param [ Integer ] value The number of documents to return.
83
+ #
84
+ # @return [ Optional ] The cloned optional.
85
+ #
86
+ # @since 1.0.0
87
+ def limit(value = nil)
88
+ option(value) do |options, query|
89
+ val = value.to_i
90
+ options.store(:limit, val)
91
+ query.pipeline.push("$limit" => val) if aggregating?
92
+ end
93
+ end
94
+
95
+ # Adds the option to limit the number of documents scanned in the
96
+ # collection.
97
+ #
98
+ # @example Add the max scan limit.
99
+ # optional.max_scan(1000)
100
+ #
101
+ # @param [ Integer ] value The max number of documents to scan.
102
+ #
103
+ # @return [ Optional ] The cloned optional.
104
+ #
105
+ # @since 1.0.0
106
+ def max_scan(value = nil)
107
+ option(value) { |options| options.store(:max_scan, value) }
108
+ end
109
+
110
+ # Adds a cumulative time limit in milliseconds for processing operations on a cursor.
111
+ #
112
+ # @example Add the max time ms option.
113
+ # optional.max_time_ms(200)
114
+ #
115
+ # @param [ Integer ] value The max time in milliseconds for processing operations on a cursor.
116
+ #
117
+ # @return [ Optional ] The cloned optional.
118
+ #
119
+ # @since 6.0.0
120
+ def max_time_ms(value = nil)
121
+ option(value) { |options| options.store(:max_time_ms, value) }
122
+ end
123
+
124
+ # Tell the query not to timeout.
125
+ #
126
+ # @example Tell the query not to timeout.
127
+ # optional.no_timeout
128
+ #
129
+ # @return [ Optional ] The cloned optional.
130
+ #
131
+ # @since 1.0.0
132
+ def no_timeout
133
+ clone.tap { |query| query.options.store(:timeout, false) }
134
+ end
135
+
136
+ # Limits the results to only contain the fields provided.
137
+ #
138
+ # @example Limit the results to the provided fields.
139
+ # optional.only(:name, :dob)
140
+ #
141
+ # @param [ Array<Symbol> ] args The fields to return.
142
+ #
143
+ # @return [ Optional ] The cloned optional.
144
+ #
145
+ # @since 1.0.0
146
+ def only(*args)
147
+ args = args.flatten
148
+ option(*args) do |options|
149
+ options.store(
150
+ :fields,
151
+ args.inject(options[:fields] || {}){ |sub, field| sub.tap { sub[field] = 1 }},
152
+ false
153
+ )
154
+ end
155
+ end
156
+
157
+ # Adds sorting criterion to the options.
158
+ #
159
+ # @example Add sorting options via a hash with integer directions.
160
+ # optional.order_by(name: 1, dob: -1)
161
+ #
162
+ # @example Add sorting options via a hash with symbol directions.
163
+ # optional.order_by(name: :asc, dob: :desc)
164
+ #
165
+ # @example Add sorting options via a hash with string directions.
166
+ # optional.order_by(name: "asc", dob: "desc")
167
+ #
168
+ # @example Add sorting options via an array with integer directions.
169
+ # optional.order_by([[ name, 1 ], [ dob, -1 ]])
170
+ #
171
+ # @example Add sorting options via an array with symbol directions.
172
+ # optional.order_by([[ name, :asc ], [ dob, :desc ]])
173
+ #
174
+ # @example Add sorting options via an array with string directions.
175
+ # optional.order_by([[ name, "asc" ], [ dob, "desc" ]])
176
+ #
177
+ # @example Add sorting options with keys.
178
+ # optional.order_by(:name.asc, :dob.desc)
179
+ #
180
+ # @example Add sorting options via a string.
181
+ # optional.order_by("name ASC, dob DESC")
182
+ #
183
+ # @param [ Array, Hash, String ] spec The sorting specification.
184
+ #
185
+ # @return [ Optional ] The cloned optional.
186
+ #
187
+ # @since 1.0.0
188
+ def order_by(*spec)
189
+ option(spec) do |options, query|
190
+ spec.compact.each do |criterion|
191
+ criterion.__sort_option__.each_pair do |field, direction|
192
+ add_sort_option(options, field, direction)
193
+ end
194
+ query.pipeline.push("$sort" => options[:sort]) if aggregating?
195
+ end
196
+ end
197
+ end
198
+ alias :order :order_by
199
+
200
+ # Instead of merging the order criteria, use this method to completely
201
+ # replace the existing ordering with the provided.
202
+ #
203
+ # @example Replace the ordering.
204
+ # optional.reorder(name: :asc)
205
+ #
206
+ # @param [ Array, Hash, String ] spec The sorting specification.
207
+ #
208
+ # @return [ Optional ] The cloned optional.
209
+ #
210
+ # @since 2.1.0
211
+ def reorder(*spec)
212
+ clone.tap do |query|
213
+ query.options.delete(:sort)
214
+ end.order_by(*spec)
215
+ end
216
+
217
+ # Add the number of documents to skip.
218
+ #
219
+ # @example Add the number to skip.
220
+ # optional.skip(100)
221
+ #
222
+ # @param [ Integer ] value The number to skip.
223
+ #
224
+ # @return [ Optional ] The cloned optional.
225
+ #
226
+ # @since 1.0.0
227
+ def skip(value = nil)
228
+ option(value) do |options, query|
229
+ val = value.to_i
230
+ options.store(:skip, val)
231
+ query.pipeline.push("$skip" => val) if aggregating?
232
+ end
233
+ end
234
+ alias :offset :skip
235
+
236
+ # Limit the returned results via slicing embedded arrays.
237
+ #
238
+ # @example Slice the returned results.
239
+ # optional.slice(aliases: [ 0, 5 ])
240
+ #
241
+ # @param [ Hash ] criterion The slice options.
242
+ #
243
+ # @return [ Optional ] The cloned optional.
244
+ #
245
+ # @since 1.0.0
246
+ def slice(criterion = nil)
247
+ option(criterion) do |options|
248
+ options.__union__(
249
+ fields: criterion.inject({}) do |option, (field, val)|
250
+ option.tap { |opt| opt.store(field, { "$slice" => val }) }
251
+ end
252
+ )
253
+ end
254
+ end
255
+
256
+ # Tell the query to operate in snapshot mode.
257
+ #
258
+ # @example Add the snapshot option.
259
+ # optional.snapshot
260
+ #
261
+ # @return [ Optional ] The cloned optional.
262
+ #
263
+ # @since 1.0.0
264
+ def snapshot
265
+ clone.tap do |query|
266
+ query.options.store(:snapshot, true)
267
+ end
268
+ end
269
+
270
+ # Limits the results to only contain the fields not provided.
271
+ #
272
+ # @example Limit the results to the fields not provided.
273
+ # optional.without(:name, :dob)
274
+ #
275
+ # @param [ Array<Symbol> ] args The fields to ignore.
276
+ #
277
+ # @return [ Optional ] The cloned optional.
278
+ #
279
+ # @since 1.0.0
280
+ def without(*args)
281
+ args = args.flatten
282
+ option(*args) do |options|
283
+ options.store(
284
+ :fields,
285
+ args.inject(options[:fields] || {}){ |sub, field| sub.tap { sub[field] = 0 }},
286
+ false
287
+ )
288
+ end
289
+ end
290
+
291
+ # Associate a comment with the query.
292
+ #
293
+ # @example Add a comment.
294
+ # optional.comment('slow query')
295
+ #
296
+ # @note Set profilingLevel to 2 and the comment will be logged in the profile
297
+ # collection along with the query.
298
+ #
299
+ # @param [ String ] comment The comment to be associated with the query.
300
+ #
301
+ # @return [ Optional ] The cloned optional.
302
+ #
303
+ # @since 2.2.0
304
+ def comment(comment = nil)
305
+ clone.tap do |query|
306
+ query.options.store(:comment, comment)
307
+ end
308
+ end
309
+
310
+ # Set the cursor type.
311
+ #
312
+ # @example Set the cursor type.
313
+ # optional.cursor_type(:tailable)
314
+ # optional.cursor_type(:tailable_await)
315
+ #
316
+ # @note The cursor can be type :tailable or :tailable_await.
317
+ #
318
+ # @param [ Symbol ] type The type of cursor to create.
319
+ #
320
+ # @return [ Optional ] The cloned optional.
321
+ #
322
+ # @since 2.2.0
323
+ def cursor_type(type)
324
+ clone.tap { |query| query.options.store(:cursor_type, type) }
325
+ end
326
+
327
+ # Set the collation.
328
+ #
329
+ # @example Set the collation.
330
+ # optional.collation(locale: 'fr', strength: 2)
331
+ #
332
+ # @param [ Hash ] collation_doc The document describing the collation to use.
333
+ #
334
+ # @return [ Optional ] The cloned optional.
335
+ #
336
+ # @since 6.1.0
337
+ def collation(collation_doc)
338
+ clone.tap { |query| query.options.store(:collation, collation_doc) }
339
+ end
340
+
341
+ private
342
+
343
+ # Add a single sort option.
344
+ #
345
+ # @api private
346
+ #
347
+ # @example Add a single sort option.
348
+ # optional.add_sort_option({}, :name, 1)
349
+ #
350
+ # @param [ Hash ] options The options.
351
+ # @param [ String ] field The field name.
352
+ # @param [ Integer ] direction The sort direction.
353
+ #
354
+ # @return [ Optional ] The cloned optional.
355
+ #
356
+ # @since 1.0.0
357
+ def add_sort_option(options, field, direction)
358
+ if driver == :mongo1x
359
+ sorting = (options[:sort] || []).dup
360
+ sorting.push([ field, direction ])
361
+ options.store(:sort, sorting)
362
+ else
363
+ sorting = (options[:sort] || {}).dup
364
+ sorting[field] = direction
365
+ options.store(:sort, sorting)
366
+ end
367
+ end
368
+
369
+ # Take the provided criterion and store it as an option in the query
370
+ # options.
371
+ #
372
+ # @api private
373
+ #
374
+ # @example Store the option.
375
+ # optional.option({ skip: 10 })
376
+ #
377
+ # @param [ Array ] args The options.
378
+ #
379
+ # @return [ Queryable ] The cloned queryable.
380
+ #
381
+ # @since 1.0.0
382
+ def option(*args)
383
+ clone.tap do |query|
384
+ unless args.compact.empty?
385
+ yield(query.options, query)
386
+ end
387
+ end
388
+ end
389
+
390
+ # Add multiple sort options at once.
391
+ #
392
+ # @api private
393
+ #
394
+ # @example Add multiple sort options.
395
+ # optional.sort_with_list(:name, :dob, 1)
396
+ #
397
+ # @param [ Array<String> ] fields The field names.
398
+ # @param [ Integer ] direction The sort direction.
399
+ #
400
+ # @return [ Optional ] The cloned optional.
401
+ #
402
+ # @since 1.0.0
403
+ def sort_with_list(*fields, direction)
404
+ option(fields) do |options, query|
405
+ fields.flatten.compact.each do |field|
406
+ add_sort_option(options, field, direction)
407
+ end
408
+ query.pipeline.push("$sort" => options[:sort]) if aggregating?
409
+ end
410
+ end
411
+
412
+ class << self
413
+
414
+ # Get the methods on the optional that can be forwarded to from a model.
415
+ #
416
+ # @example Get the forwardable methods.
417
+ # Optional.forwardables
418
+ #
419
+ # @return [ Array<Symbol> ] The names of the forwardable methods.
420
+ #
421
+ # @since 1.0.0
422
+ def forwardables
423
+ public_instance_methods(false) - [ :options, :options= ]
424
+ end
425
+ end
426
+ end
427
+ end
428
+ end
429
+ end
@@ -0,0 +1,153 @@
1
+ # encoding: utf-8
2
+ module Mongoid
3
+ class Criteria
4
+ module Queryable
5
+
6
+ # The options is a hash representation of options passed to MongoDB queries,
7
+ # such as skip, limit, and sorting criteria.
8
+ class Options < Smash
9
+
10
+ # Convenience method for getting the field options.
11
+ #
12
+ # @example Get the fields options.
13
+ # options.fields
14
+ #
15
+ # @return [ Hash ] The fields options.
16
+ #
17
+ # @since 1.0.0
18
+ def fields
19
+ self[:fields]
20
+ end
21
+
22
+ # Convenience method for getting the limit option.
23
+ #
24
+ # @example Get the limit option.
25
+ # options.limit
26
+ #
27
+ # @return [ Integer ] The limit option.
28
+ #
29
+ # @since 1.0.0
30
+ def limit
31
+ self[:limit]
32
+ end
33
+
34
+ # Convenience method for getting the skip option.
35
+ #
36
+ # @example Get the skip option.
37
+ # options.skip
38
+ #
39
+ # @return [ Integer ] The skip option.
40
+ #
41
+ # @since 1.0.0
42
+ def skip
43
+ self[:skip]
44
+ end
45
+
46
+ # Convenience method for getting the sort options.
47
+ #
48
+ # @example Get the sort options.
49
+ # options.sort
50
+ #
51
+ # @return [ Hash ] The sort options.
52
+ #
53
+ # @since 1.0.0
54
+ def sort
55
+ self[:sort]
56
+ end
57
+
58
+ # Store the value in the options for the provided key. The options will
59
+ # handle all necessary serialization and localization in this step.
60
+ #
61
+ # @example Store a value in the options.
62
+ # options.store(:key, "testing")
63
+ #
64
+ # @param [ String, Symbol ] key The name of the attribute.
65
+ # @param [ Object ] value The value to add.
66
+ #
67
+ # @return [ Object ] The stored object.
68
+ #
69
+ # @since 1.0.0
70
+ def store(key, value, localize = true)
71
+ super(key, evolve(value, localize))
72
+ end
73
+ alias :[]= :store
74
+
75
+ # Convert the options to aggregation pipeline friendly options.
76
+ #
77
+ # @example Convert the options to a pipeline.
78
+ # options.to_pipeline
79
+ #
80
+ # @return [ Array<Hash> ] The options in pipeline form.
81
+ #
82
+ # @since 2.0.0
83
+ def to_pipeline
84
+ pipeline = []
85
+ pipeline.push({ "$skip" => skip }) if skip
86
+ pipeline.push({ "$limit" => limit }) if limit
87
+ pipeline.push({ "$sort" => sort }) if sort
88
+ pipeline
89
+ end
90
+
91
+ # Perform a deep copy of the options.
92
+ #
93
+ # @example Perform a deep copy.
94
+ # options.__deep_copy__
95
+ #
96
+ # @return [ Options ] The copied options.
97
+ #
98
+ # @since 6.1.1
99
+ def __deep_copy__
100
+ self.class.new(aliases, serializers) do |copy|
101
+ each_pair do |key, value|
102
+ copy.merge!(key => value.__deep_copy__)
103
+ end
104
+ end
105
+ end
106
+
107
+ private
108
+
109
+ # Evolve a single key selection with various types of values.
110
+ #
111
+ # @api private
112
+ #
113
+ # @example Evolve a simple selection.
114
+ # options.evolve(field, 5)
115
+ #
116
+ # @param [ Object ] value The value to serialize.
117
+ #
118
+ # @return [ Object ] The serialized object.
119
+ #
120
+ # @since 1.0.0
121
+ def evolve(value, localize = true)
122
+ case value
123
+ when Hash
124
+ evolve_hash(value, localize)
125
+ else
126
+ value
127
+ end
128
+ end
129
+
130
+ # Evolve a single key selection with hash values.
131
+ #
132
+ # @api private
133
+ #
134
+ # @example Evolve a simple selection.
135
+ # options.evolve(field, { "$gt" => 5 })
136
+ #
137
+ # @param [ Hash ] value The hash to serialize.
138
+ #
139
+ # @return [ Object ] The serialized hash.
140
+ #
141
+ # @since 1.0.0
142
+ def evolve_hash(value, localize = true)
143
+ value.inject({}) do |hash, (field, _value)|
144
+ name, serializer = storage_pair(field)
145
+ name = localized_key(name, serializer) if localize
146
+ hash[name] = _value
147
+ hash
148
+ end
149
+ end
150
+ end
151
+ end
152
+ end
153
+ end