Fix potential metric loss when open_buffer is combined with disable_buffering=False

datadog/dogstatsd/base.py

@@ -734,14 +734,11 @@ def open_buffer(self, max_buffer_size=None):
 
         self._config_lock.acquire()
 
-        # XXX Remove if `disable_buffering` default is changed to False
         self._send = self._send_to_buffer
 
         if max_buffer_size is not None:
             log.warning("The parameter max_buffer_size is now deprecated and is not used anymore")
 
-        self._reset_buffer()
-
     def close_buffer(self):
         """
         Flush the buffer and switch back to single metric packets.
@@ -752,7 +749,6 @@ def close_buffer(self):
         try:
             self.flush()
         finally:
-            # XXX Remove if `disable_buffering` default is changed to False
             if self._disable_buffering:
                 self._send = self._send_to_server
 

tests/integration/dogstatsd/test_statsd_sender.py

@@ -83,3 +83,21 @@ def test_fork_hooks(disable_background_sender, disable_buffering):
 
     foo.close()
     bar.close()
+
+
+def test_buffering_with_context():
+    statsd = DogStatsd(
+        telemetry_min_flush_interval=0,
+        disable_buffering=False,
+    )
+
+    foo, bar = socket.socketpair(socket.AF_UNIX, socket.SOCK_DGRAM, 0)
+    statsd.socket = foo
+
+    statsd.increment("first")
+    with statsd: # should not erase previously buffered metrics
+        pass
+
+    bar.settimeout(5)
+    msg = bar.recv(8192)
+    assert msg == b"first:1|c\n"