def description_changed(self, event):
previous_server_type = event.previous_description.server_type
new_server_type = event.new_description.server_type
if new_server_type != previous_server_type:
logger.debug(
f"Server {event.server_address} changed type from {event.previous_description.server_type_name} to "
f"{event.new_description.server_type_name}")
def test_logging():
logger.trace(msg="TRACE MESSAGE")
logger.debug(msg="DEBUG MESSAGE")
logger.info(msg="INFO MESSAGE")
logger.success(msg="SUCCESS MESSAGE")
logger.warning(msg="WARNING MESSAGE")
logger.error(msg="ERROR MESSAGE")
logger.critical(msg="CRITICAL MESSAGE")
def test_log(self):
with self.assertLogs(logger=logger, level=TRACE) as logger_context:
logger.trace(msg="TESTING TRACE")
logger.debug(msg="TESTING DEBUG")
logger.success(msg="TESTING SUCCESS")
logger.info(msg="TESTING INFO")
logger.warning(msg="TESTING WARNING")
logger.error(msg="TESTING ERROR")
logger.critical(msg="TESTING CRITICAL")
self.assertEqual(
[
f"TRACE:{logger.name}:TESTING TRACE",
f"DEBUG:{logger.name}:TESTING DEBUG",
f"SUCCESS:{logger.name}:TESTING SUCCESS",
f"INFO:{logger.name}:TESTING INFO",
f"WARNING:{logger.name}:TESTING WARNING",
f"ERROR:{logger.name}:TESTING ERROR",
f"CRITICAL:{logger.name}:TESTING CRITICAL",
],
logger_context.output,
)
def description_changed(self, event):
logger.debug(
f"Topology description updated for topology id {event.topology_id}"
)
previous_topology_type = event.previous_description.topology_type
new_topology_type = event.new_description.topology_type
if new_topology_type != previous_topology_type:
logger.debug(
f"Topology {event.topology_id} changed type from {event.previous_description.topology_type_name} to "
f"{event.new_description.topology_type_name}")
if not event.new_description.has_writable_server():
logger.debug("No writable servers available.")
if not event.new_description.has_readable_server():
logger.debug("No readable servers available.")
def _end_trace_malloc(self):
logger.debug(msg="=== START SNAPSHOT ===")
snapshot = tracemalloc.take_snapshot()
snapshot = snapshot.filter_traces(filters=self._get_trace_malloc_filters())
for stat in snapshot.statistics(key_type="lineno", cumulative=True):
logger.debug(msg=f"{stat}")
if self.show_memory:
size, peak = tracemalloc.get_traced_memory()
snapshot_size = tracemalloc.get_tracemalloc_memory()
logger.debug(
msg=f"❕size={self._bytes_to_megabytes(size=size)}, "
f"❗peak={self._bytes_to_megabytes(size=peak)}, "
f"💾snapshot_size={self._bytes_to_megabytes(size=snapshot_size)}"
)
if self.clear_traces:
tracemalloc.clear_traces()
logger.debug(msg="=== END SNAPSHOT ===")
def connection_check_out_failed(self, event):
logger.debug(
f"[pool {event.address}] connection check out failed, reason: {event.reason}"
)
def connection_closed(self, event):
logger.debug(
f"[pool {event.address}][conn #{event.connection_id}] connection closed, reason: {event.reason}"
)
def connection_check_out_started(self, event):
logger.debug(f"[pool {event.address}] connection check out started")
def opened(self, event):
logger.debug(f"Topology with id {event.topology_id} opened")
def connection_ready(self, event):
logger.debug(
f"[pool {event.address}][conn #{event.connection_id}] connection setup succeeded"
)
def opened(self, event):
logger.debug(
f"Server {event.server_address} added to topology {event.topology_id}"
)
def closed(self, event):
logger.debug(
f"Server {event.server_address} removed from topology {event.topology_id}"
)
def started(self, event):
logger.debug(
f"Command '{event.command_name}' with request id {event.request_id} started on server {event.connection_id}"
)
def _print_timing(self, name: str, precision: int = 5):
logger.debug(f"📊{name} ⏱: {self._end_time - self._start_time:.{precision}f} seconds")
def create_client(cls):
"""Creating MongoDB client"""
logger.debug(msg="Creating MongoDB client")
cls.client = motor.motor_asyncio.AsyncIOMotorClient(
settings.Settings.MONGO_URL)
def delete_client(cls):
"""Closing MongoDB client"""
logger.debug(msg="Disconnecting from MongoDB")
cls.client.close()
cls.client = None # noqa
def retrieve_client(cls) -> pymongo.MongoClient:
"""Retrieve existing MongoDB client or create it (at first call)"""
if cls.client is None:
logger.debug(msg="Initialization of MongoDB")
cls.create_client()
return cls.client
def closed(self, event):
logger.debug(f"Topology with id {event.topology_id} closed")
def succeeded(self, event):
logger.debug(
f"Heartbeat to server {event.connection_id} succeeded with reply {event.reply.document}"
)
def connection_checked_out(self, event):
logger.debug(
f"[pool {event.address}][conn #{event.connection_id}] connection checked out of pool"
)
def failed(self, event):
logger.debug(
f"Command {event.command_name} with request id {event.request_id} on server {event.connection_id} failed "
f"in {event.duration_micros} microseconds")
def connection_checked_in(self, event):
logger.debug(
f"[pool {event.address}][conn #{event.connection_id}] connection checked into pool"
)
def pool_closed(self, event):
logger.debug(f"[pool {event.address}] pool closed")
def started(self, event):
logger.debug(f"Heartbeat sent to server {event.connection_id}")
def connection_created(self, event):
logger.debug(
f"[pool {event.address}][conn #{event.connection_id}] connection created"
)
def succeeded(self, event):
logger.debug(
f"Command '{event.command_name}' with request id {event.request_id} on server {event.connection_id} "
f"succeeded in {event.duration_micros} microseconds")
def failed(self, event):
logger.debug(
f"Heartbeat to server {event.connection_id} failed with error {event.reply}"
)
