add flush function

DataDog · Aug 12, 2024 · 13d9d39 · 13d9d39
1 parent aa74073
commit 13d9d39
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/datadog/dogstatsd/base.py b/datadog/dogstatsd/base.py
@@ -1533,7 +1533,16 @@ def pre_fork(self):
 
     def post_fork_parent(self):
         """Restore the client state after a fork in the parent process."""
-        self._start_flush_thread(self._flush_interval)
+        if self._disable_aggregating:
+            self._start_flush_thread(
+                self._flush_interval,
+                self.flush_buffered_metrics,
+            )
+        else:
+            self._start_flush_thread(
+                self._flush_interval,
+                self.flush_aggregated_metrics,
+            )
         self._start_sender_thread()
         self._config_lock.release()