-
Notifications
You must be signed in to change notification settings - Fork 533
/
Copy pathclient.rb
1641 lines (1497 loc) · 64.3 KB
/
client.rb
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
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
# 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
# The client is the entry point to the driver and is the main object that
# will be interacted with.
#
# @since 2.0.0
class Client
extend Forwardable
include Loggable
# The options that do not affect the behavior of a cluster and its
# subcomponents.
#
# @since 2.1.0
CRUD_OPTIONS = [
:auto_encryption_options,
:database,
:read, :read_concern,
:write, :write_concern,
:retry_reads, :max_read_retries, :read_retry_interval,
:retry_writes, :max_write_retries,
# Options which cannot currently be here:
#
# :server_selection_timeout
# Server selection timeout is used by cluster constructor to figure out
# how long to wait for initial scan in compatibility mode, but once
# the cluster is initialized it no longer uses this timeout.
# Unfortunately server selector reads server selection timeout out of
# the cluster, and this behavior is required by Cluster#next_primary
# which takes no arguments. When next_primary is removed we can revsit
# using the same cluster object with different server selection timeouts.
].freeze
# Valid client options.
#
# @since 2.1.2
VALID_OPTIONS = [
:app_name,
:auth_mech,
:auth_mech_properties,
:auth_source,
:auto_encryption_options,
:bg_error_backtrace,
:cleanup,
:compressors,
:direct_connection,
:connect,
:connect_timeout,
:database,
:heartbeat_frequency,
:id_generator,
:load_balanced,
:local_threshold,
:logger,
:log_prefix,
:max_connecting,
:max_idle_time,
:max_pool_size,
:max_read_retries,
:max_write_retries,
:min_pool_size,
:monitoring,
:monitoring_io,
:password,
:platform,
:populator_io,
:read,
:read_concern,
:read_retry_interval,
:replica_set,
:resolv_options,
:retry_reads,
:retry_writes,
:scan,
:sdam_proc,
:server_api,
:server_selection_timeout,
:socket_timeout,
:srv_max_hosts,
:srv_service_name,
:ssl,
:ssl_ca_cert,
:ssl_ca_cert_object,
:ssl_ca_cert_string,
:ssl_cert,
:ssl_cert_object,
:ssl_cert_string,
:ssl_key,
:ssl_key_object,
:ssl_key_pass_phrase,
:ssl_key_string,
:ssl_verify,
:ssl_verify_certificate,
:ssl_verify_hostname,
:ssl_verify_ocsp_endpoint,
:truncate_logs,
:user,
:wait_queue_timeout,
:wrapping_libraries,
:write,
:write_concern,
:zlib_compression_level,
].freeze
# The compression algorithms supported by the driver.
#
# @since 2.5.0
VALID_COMPRESSORS = [
Mongo::Protocol::Compressed::ZSTD,
Mongo::Protocol::Compressed::SNAPPY,
Mongo::Protocol::Compressed::ZLIB
].freeze
# The known server API versions.
VALID_SERVER_API_VERSIONS = %w(
1
).freeze
# @return [ Mongo::Cluster ] cluster The cluster of servers for the client.
attr_reader :cluster
# @return [ Mongo::Database ] database The database the client is operating on.
attr_reader :database
# @return [ Hash ] options The configuration options.
attr_reader :options
# @return [ Mongo::Crypt::AutoEncrypter ] The object that encapsulates
# auto-encryption behavior
attr_reader :encrypter
# Delegate command and collections execution to the current database.
def_delegators :@database, :command, :collections
# Delegate subscription to monitoring.
def_delegators :monitoring, :subscribe, :unsubscribe
# @return [ Monitoring ] monitoring The monitoring.
# @api private
def monitoring
if cluster
cluster.monitoring
else
@monitoring
end
end
private :monitoring
# Determine if this client is equivalent to another object.
#
# @example Check client equality.
# client == other
#
# @param [ Object ] other The object to compare to.
#
# @return [ true, false ] If the objects are equal.
#
# @since 2.0.0
def ==(other)
return false unless other.is_a?(Client)
cluster == other.cluster && options == other.options
end
alias_method :eql?, :==
# Get a collection object for the provided collection name.
#
# @example Get the collection.
# client[:users]
#
# @param [ String, Symbol ] collection_name The name of the collection.
# @param [ Hash ] options The options to the collection.
#
# @return [ Mongo::Collection ] The collection.
#
# @since 2.0.0
def [](collection_name, options = {})
database[collection_name, options]
end
# Get the hash value of the client.
#
# @example Get the client hash value.
# client.hash
#
# @return [ Integer ] The client hash value.
#
# @since 2.0.0
def hash
[cluster, options].hash
end
# Instantiate a new driver client.
#
# @example Instantiate a single server or mongos client.
# Mongo::Client.new(['127.0.0.1:27017'])
#
# @example Instantiate a client for a replica set.
# Mongo::Client.new(['127.0.0.1:27017', '127.0.0.1:27021'])
#
# @example Directly connect to a mongod in a replica set
# Mongo::Client.new(['127.0.0.1:27017'], :connect => :direct)
# # without `:connect => :direct`, Mongo::Client will discover and
# # connect to the replica set if given the address of a server in
# # a replica set
#
# @param [ Array<String> | String ] addresses_or_uri The array of server addresses in the
# form of host:port or a MongoDB URI connection string.
# @param [ Hash ] options The options to be used by the client. If a MongoDB URI
# connection string is also provided, these options take precedence over any
# analogous options present in the URI string.
#
# @option options [ String, Symbol ] :app_name Application name that is
# printed to the mongod logs upon establishing a connection in server
# versions >= 3.4.
# @option options [ Symbol ] :auth_mech The authentication mechanism to
# use. One of :mongodb_cr, :mongodb_x509, :plain, :scram, :scram256
# @option options [ Hash ] :auth_mech_properties
# @option options [ String ] :auth_source The source to authenticate from.
# @option options [ true | false | nil | Integer ] :bg_error_backtrace
# Experimental. Set to true to log complete backtraces for errors in
# background threads. Set to false or nil to not log backtraces. Provide
# a positive integer to log up to that many backtrace lines.
# @option options [ Array<String> ] :compressors A list of potential
# compressors to use, in order of preference. The driver chooses the
# first compressor that is also supported by the server. Currently the
# driver only supports 'zstd, 'snappy' and 'zlib'.
# @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 [ Float ] :connect_timeout The timeout, in seconds, to
# attempt a connection.
# @option options [ String ] :database The database to connect to.
# @option options [ Float ] :heartbeat_frequency The interval, in seconds,
# for the server monitor to refresh its description via hello.
# @option options [ Object ] :id_generator A custom object to generate ids
# for documents. Must respond to #generate.
# @option options [ true | false ] :load_balanced Whether to expect to
# connect to a load balancer.
# @option options [ Integer ] :local_threshold The local threshold boundary
# in seconds for selecting a near server for an operation.
# @option options [ Logger ] :logger A custom logger to use.
# @option options [ String ] :log_prefix A custom log prefix to use when
# logging. This option is experimental and subject to change in a future
# version of the driver.
# @option options [ Integer ] :max_connecting The maximum number of
# connections that can be connecting simultaneously. The default is 2.
# This option should be increased if there are many threads that share
# the same client and the application is experiencing timeouts
# while waiting for connections to be established.
# selecting a server for an operation. The default is 2.
# @option options [ Integer ] :max_idle_time The maximum seconds a socket can remain idle
# since it has been checked in to the pool.
# @option options [ Integer ] :max_pool_size The maximum size of the
# connection pool. Setting this option to zero creates an unlimited connection pool.
# @option options [ Integer ] :max_read_retries The maximum number of read
# retries when legacy read retries are in use.
# @option options [ Integer ] :max_write_retries The maximum number of write
# retries when legacy write retries are in use.
# @option options [ Integer ] :min_pool_size The minimum size of the
# connection pool.
# @option options [ true, false ] :monitoring If false is given, the
# client is initialized without global SDAM event subscribers and
# will not publish SDAM events. Command monitoring and legacy events
# will still be published, and the driver will still perform SDAM and
# monitor its cluster in order to perform server selection. Built-in
# driver logging of SDAM events will be disabled because it is
# implemented through SDAM event subscription. Client#subscribe will
# succeed for all event types, but subscribers to SDAM events will
# not be invoked. Values other than false result in default behavior
# which is to perform normal SDAM event publication.
# @option options [ true, false ] :monitoring_io For internal driver
# use only. Set to false to prevent SDAM-related I/O from being
# done by this client or servers under it. Note: setting this option
# to false will make the client non-functional. It is intended for
# use in tests which manually invoke SDAM state transitions.
# @option options [ true | false ] :cleanup For internal driver use only.
# Set to false to prevent endSessions command being sent to the server
# to clean up server sessions when the cluster is disconnected, and to
# to not start the periodic executor. If :monitoring_io is false,
# :cleanup automatically defaults to false as well.
# @option options [ String ] :password The user's password.
# @option options [ String ] :platform Platform information to include in
# the metadata printed to the mongod logs upon establishing a connection
# in server versions >= 3.4.
# @option options [ Hash ] :read The read preference options. The hash
# may have the following items:
# - *:mode* -- read preference specified as a symbol; valid values are
# *:primary*, *:primary_preferred*, *:secondary*, *:secondary_preferred*
# and *:nearest*.
# - *:tag_sets* -- an array of hashes.
# - *:local_threshold*.
# @option options [ Hash ] :read_concern The read concern option.
# @option options [ Float ] :read_retry_interval The interval, in seconds,
# in which reads on a mongos are retried.
# @option options [ Symbol ] :replica_set The name of the replica set to
# connect to. Servers not in this replica set will be ignored.
# @option options [ true | false ] :retry_reads If true, modern retryable
# reads are enabled (which is the default). If false, modern retryable
# reads are disabled and legacy retryable reads are enabled.
# @option options [ true | false ] :retry_writes Retry writes once when
# connected to a replica set or sharded cluster versions 3.6 and up.
# (Default is true.)
# @option options [ true | false ] :scan Whether to scan all seeds
# in constructor. The default in driver version 2.x is to do so;
# driver version 3.x will not scan seeds in constructor. Opt in to the
# new behavior by setting this option to false. *Note:* setting
# this option to nil enables scanning seeds in constructor in driver
# version 2.x. Driver version 3.x will recognize this option but
# will ignore it and will never scan seeds in the constructor.
# @option options [ Proc ] :sdam_proc A Proc to invoke with the client
# as the argument prior to performing server discovery and monitoring.
# Use this to set up SDAM event listeners to receive events published
# during client construction.
#
# Note: the client is not fully constructed when sdam_proc is invoked,
# in particular the cluster is nil at this time. sdam_proc should
# limit itself to calling #subscribe and #unsubscribe methods on the
# client only.
# @option options [ Hash ] :server_api The requested server API version.
# This hash can have the following items:
# - *:version* -- string
# - *:strict* -- boolean
# - *:deprecation_errors* -- boolean
# @option options [ Integer ] :server_selection_timeout The timeout in seconds
# for selecting a server for an operation.
# @option options [ Float ] :socket_timeout The timeout, in seconds, to
# execute operations on a socket.
# @option options [ Integer ] :srv_max_hosts The maximum number of mongoses
# that the driver will communicate with for sharded topologies. If this
# option is 0, then there will be no maximum number of mongoses. If the
# given URI resolves to more hosts than ``:srv_max_hosts``, the client
# will ramdomly choose an ``:srv_max_hosts`` sized subset of hosts.
# @option options [ String ] :srv_service_name The service name to use in
# the SRV DNS query.
# @option options [ true, false ] :ssl Whether to use TLS.
# @option options [ String ] :ssl_ca_cert The file containing concatenated
# certificate authority certificates used to validate certs passed from the
# other end of the connection. Intermediate certificates should NOT be
# specified in files referenced by this option. One of :ssl_ca_cert,
# :ssl_ca_cert_string or :ssl_ca_cert_object (in order of priority) is
# required when using :ssl_verify.
# @option options [ Array<OpenSSL::X509::Certificate> ] :ssl_ca_cert_object
# An array of OpenSSL::X509::Certificate objects representing the
# certificate authority certificates used to validate certs passed from
# the other end of the connection. Intermediate certificates should NOT
# be specified in files referenced by this option. One of :ssl_ca_cert,
# :ssl_ca_cert_string or :ssl_ca_cert_object (in order of priority)
# is required when using :ssl_verify.
# @option options [ String ] :ssl_ca_cert_string A string containing
# certificate authority certificate used to validate certs passed from the
# other end of the connection. This option allows passing only one CA
# certificate to the driver. Intermediate certificates should NOT
# be specified in files referenced by this option. One of :ssl_ca_cert,
# :ssl_ca_cert_string or :ssl_ca_cert_object (in order of priority) is
# required when using :ssl_verify.
# @option options [ String ] :ssl_cert The certificate file used to identify
# the connection against MongoDB. A certificate chain may be passed by
# specifying the client certificate first followed by any intermediate
# certificates up to the CA certificate. The file may also contain the
# certificate's private key, which will be ignored. This option, if present,
# takes precedence over the values of :ssl_cert_string and :ssl_cert_object
# @option options [ OpenSSL::X509::Certificate ] :ssl_cert_object The OpenSSL::X509::Certificate
# used to identify the connection against MongoDB. Only one certificate
# may be passed through this option.
# @option options [ String ] :ssl_cert_string A string containing the PEM-encoded
# certificate used to identify the connection against MongoDB. A certificate
# chain may be passed by specifying the client certificate first followed
# by any intermediate certificates up to the CA certificate. The string
# may also contain the certificate's private key, which will be ignored,
# This option, if present, takes precedence over the value of :ssl_cert_object
# @option options [ String ] :ssl_key The private keyfile used to identify the
# connection against MongoDB. Note that even if the key is stored in the same
# file as the certificate, both need to be explicitly specified. This option,
# if present, takes precedence over the values of :ssl_key_string and :ssl_key_object
# @option options [ OpenSSL::PKey ] :ssl_key_object The private key used to identify the
# connection against MongoDB
# @option options [ String ] :ssl_key_pass_phrase A passphrase for the private key.
# @option options [ String ] :ssl_key_string A string containing the PEM-encoded private key
# used to identify the connection against MongoDB. This parameter, if present,
# takes precedence over the value of option :ssl_key_object
# @option options [ true, false ] :ssl_verify Whether to perform peer certificate validation and
# hostname verification. Note that the decision of whether to validate certificates will be
# overridden if :ssl_verify_certificate is set, and the decision of whether to validate
# hostnames will be overridden if :ssl_verify_hostname is set.
# @option options [ true, false ] :ssl_verify_certificate Whether to perform peer certificate
# validation. This setting overrides :ssl_verify with respect to whether certificate
# validation is performed.
# @option options [ true, false ] :ssl_verify_hostname Whether to perform peer hostname
# validation. This setting overrides :ssl_verify with respect to whether hostname validation
# is performed.
# @option options [ true, false ] :truncate_logs Whether to truncate the
# logs at the default 250 characters.
# @option options [ String ] :user The user name.
# @option options [ Float ] :wait_queue_timeout The time to wait, in
# seconds, in the connection pool for a connection to be checked in.
# @option options [ Array<Hash> ] :wrapping_libraries Information about
# libraries such as ODMs that are wrapping the driver, to be added to
# metadata sent to the server. Specify the lower level libraries first.
# Allowed hash keys: :name, :version, :platform.
# @option options [ Hash ] :write Deprecated. Equivalent to :write_concern
# option.
# @option options [ Hash ] :write_concern The write concern options.
# Can be :w => Integer|String, :wtimeout => Integer (in milliseconds),
# :j => Boolean, :fsync => Boolean.
# @option options [ Integer ] :zlib_compression_level The Zlib compression level to use, if using compression.
# See Ruby's Zlib module for valid levels.
# @option options [ Hash ] :resolv_options For internal driver use only.
# Options to pass through to Resolv::DNS constructor for SRV lookups.
# @option options [ Hash ] :auto_encryption_options Auto-encryption related
# options.
# - :key_vault_client => Client | nil, a client connected to the MongoDB
# instance containing the encryption key vault
# - :key_vault_namespace => String, the namespace of the key vault in the
# format database.collection
# - :kms_providers => Hash, A hash of key management service (KMS) configuration
# information. Valid hash keys are :aws, :azure, :gcp, :kmip, :local.
# There may be more than one kms provider specified.
# - :kms_tls_options => Hash, A hash of TLS options to authenticate to
# KMS providers, usually used for KMIP servers. Valid hash keys
# are :aws, :azure, :gcp, :kmip, :local. There may be more than one
# kms provider specified.
# - :schema_map => Hash | nil, JSONSchema for one or more collections
# specifying which fields should be encrypted. This option is
# mutually exclusive with :schema_map_path.
# - Note: Schemas supplied in the schema_map only apply to configuring
# automatic encryption for client side encryption. Other validation
# rules in the JSON schema will not be enforced by the driver and will
# result in an error.
# - Note: Supplying a schema_map provides more security than relying on
# JSON Schemas obtained from the server. It protects against a
# malicious server advertising a false JSON Schema, which could trick
# the client into sending unencrypted data that should be encrypted.
# - Note: If a collection is present on both the :encrypted_fields_map
# and :schema_map, an error will be raised.
# - :schema_map_path => String | nil A path to a file contains the JSON schema
# of the collection that stores auto encrypted documents. This option is
# mutually exclusive with :schema_map.
# - :bypass_auto_encryption => Boolean, when true, disables auto encryption;
# defaults to false.
# - :extra_options => Hash | nil, options related to spawning mongocryptd
# (this part of the API is subject to change).
# - :encrypted_fields_map => Hash | nil, maps a collection namespace to
# a hash describing encrypted fields for queryable encryption.
# - Note: If a collection is present on both the encryptedFieldsMap
# and schemaMap, an error will be raised.
# - :bypass_query_analysis => Boolean | nil, when true disables automatic
# analysis of outgoing commands.
# - :crypt_shared_lib_path => [ String | nil ] Path that should
# be the used to load the crypt shared library. Providing this option
# overrides default crypt shared library load paths for libmongocrypt.
# - :crypt_shared_lib_required => [ Boolean | nil ] Whether
# crypt shared library is required. If 'true', an error will be raised
# if a crypt_shared library cannot be loaded by libmongocrypt.
#
# Notes on automatic encryption:
# - Automatic encryption is an enterprise only feature that only applies
# to operations on a collection.
# - Automatic encryption is not supported for operations on a database or
# view.
# - Automatic encryption requires the authenticated user to have the
# listCollections privilege.
# - At worst, automatic encryption may triple the number of connections
# used by the Client at any one time.
# - If automatic encryption fails on an operation, use a MongoClient
# configured with bypass_auto_encryption: true and use
# ClientEncryption.encrypt to manually encrypt values.
# - Enabling Client Side Encryption reduces the maximum write batch size
# and may have a negative performance impact.
#
# @since 2.0.0
def initialize(addresses_or_uri, options = nil)
options = options ? options.dup : {}
srv_uri = nil
if addresses_or_uri.is_a?(::String)
uri = URI.get(addresses_or_uri, options)
if uri.is_a?(URI::SRVProtocol)
# If the URI is an SRV URI, note this so that we can start
# SRV polling if the topology is a sharded cluster.
srv_uri = uri
end
addresses = uri.servers
uri_options = uri.client_options.dup
# Special handing for :write and :write_concern: allow client Ruby
# options to override URI options, even when the Ruby option uses the
# deprecated :write key and the URI option uses the current
# :write_concern key
if options[:write]
uri_options.delete(:write_concern)
end
options = uri_options.merge(options)
@srv_records = uri.srv_records
else
addresses = addresses_or_uri
addresses.each do |addr|
if addr =~ /\Amongodb(\+srv)?:\/\//i
raise ArgumentError, "Host '#{addr}' should not contain protocol. Did you mean to not use an array?"
end
end
@srv_records = nil
end
options = self.class.canonicalize_ruby_options(options)
# The server API version is specified to be a string.
# However, it is very annoying to always provide the number 1 as a string,
# therefore cast to the string type here.
if server_api = options[:server_api]
if server_api.is_a?(Hash)
server_api = Options::Redacted.new(server_api)
if (version = server_api[:version]).is_a?(Integer)
options[:server_api] = server_api.merge(version: version.to_s)
end
end
end
# Special handling for sdam_proc as it is only used during client
# construction
sdam_proc = options.delete(:sdam_proc)
# For gssapi service_name, the default option is given in a hash
# (one level down from the top level).
merged_options = default_options(options)
options.each do |k, v|
default_v = merged_options[k]
if Hash === default_v
v = default_v.merge(v)
end
merged_options[k] = v
end
options = merged_options
options.keys.each do |k|
if options[k].nil?
options.delete(k)
end
end
@options = validate_new_options!(options)
=begin WriteConcern object support
if @options[:write_concern].is_a?(WriteConcern::Base)
# Cache the instance so that we do not needlessly reconstruct it.
@write_concern = @options[:write_concern]
@options[:write_concern] = @write_concern.options
end
=end
@options.freeze
validate_options!(addresses, is_srv: uri.is_a?(URI::SRVProtocol))
validate_authentication_options!
database_options = @options.dup
database_options.delete(:server_api)
@database = Database.new(self, @options[:database], database_options)
# Temporarily set monitoring so that event subscriptions can be
# set up without there being a cluster
@monitoring = Monitoring.new(@options)
if sdam_proc
sdam_proc.call(self)
end
@connect_lock = Mutex.new
@connect_lock.synchronize do
@cluster = Cluster.new(addresses, @monitoring,
cluster_options.merge(srv_uri: srv_uri))
end
begin
# Unset monitoring, it will be taken out of cluster from now on
remove_instance_variable('@monitoring')
if @options[:auto_encryption_options]
@connect_lock.synchronize do
build_encrypter
end
end
rescue
begin
@cluster.close
rescue => e
log_warn("Eror closing cluster in client constructor's exception handler: #{e.class}: #{e}")
# Drop this exception so that the original exception is raised
end
raise
end
if block_given?
begin
yield(self)
ensure
close
end
end
end
# @api private
def cluster_options
# We share clusters when a new client with different CRUD_OPTIONS
# is requested; therefore, cluster should not be getting any of these
# options upon instantiation
options.reject do |key, value|
CRUD_OPTIONS.include?(key.to_sym)
end.merge(
# but need to put the database back in for auth...
database: options[:database],
# Put these options in for legacy compatibility, but note that
# their values on the client and the cluster do not have to match -
# applications should read these values from client, not from cluster
max_read_retries: options[:max_read_retries],
read_retry_interval: options[:read_retry_interval],
).tap do |options|
# If the client has a cluster already, forward srv_uri to the new
# cluster to maintain SRV monitoring. If the client is brand new,
# its constructor sets srv_uri manually.
if cluster
options.update(srv_uri: cluster.options[:srv_uri])
end
end
end
# Get the maximum number of times the client can retry a read operation
# when using legacy read retries.
#
# @return [ Integer ] The maximum number of retries.
#
# @api private
def max_read_retries
options[:max_read_retries] || Cluster::MAX_READ_RETRIES
end
# Get the interval, in seconds, in which read retries when using legacy
# read retries.
#
# @return [ Float ] The interval.
#
# @api private
def read_retry_interval
options[:read_retry_interval] || Cluster::READ_RETRY_INTERVAL
end
# Get the maximum number of times the client can retry a write operation
# when using legacy write retries.
#
# @return [ Integer ] The maximum number of retries.
#
# @api private
def max_write_retries
options[:max_write_retries] || Cluster::MAX_WRITE_RETRIES
end
# Get an inspection of the client as a string.
#
# @example Inspect the client.
# client.inspect
#
# @return [ String ] The inspection string.
#
# @since 2.0.0
def inspect
"#<Mongo::Client:0x#{object_id} cluster=#{cluster.summary}>"
end
# Get a summary of the client state.
#
# @note The exact format and layout of the returned summary string is
# not part of the driver's public API and may be changed at any time.
#
# @return [ String ] The summary string.
#
# @since 2.7.0
def summary
"#<Client cluster=#{cluster.summary}>"
end
# Get the server selector. It either uses the read preference
# defined in the client options or defaults to a Primary server selector.
#
# @example Get the server selector.
# client.server_selector
#
# @return [ Mongo::ServerSelector ] The server selector using the
# user-defined read preference or a Primary server selector default.
#
# @since 2.5.0
def server_selector
@server_selector ||= if read_preference
ServerSelector.get(read_preference)
else
ServerSelector.primary
end
end
# Get the read preference from the options passed to the client.
#
# @example Get the read preference.
# client.read_preference
#
# @return [ BSON::Document ] The user-defined read preference.
# The document may have the following fields:
# - *:mode* -- read preference specified as a symbol; valid values are
# *:primary*, *:primary_preferred*, *:secondary*, *:secondary_preferred*
# and *:nearest*.
# - *:tag_sets* -- an array of hashes.
# - *:local_threshold*.
#
# @since 2.0.0
def read_preference
@read_preference ||= options[:read]
end
# Creates a new client configured to use the database with the provided
# name, and using the other options configured in this client.
#
# @note The new client shares the cluster with the original client,
# and as a result also shares the monitoring instance and monitoring
# event subscribers.
#
# @example Create a client for the `users' database.
# client.use(:users)
#
# @param [ String, Symbol ] name The name of the database to use.
#
# @return [ Mongo::Client ] A new client instance.
#
# @since 2.0.0
def use(name)
with(database: name)
end
# Creates a new client with the passed options merged over the existing
# options of this client. Useful for one-offs to change specific options
# without altering the original client.
#
# @note Depending on options given, the returned client may share the
# cluster with the original client or be created with a new cluster.
# If a new cluster is created, the monitoring event subscribers on
# the new client are set to the default event subscriber set and
# none of the subscribers on the original client are copied over.
#
# @example Get a client with changed options.
# client.with(:read => { :mode => :primary_preferred })
#
# @param [ Hash ] new_options The new options to use.
#
# @return [ Mongo::Client ] A new client instance.
#
# @since 2.0.0
def with(new_options = nil)
clone.tap do |client|
opts = client.update_options(new_options || Options::Redacted.new)
Database.create(client)
# We can't use the same cluster if some options that would affect it
# have changed.
if cluster_modifying?(opts)
Cluster.create(client, monitoring: opts[:monitoring])
end
end
end
# Updates this client's options from new_options, validating all options.
#
# The new options may be transformed according to various rules.
# The final hash of options actually applied to the client is returned.
#
# If options fail validation, this method may warn or raise an exception.
# If this method raises an exception, the client should be discarded
# (similarly to if a constructor raised an exception).
#
# @param [ Hash ] new_options The new options to use.
#
# @return [ Hash ] Modified new options written into the client.
#
# @api private
def update_options(new_options)
old_options = @options
new_options = self.class.canonicalize_ruby_options(new_options || {})
validate_new_options!(new_options).tap do |opts|
# Our options are frozen
options = @options.dup
if options[:write] && opts[:write_concern]
options.delete(:write)
end
if options[:write_concern] && opts[:write]
options.delete(:write_concern)
end
options.update(opts)
@options = options.freeze
auto_encryption_options_changed =
@options[:auto_encryption_options] != old_options[:auto_encryption_options]
# If there are new auto_encryption_options, create a new encrypter.
# Otherwise, allow the new client to share an encrypter with the
# original client.
#
# If auto_encryption_options are nil, set @encrypter to nil, but do not
# close the encrypter because it may still be used by the original client.
if @options[:auto_encryption_options] && auto_encryption_options_changed
@connect_lock.synchronize do
build_encrypter
end
elsif @options[:auto_encryption_options].nil?
@connect_lock.synchronize do
@encrypter = nil
end
end
validate_options!
validate_authentication_options!
end
end
# Get the read concern for this client.
#
# @example Get the client read concern.
# client.read_concern
#
# @return [ Hash ] The read concern.
#
# @since 2.6.0
def read_concern
options[:read_concern]
end
# Get the write concern for this client. If no option was provided, then a
# default single server acknowledgement will be used.
#
# @example Get the client write concern.
# client.write_concern
#
# @return [ Mongo::WriteConcern ] The write concern.
#
# @since 2.0.0
def write_concern
@write_concern ||= WriteConcern.get(options[:write_concern] || options[:write])
end
def closed?
!!@closed
end
# Close all connections.
#
# @return [ true ] Always true.
#
# @since 2.1.0
def close
@connect_lock.synchronize do
@closed = true
do_close
end
true
end
# Close encrypter and clean up auto-encryption resources.
#
# @return [ true ] Always true.
def close_encrypter
@encrypter.close if @encrypter
true
end
# Reconnect the client.
#
# @example Reconnect the client.
# client.reconnect
#
# @return [ true ] Always true.
#
# @since 2.1.0
def reconnect
addresses = cluster.addresses.map(&:to_s)
@connect_lock.synchronize do
do_close rescue nil
@cluster = Cluster.new(addresses, monitoring, cluster_options)
if @options[:auto_encryption_options]
build_encrypter
end
@closed = false
end
true
end
# Get the names of all databases.
#
# @example Get the database names.
# client.database_names
#
# @param [ Hash ] filter The filter criteria for getting a list of databases.
# @param [ Hash ] opts The command options.
#
# @option opts [ true, false ] :authorized_databases A flag that determines
# which databases are returned based on user privileges when access control
# is enabled
#
# See https://mongodb.com/docs/manual/reference/command/listDatabases/
# for more information and usage.
# @option opts [ Session ] :session The session to use.
# @option options [ Object ] :comment A user-provided
# comment to attach to this command.
#
# @return [ Array<String> ] The names of the databases.
#
# @since 2.0.5
def database_names(filter = {}, opts = {})
list_databases(filter, true, opts).collect{ |info| info['name'] }
end
# Get info for each database.
#
# @example Get the info for each database.
# client.list_databases
#
# @param [ Hash ] filter The filter criteria for getting a list of databases.
# @param [ true, false ] name_only Whether to only return each database name without full metadata.
# @param [ Hash ] opts The command options.
#
# @option opts [ true, false ] :authorized_databases A flag that determines
# which databases are returned based on user privileges when access control
# is enabled
#
# See https://mongodb.com/docs/manual/reference/command/listDatabases/
# for more information and usage.
# @option opts [ Session ] :session The session to use.
# @option options [ Object ] :comment A user-provided
# comment to attach to this command.
#
# @return [ Array<Hash> ] The info for each database.
#
# @since 2.0.5
def list_databases(filter = {}, name_only = false, opts = {})
cmd = { listDatabases: 1 }
cmd[:nameOnly] = !!name_only
cmd[:filter] = filter unless filter.empty?
cmd[:authorizedDatabases] = true if opts[:authorized_databases]
use(Database::ADMIN).database.read_command(cmd, opts).first[Database::DATABASES]
end
# Returns a list of Mongo::Database objects.
#
# @example Get a list of Mongo::Database objects.
# client.list_mongo_databases
#
# @param [ Hash ] filter The filter criteria for getting a list of databases.
# @param [ Hash ] opts The command options.
#
# @option opts [ Session ] :session The session to use.
# @option options [ Object ] :comment A user-provided
# comment to attach to this command.
#
# @return [ Array<Mongo::Database> ] The list of database objects.
#
# @since 2.5.0
def list_mongo_databases(filter = {}, opts = {})
database_names(filter, opts).collect do |name|
Database.new(self, name, options)
end
end
# Start a session.
#
# If the deployment does not support sessions, raises