File: op_query.rb

package info (click to toggle)
ruby-mongo 2.5.1-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 4,332 kB
  • sloc: ruby: 45,579; makefile: 5
file content (83 lines) | stat: -rw-r--r-- 2,521 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
# Copyright (C) 2015-2017 MongoDB, Inc.
#
# 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
  class Collection
    class View
      module Builder

        # Builds a legacy OP_QUERY specification from options.
        #
        # @since 2.2.0
        class OpQuery
          extend Forwardable

          def_delegators :@view, :cluster, :collection, :database, :filter, :options, :read

          # @return [ BSON::Document ] modifiers The server modifiers.
          attr_reader :modifiers

          # Create the new legacy query builder.
          #
          # @example Create the query builder.
          #   QueryBuilder.new(view)
          #
          # @param [ Collection::View ] view The collection view.
          #
          # @since 2.2.2
          def initialize(view)
            @view = view
            @modifiers = Modifiers.map_server_modifiers(options)
          end

          def specification
            {
              :selector  => requires_special_filter? ? special_filter : filter,
              :read      => read,
              :options   => query_options,
              :db_name   => database.name,
              :coll_name => collection.name
            }
          end

          private

          def query_options
            BSON::Document.new(
              project: options[:projection],
              skip: options[:skip],
              limit: options[:limit],
              flags: Flags.map_flags(options),
              batch_size: options[:batch_size]
            )
          end

          def requires_special_filter?
            !modifiers.empty? || cluster.sharded?
          end

          def read_pref_formatted
            @read_formatted ||= ServerSelector.get(read).to_mongos if read
          end

          def special_filter
            sel = BSON::Document.new(:$query => filter).merge!(modifiers)
            sel[:$readPreference] = read_pref_formatted unless read_pref_formatted.nil?
            sel
          end
        end
      end
    end
  end
end