160 lines
4.9 KiB
Python
160 lines
4.9 KiB
Python
|
# Copyright 2014-present 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.
|
||
|
|
||
|
"""Represent MongoClient's configuration."""
|
||
|
|
||
|
import threading
|
||
|
import traceback
|
||
|
|
||
|
from bson.objectid import ObjectId
|
||
|
from pymongo import common, monitor, pool
|
||
|
from pymongo.common import LOCAL_THRESHOLD_MS, SERVER_SELECTION_TIMEOUT
|
||
|
from pymongo.errors import ConfigurationError
|
||
|
from pymongo.pool import PoolOptions
|
||
|
from pymongo.server_description import ServerDescription
|
||
|
from pymongo.topology_description import TOPOLOGY_TYPE
|
||
|
|
||
|
|
||
|
class TopologySettings:
|
||
|
def __init__(
|
||
|
self,
|
||
|
seeds=None,
|
||
|
replica_set_name=None,
|
||
|
pool_class=None,
|
||
|
pool_options=None,
|
||
|
monitor_class=None,
|
||
|
condition_class=None,
|
||
|
local_threshold_ms=LOCAL_THRESHOLD_MS,
|
||
|
server_selection_timeout=SERVER_SELECTION_TIMEOUT,
|
||
|
heartbeat_frequency=common.HEARTBEAT_FREQUENCY,
|
||
|
server_selector=None,
|
||
|
fqdn=None,
|
||
|
direct_connection=False,
|
||
|
load_balanced=None,
|
||
|
srv_service_name=common.SRV_SERVICE_NAME,
|
||
|
srv_max_hosts=0,
|
||
|
):
|
||
|
"""Represent MongoClient's configuration.
|
||
|
|
||
|
Take a list of (host, port) pairs and optional replica set name.
|
||
|
"""
|
||
|
if heartbeat_frequency < common.MIN_HEARTBEAT_INTERVAL:
|
||
|
raise ConfigurationError(
|
||
|
"heartbeatFrequencyMS cannot be less than %d"
|
||
|
% (common.MIN_HEARTBEAT_INTERVAL * 1000,)
|
||
|
)
|
||
|
|
||
|
self._seeds = seeds or [("localhost", 27017)]
|
||
|
self._replica_set_name = replica_set_name
|
||
|
self._pool_class = pool_class or pool.Pool
|
||
|
self._pool_options = pool_options or PoolOptions()
|
||
|
self._monitor_class = monitor_class or monitor.Monitor
|
||
|
self._condition_class = condition_class or threading.Condition
|
||
|
self._local_threshold_ms = local_threshold_ms
|
||
|
self._server_selection_timeout = server_selection_timeout
|
||
|
self._server_selector = server_selector
|
||
|
self._fqdn = fqdn
|
||
|
self._heartbeat_frequency = heartbeat_frequency
|
||
|
self._direct = direct_connection
|
||
|
self._load_balanced = load_balanced
|
||
|
self._srv_service_name = srv_service_name
|
||
|
self._srv_max_hosts = srv_max_hosts or 0
|
||
|
|
||
|
self._topology_id = ObjectId()
|
||
|
# Store the allocation traceback to catch unclosed clients in the
|
||
|
# test suite.
|
||
|
self._stack = "".join(traceback.format_stack())
|
||
|
|
||
|
@property
|
||
|
def seeds(self):
|
||
|
"""List of server addresses."""
|
||
|
return self._seeds
|
||
|
|
||
|
@property
|
||
|
def replica_set_name(self):
|
||
|
return self._replica_set_name
|
||
|
|
||
|
@property
|
||
|
def pool_class(self):
|
||
|
return self._pool_class
|
||
|
|
||
|
@property
|
||
|
def pool_options(self):
|
||
|
return self._pool_options
|
||
|
|
||
|
@property
|
||
|
def monitor_class(self):
|
||
|
return self._monitor_class
|
||
|
|
||
|
@property
|
||
|
def condition_class(self):
|
||
|
return self._condition_class
|
||
|
|
||
|
@property
|
||
|
def local_threshold_ms(self):
|
||
|
return self._local_threshold_ms
|
||
|
|
||
|
@property
|
||
|
def server_selection_timeout(self):
|
||
|
return self._server_selection_timeout
|
||
|
|
||
|
@property
|
||
|
def server_selector(self):
|
||
|
return self._server_selector
|
||
|
|
||
|
@property
|
||
|
def heartbeat_frequency(self):
|
||
|
return self._heartbeat_frequency
|
||
|
|
||
|
@property
|
||
|
def fqdn(self):
|
||
|
return self._fqdn
|
||
|
|
||
|
@property
|
||
|
def direct(self):
|
||
|
"""Connect directly to a single server, or use a set of servers?
|
||
|
|
||
|
True if there is one seed and no replica_set_name.
|
||
|
"""
|
||
|
return self._direct
|
||
|
|
||
|
@property
|
||
|
def load_balanced(self):
|
||
|
"""True if the client was configured to connect to a load balancer."""
|
||
|
return self._load_balanced
|
||
|
|
||
|
@property
|
||
|
def srv_service_name(self):
|
||
|
"""The srvServiceName."""
|
||
|
return self._srv_service_name
|
||
|
|
||
|
@property
|
||
|
def srv_max_hosts(self):
|
||
|
"""The srvMaxHosts."""
|
||
|
return self._srv_max_hosts
|
||
|
|
||
|
def get_topology_type(self):
|
||
|
if self.load_balanced:
|
||
|
return TOPOLOGY_TYPE.LoadBalanced
|
||
|
elif self.direct:
|
||
|
return TOPOLOGY_TYPE.Single
|
||
|
elif self.replica_set_name is not None:
|
||
|
return TOPOLOGY_TYPE.ReplicaSetNoPrimary
|
||
|
else:
|
||
|
return TOPOLOGY_TYPE.Unknown
|
||
|
|
||
|
def get_server_descriptions(self):
|
||
|
"""Initial dict of (address, ServerDescription) for all seeds."""
|
||
|
return {address: ServerDescription(address) for address in self.seeds}
|