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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
# frozen_string_literal: true
# rubocop:todo all
# Copyright (C) 2014-2020 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 Cluster
# Defines behavior for getting servers.
#
# Topologies are associated with their clusters - for example, a
# ReplicaSet topology contains the replica set name. A topology
# object therefore cannot be used with multiple cluster objects.
#
# At the same time, topology objects do not know anything about
# specific *servers* in a cluster, despite what their constructor
# may suggest. Which means, in particular, that topology change events
# require the application to maintain cluster references on its own
# if it wants to track server changes within a replica set.
#
# @since 2.0.0
module Topology
extend self
end
end
end
require 'mongo/cluster/topology/base'
require 'mongo/cluster/topology/no_replica_set_options'
require 'mongo/cluster/topology/load_balanced'
require 'mongo/cluster/topology/replica_set_no_primary'
require 'mongo/cluster/topology/replica_set_with_primary'
require 'mongo/cluster/topology/sharded'
require 'mongo/cluster/topology/single'
require 'mongo/cluster/topology/unknown'
module Mongo
class Cluster
module Topology
# The various topologies for server selection.
#
# @since 2.0.0
# @api private
OPTIONS = {
direct: Single,
load_balanced: LoadBalanced,
replica_set: ReplicaSetNoPrimary,
sharded: Sharded,
}.freeze
# Get the initial cluster topology for the provided options.
#
# @example Get the initial cluster topology.
# Topology.initial(topology: :replica_set)
#
# @param [ Cluster ] cluster The cluster.
# @param [ Monitoring ] monitoring The monitoring.
# @param [ Hash ] options The cluster options.
#
# @option options [ true | false ] :direct_connection Whether to connect
# directly to the specified seed, bypassing topology discovery. Exactly
# one seed must be provided.
# @option options [ Symbol ] :connect Deprecated - use :direct_connection
# option instead of this option. The connection method to use. This
# forces the cluster to behave in the specified way instead of
# auto-discovering. One of :direct, :replica_set, :sharded,
# :load_balanced. If :connect is set to :load_balanced, the driver
# will behave as if the server is a load balancer even if it isn't
# connected to a load balancer.
# @option options [ true | false ] :load_balanced Whether to expect to
# connect to a load balancer.
# @option options [ Symbol ] :replica_set The name of the replica set to
# connect to. Servers not in this replica set will be ignored.
#
# @return [ ReplicaSet, Sharded, Single, LoadBalanced ] The topology.
#
# @since 2.0.0
# @api private
def initial(cluster, monitoring, options)
connect = options[:connect]&.to_sym
cls = if options[:direct_connection]
if connect && connect != :direct
raise ArgumentError, "Conflicting topology options: direct_connection=true and connect=#{connect}"
end
if options[:load_balanced]
raise ArgumentError, "Conflicting topology options: direct_connection=true and load_balanced=true"
end
Single
elsif options[:direct_connection] == false && connect && connect == :direct
raise ArgumentError, "Conflicting topology options: direct_connection=false and connect=#{connect}"
elsif connect && connect != :load_balanced
if options[:load_balanced]
raise ArgumentError, "Conflicting topology options: connect=#{options[:connect].inspect} and load_balanced=true"
end
OPTIONS.fetch(options[:connect].to_sym)
elsif options.key?(:replica_set) || options.key?(:replica_set_name)
if options[:load_balanced]
raise ArgumentError, "Conflicting topology options: replica_set/replica_set_name and load_balanced=true"
end
ReplicaSetNoPrimary
elsif options[:load_balanced] || connect == :load_balanced
LoadBalanced
else
Unknown
end
# Options here are client/cluster/server options.
# In particular the replica set name key is different for
# topology.
# If replica_set_name is given (as might be internally by driver),
# use that key.
# Otherwise (e.g. options passed down from client),
# move replica_set to replica_set_name.
if (cls <= ReplicaSetNoPrimary || cls == Single) && !options[:replica_set_name]
options = options.dup
options[:replica_set_name] = options.delete(:replica_set)
end
cls.new(options, monitoring, cluster)
end
end
end
end
|
