fix: remove atexit

Author: Kl0ven

sentry_dynamic_sampling_lib/sampler.py

@@ -1,6 +1,5 @@
-import atexit
 import logging
-from threading import Event, Thread
+from threading import Thread
 from time import sleep
 
 import schedule
@@ -12,19 +11,13 @@
 LOGGER = logging.getLogger("SentryWrapper")
 
 
-def on_exit(event, thread):
-    event.set()
-    thread.join()
-
-
 class ControllerClient(Thread):
-    def __init__(self, stop, config, metric, *args, **kwargs) -> None:
+    def __init__(self, config, metric, *args, **kwargs) -> None:
         self.poll_interval = kwargs.pop("poll_interval")
         self.metric_interval = kwargs.pop("metric_interval")
         self.controller_endpoint = kwargs.pop("controller_endpoint")
         self.metric_endpoint = kwargs.pop("metric_endpoint")
         self.app_key = kwargs.pop("app_key")
-        self.stop: Event = stop
         self.config: Config = config
         self.metrics: Metric = metric
         self.session = CachedSession(backend="memory", cache_control=True)
@@ -37,7 +30,7 @@ def run(self):
         sleep(5)
         schedule.every(self.poll_interval).seconds.do(self.update_config)
         schedule.every(self.metric_interval).seconds.do(self.update_metrics)
-        while not self.stop.is_set():
+        while True:
             schedule.run_pending()
             sleep(1)
 
@@ -79,19 +72,11 @@ def update_metrics(self):
 
 class TraceSampler:
     def __init__(self, *args, **kwargs) -> None:
-        self.stop = Event()
         self.config = Config()
         self.metrics = Metric()
-        self.controller = ControllerClient(
-            *args, self.stop, self.config, self.metrics, **kwargs
-        )
+        self.controller = ControllerClient(*args, self.config, self.metrics, **kwargs)
         self.controller.start()
 
-        atexit.register(on_exit, self.stop, self.controller)
-
-    def __del__(self):
-        on_exit(self.stop, self.controller)
-
     def __call__(self, sampling_context):
         if sampling_context:
             if "wsgi_environ" in sampling_context: