Add functionality to track integrations being used (#1065)

Author: lzchen

CHANGELOG.md

@@ -2,6 +2,9 @@
 
 ## Unreleased
 
+- Added integration tracking functionality
+([#1065](https://github.com/census-instrumentation/opencensus-python/pull/1065))
+
 # 0.7.13
 Released 2021-05-13
 
@@ -31,7 +34,7 @@ Released 2020-06-29
 Released 2020-06-17
 
 - Hotfix for breaking change
-  ([#915](https://github.com/census-instrumentation/opencensus-python/pull/915),
+  ([#915](https://github.com/census-instrumentation/opencensus-python/pull/915))
 
 ## 0.7.8
 Released 2020-06-17

contrib/opencensus-ext-django/opencensus/ext/django/middleware.py

@@ -29,6 +29,7 @@
 from opencensus.trace import (
     attributes_helper,
     execution_context,
+    integrations,
     print_exporter,
     samplers,
 )
@@ -172,6 +173,9 @@ def __init__(self, get_response=None):
         if django.VERSION >= (2,):  # pragma: NO COVER
             connection.execute_wrappers.append(_trace_db_call)
 
+        # pylint: disable=protected-access
+        integrations.add_integration(integrations._Integrations.DJANGO)
+
     def process_request(self, request):
         """Called on each request, before Django decides which view to execute.
 

contrib/opencensus-ext-flask/opencensus/ext/flask/flask_middleware.py

@@ -24,6 +24,7 @@
 from opencensus.trace import (
     attributes_helper,
     execution_context,
+    integrations,
     print_exporter,
     samplers,
 )
@@ -119,6 +120,9 @@ def init_app(self, app):
 
         self.setup_trace()
 
+        # pylint: disable=protected-access
+        integrations.add_integration(integrations._Integrations.FLASK)
+
     def setup_trace(self):
         self.app.before_request(self._before_request)
         self.app.after_request(self._after_request)

contrib/opencensus-ext-google-cloud-clientlibs/opencensus/ext/google_cloud_clientlibs/trace.py

@@ -20,6 +20,7 @@
 
 from opencensus.ext.grpc.client_interceptor import OpenCensusClientInterceptor
 from opencensus.ext.requests.trace import trace_integration as trace_requests
+from opencensus.trace import integrations
 
 log = logging.getLogger(__name__)
 
@@ -42,6 +43,9 @@ def trace_integration(tracer=None):
     # Integrate with HTTP
     trace_http(tracer)
 
+    # pylint: disable=protected-access
+    integrations.add_integration(integrations._Integrations.GOOGLE_CLOUD)
+
 
 def trace_grpc(tracer=None):
     """Integrate with gRPC."""

contrib/opencensus-ext-httplib/opencensus/ext/httplib/trace.py

@@ -15,7 +15,7 @@
 import logging
 import sys
 
-from opencensus.trace import attributes_helper, execution_context
+from opencensus.trace import attributes_helper, execution_context, integrations
 from opencensus.trace import span as span_module
 from opencensus.trace import utils
 
@@ -53,6 +53,9 @@ def trace_integration(tracer=None):
     wrapped_response = wrap_httplib_response(response_func)
     setattr(httplib.HTTPConnection, response_func.__name__, wrapped_response)
 
+    # pylint: disable=protected-access
+    integrations.add_integration(integrations._Integrations.HTTP_LIB)
+
 
 def wrap_httplib_request(request_func):
     """Wrap the httplib request function to trace. Create a new span and update

contrib/opencensus-ext-logging/opencensus/ext/logging/trace.py

@@ -15,6 +15,7 @@
 import logging
 
 from opencensus.log import TraceLogger
+from opencensus.trace import integrations
 
 
 def trace_integration(tracer=None):
@@ -25,3 +26,5 @@ def trace_integration(tracer=None):
     context.
     """
     logging.setLoggerClass(TraceLogger)
+    # pylint: disable=protected-access
+    integrations.add_integration(integrations._Integrations.LOGGING)

contrib/opencensus-ext-mysql/opencensus/ext/mysql/trace.py

@@ -18,6 +18,7 @@
 import mysql.connector
 
 from opencensus.ext.dbapi import trace
+from opencensus.trace import integrations
 
 MODULE_NAME = 'mysql'
 
@@ -31,3 +32,5 @@ def trace_integration(tracer=None):
     conn_module = inspect.getmodule(conn_func)
     wrapped = trace.wrap_conn(conn_func)
     setattr(conn_module, CONN_WRAP_METHOD, wrapped)
+    # pylint: disable=protected-access
+    integrations.add_integration(integrations._Integrations.MYSQL)

contrib/opencensus-ext-postgresql/opencensus/ext/postgresql/trace.py

@@ -19,7 +19,7 @@
 from psycopg2 import connect as pg_connect
 from psycopg2.extensions import cursor as pgcursor
 
-from opencensus.trace import execution_context
+from opencensus.trace import execution_context, integrations
 from opencensus.trace import span as span_module
 
 log = logging.getLogger(__name__)
@@ -37,6 +37,8 @@ def trace_integration(tracer=None):
     conn_func = getattr(psycopg2, CONN_WRAP_METHOD)
     conn_module = inspect.getmodule(conn_func)
     setattr(conn_module, conn_func.__name__, connect)
+    # pylint: disable=protected-access
+    integrations.add_integration(integrations._Integrations.POSTGRESQL)
 
 
 def connect(*args, **kwargs):

contrib/opencensus-ext-pymongo/opencensus/ext/pymongo/trace.py

@@ -17,7 +17,7 @@
 from google.rpc import code_pb2
 from pymongo import monitoring
 
-from opencensus.trace import execution_context
+from opencensus.trace import execution_context, integrations
 from opencensus.trace import span as span_module
 from opencensus.trace import status as status_module
 
@@ -32,6 +32,8 @@ def trace_integration(tracer=None):
     """Integrate with pymongo to trace it using event listener."""
     log.info('Integrated module: {}'.format(MODULE_NAME))
     monitoring.register(MongoCommandListener(tracer=tracer))
+    # pylint: disable=protected-access
+    integrations.add_integration(integrations._Integrations.PYMONGO)
 
 
 class MongoCommandListener(monitoring.CommandListener):

contrib/opencensus-ext-pymysql/opencensus/ext/pymysql/trace.py

@@ -18,6 +18,7 @@
 import pymysql
 
 from opencensus.ext.dbapi import trace
+from opencensus.trace import integrations
 
 MODULE_NAME = 'pymysql'
 
@@ -31,3 +32,5 @@ def trace_integration(tracer=None):
     conn_module = inspect.getmodule(conn_func)
     wrapped = trace.wrap_conn(conn_func)
     setattr(conn_module, CONN_WRAP_METHOD, wrapped)
+    # pylint: disable=protected-access
+    integrations.add_integration(integrations._Integrations.PYMYSQL)