flush

Author: lzchen

contrib/opencensus-ext-azure/examples/logs/error.py

@@ -32,4 +32,3 @@ def main():
 
 if __name__ == '__main__':
     main()
-    input("...")

contrib/opencensus-ext-azure/examples/logs/event.py

@@ -23,5 +23,3 @@
 logger.addHandler(AzureEventHandler())
 logger.setLevel(logging.INFO)
 logger.info('Hello, World!')
-
-input("...")

contrib/opencensus-ext-azure/examples/logs/properties.py

@@ -32,5 +32,3 @@
     result = 1 / 0  # generate a ZeroDivisionError
 except Exception:
     logger.exception('Captured an exception.', extra=properties)
-
-input("...")

contrib/opencensus-ext-azure/examples/logs/simple.py

@@ -21,6 +21,3 @@
 # and place it in the APPLICATIONINSIGHTS_CONNECTION_STRING
 # environment variable.
 logger.addHandler(AzureLogHandler())
-logger.warning('Hello, World!')
-
-input("...")

contrib/opencensus-ext-azure/examples/metrics/simple.py

@@ -50,7 +50,6 @@ def main():
 
     mmap.measure_int_put(CARROTS_MEASURE, 1000)
     mmap.record(tmap)
-    time.sleep(60)
 
     print("Done recording metrics")
 

contrib/opencensus-ext-azure/examples/traces/credential.py

@@ -35,4 +35,3 @@
 
 with tracer.span(name='foo'):
     print('Hello, World!')
-input(...)

contrib/opencensus-ext-azure/examples/traces/simple.py

@@ -23,4 +23,3 @@
 
 with tracer.span(name='foo'):
     print('Hello, World!')
-input(...)

contrib/opencensus-ext-azure/opencensus/ext/azure/log_exporter/__init__.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import atexit
 import logging
 import os
 import random
@@ -63,6 +64,7 @@ def __init__(self, **options):
         self._queue = Queue(capacity=self.options.queue_capacity)
         self._worker = Worker(self._queue, self)
         self._worker.start()
+        # atexit.register(self.close, self.options.grace_period)
         # start statsbeat on exporter instantiation
         if not os.environ.get("APPLICATIONINSIGHTS_STATSBEAT_DISABLED_ALL"):
             statsbeat_metrics.collect_statsbeat_metrics(self.options)
@@ -88,11 +90,11 @@ def _export(self, batch, event=None):  # pragma: NO COVER
             if event:
                 event.set()
 
-    def close(self):
+    def close(self, timeout=None):
         if self.storage:
             self.storage.close()
         if self._worker:
-            self._worker.stop()
+            self._worker.stop(timeout)
 
     def createLock(self):
         self.lock = None

contrib/opencensus-ext-azure/opencensus/ext/azure/metrics_exporter/__init__.py

@@ -58,7 +58,7 @@ def __init__(self, is_stats=False, **options):
                 retention_period=self.options.storage_retention_period,
                 source=self.__class__.__name__,
             )
-        self._atexit_handler = atexit.register(self.shutdown)
+        self._atexit_handler = atexit.register(self.shutdown, self.options.grace_period)
         self.exporter_thread = None
         # For redirects
         self._consecutive_redirects = 0  # To prevent circular redirects