finish disabling aggregation functions

DataDog · Jul 22, 2024 · 78ce146 · 78ce146
1 parent eedbec5
commit 78ce146
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/datadog/dogstatsd/base.py b/datadog/dogstatsd/base.py
@@ -666,12 +666,11 @@ def disable_aggregation(self, is_disabled):
 
             # If aggregation has been disabled, flush and kill the background thread
             # otherwise start up the flushing thread and enable aggregation.
+            self._send = self._send_to_server
             if is_disabled:
-                self._send = self._send_to_server
                 self._stop_aggregation_flush_thread()
                 log.debug("Statsd aggregation is disabled")
             else:
-                self._send = self._send_to_aggregator
                 self._start_flush_thread(self._aggregation_flush_interval, MIN_AGGREGATION_FLUSH_INTERVAL, self.flush_aggregated_metrics, self._aggregation_flush_thread,
                     self._aggregation_flush_thread_stop)