Remove thread-unsafe event logging code (#95)

This would occasionally cause errors because of dask/distributed#5552, and wasn't very useful anyway.

Author: gjoseph92

stackstac/rio_reader.py

@@ -1,10 +1,8 @@
 from __future__ import annotations
 
-import functools
 import logging
 import threading
 import warnings
-import weakref
 from typing import TYPE_CHECKING, Optional, Protocol, Tuple, Type, TypedDict, Union
 
 import numpy as np
@@ -32,16 +30,6 @@ def _curthread():
     return threading.current_thread().name
 
 
-def log_event(topic: str, msg: dict) -> None:
-    try:
-        import distributed
-
-        worker = distributed.get_worker()
-    except (ImportError, ValueError):
-        return
-    worker.log_event(topic, dict(msg, thread=_curthread()))
-
-
 # /TODO
 
 
@@ -203,8 +191,6 @@ def __init__(
         # but because `close` closes datasets across all threads by simply deleting the current threadlocal
         # and replacing it with an empty one, we have to synchronize all access to `self._threadlocal`.
 
-        log_event("create_ThreadLocalRioDataset", dict(url=self._url, vrt=bool(vrt)))
-
     def _open(self) -> Union[SelfCleaningDatasetReader, WarpedVRT]:
         with self._env.open:
             with time(f"Reopen {self._url!r} in {_curthread()}: {{t}}"):
@@ -214,23 +200,16 @@ def _open(self) -> Union[SelfCleaningDatasetReader, WarpedVRT]:
                     driver=self._driver,
                     **self._open_options,
                 )
-                log_event("open_dataset", dict(url=self._url))
             if self._vrt_params:
                 with self._env.open_vrt:
                     result = vrt = WarpedVRT(ds, sharing=False, **self._vrt_params)
-                    log_event("open_vrt", dict(url=self._url))
             else:
                 vrt = None
 
         with self._lock:
             self._threadlocal.ds = ds
             self._threadlocal.vrt = vrt
 
-        weakref.ref(
-            ds, functools.partial(log_event, "close_dataset", dict(url=self._url))
-        )
-        weakref.ref(vrt, functools.partial(log_event, "close_vrt", dict(url=self._url)))
-        # NOTE: functools.partial to hopefully avoid taking a closure over `self`
         return result
 
     @property
@@ -277,7 +256,6 @@ def close(self) -> None:
         # datasets.
         # NOTE: we're assuming here that closing a GDAL dataset from a thread other than the one that created
         # it is safe to do, which, knowing GDAL, is quite possibly untrue.
-        log_event("close_ThreadLocalRioDataset", dict(url=self._url))
         with self._lock:
             self._threadlocal = threading.local()
 
@@ -365,14 +343,6 @@ def _open(self) -> ThreadsafeRioDataset:
                     "a separate STAC asset), so you'll need to exclude this asset from your analysis."
                 )
 
-            log_event("open_dataset_initial", dict(url=self.url))
-            weakref.ref(
-                ds,
-                functools.partial(
-                    log_event, "close_dataset_initial", dict(url=self.url)
-                ),
-            )
-
             # Only make a VRT if the dataset doesn't match the spatial spec we want
             if self.spec.vrt_params != {
                 "crs": ds.crs.to_epsg(),
@@ -387,13 +357,6 @@ def _open(self) -> ThreadsafeRioDataset:
                         resampling=self.resampling,
                         **self.spec.vrt_params,
                     )
-                    log_event("open_vrt_initial", dict(url=self.url))
-                    weakref.ref(
-                        vrt,
-                        functools.partial(
-                            log_event, "close_vrt_initial", dict(url=self.url)
-                        ),
-                    )
             else:
                 logger.info(f"Skipping VRT for {self.url!r}")
                 vrt = None