Enable defining a handler for buffer overflow when using the library as a python logging handler

montaro · montaro · commit 3ab3aa2b40b2 · 2017-01-26T02:38:49.000+02:00
diff --git a/.gitignore b/.gitignore
@@ -9,3 +9,4 @@
 /.tox
 /build
 /dist
+/.idea
diff --git a/README.rst b/README.rst
@@ -188,7 +188,7 @@ module.
 
     logging.basicConfig(level=logging.INFO)
     l = logging.getLogger('fluent.test')
-    h = handler.FluentHandler('app.follow', host='host', port=24224)
+    h = handler.FluentHandler('app.follow', host='host', port=24224, buffer_overflow_handler=handler)
     formatter = handler.FluentRecordFormatter(custom_format)
     h.setFormatter(formatter)
     l.addHandler(h)
@@ -242,6 +242,7 @@ A sample configuration ``logging.yaml`` would be:
                 host: localhost
                 port: 24224
                 tag: test.logging
+                buffer_overflow_handler: handler
                 formatter: fluent_fmt
                 level: DEBUG
             none:
diff --git a/fluent/handler.py b/fluent/handler.py
@@ -115,12 +115,14 @@ def __init__(self,
                  host='localhost',
                  port=24224,
                  timeout=3.0,
-                 verbose=False):
+                 verbose=False,
+                 buffer_overflow_handler=None):
 
         self.tag = tag
         self.sender = sender.FluentSender(tag,
                                           host=host, port=port,
-                                          timeout=timeout, verbose=verbose)
+                                          timeout=timeout, verbose=verbose,
+                                          buffer_overflow_handler=buffer_overflow_handler)
         logging.Handler.__init__(self)
 
     def emit(self, record):

-Original file line number
+Diff line change
 /.tox
 /build
 /dist
 +/.idea