Add support for logs and resources.

Author: michaelsafyan

opentelemetry-configurator-gcp/src/opentelemetry/configurator/gcp/gcloud_env.py

@@ -32,32 +32,14 @@ def _is_likely_gke():
     return ('KUBERNETES_SERVICE_HOST' in os.environ)
 
 
-_NOT_GOOGLE_CLOUD = 'NOT_GCLOUD'
-_GAE = 'GOOGLE_APP_ENGINE'
-_CLOUD_RUN = 'CLOUD_RUN'
-_GKE = 'GKE'
-_GCE = 'GCE'
-
-_detected_environ = None
-
-def _detect_environ():
-    global _detected_environ
-    if _detected_environ is not None:
-        return _detected_environ
-    if not _can_resolve_metadata_server():
-        _detected_environ = _NOT_GOOGLE_CLOUD
-    elif _is_likely_gae():
-        _detected_environ = _GAE
-    elif _is_likely_gke():
-        _detected_environ = _GKE
-    elif _is_likely_cloud_run():
-        _detected_environ = _CLOUD_RUN
-    elif _is_likely_gce():
-        _detected_environ = _GCE
-    else:
-        _detected_environ  = _NOT_GOOGLE_CLOUD
-    return _detected_environ
-
-
 def is_running_on_gcp():
-    return _detect_environ() != _NOT_GOOGLE_CLOUD
+    return (
+        _can_resolve_metadata_server() and
+        (
+            _is_likely_gke() or
+            _is_likely_cloud_run() or
+            _is_likely_gce() or
+            _is_likely_gae()
+        )
+    )
+

opentelemetry-configurator-gcp/src/opentelemetry/configurator/gcp/logs.py

@@ -1,4 +1,53 @@
+import sys
+import os
+import os.path
+import logging
+from opentelemetry.exporter.cloud_logging import (
+    CloudLoggingExporter,
+)
+from opentelemetry._logs import set_logger_provider
+from opentelemetry.sdk import environment_variables as otel_env_vars
+from opentelemetry.sdk._logs import LoggerProvider, LoggingHandler
+from opentelemetry.sdk._logs.export import BatchLogRecordProcessor
+
+
+_LEVEL_NAME_TO_LEVEL = {
+    'info': logging.INFO,
+    'error': logging.ERROR,
+    'debug': logging.DEBUG,
+    'warning': logging.WARNING,
+}
+
+
+def _get_entrypoint_script_name():
+    main_script_path = sys.argv[0]
+    if not main_script_path:
+        main_script_path = sys.executable
+    simple_script_name = os.path.basename(main_scripot_path).rstrip('.py')
+    return simple_script_name
+
+
+def _get_log_name():
+    log_name = os.getenv('OTEL_GCP_LOG_NAME')
+    if log_name:
+        return log_name
+    return _get_entrypoint_script_name()
+
+
+def _get_log_level():
+    level = os.getenv(otel_env_vars.OTEL_LOG_LEVEL)
+    if level is None:
+        return logging.INFO
+    level_value = _LEVEL_NAME_TO_LEVEL.get(level)
+    if level_value is None:
+        return logging.INFO
+    return level_value
+
 
 def configure_logs_exporter(resource=None):
-    # TODO:...
-    pass
+    provider = LoggerProvider(resource=resource)
+    provider.add_log_record_processor(BatchLogRecordProcessor(
+        CloudLoggingExporter(default_log_name=_get_log_name())))
+    set_logger_provider(provider)
+    handler = LoggingHandler(level=_get_log_level(), logger_provider=provider)
+    logging.getLogger().addHandler(handler)

opentelemetry-configurator-gcp/src/opentelemetry/configurator/gcp/resource.py

@@ -1,4 +1,13 @@
+from opentelemetry.sdk import resources as otel_resources_sdk
+from opentelemetry.resourcedetector import gcp_resource_detector
+
 
 def get_resource(include_gcp_detector=False):
-    # TODO: ...
-    pass
+  detectors = [
+      otel_resources_sdk.OTELResourceDetector(),
+      otel_resources_sdk.ProcessResourceDetector(),
+      otel_resources_sdk.OsResourceDetector(),
+  ]
+  if include_gcp_detector:
+      detectors.append(gcp_resource_detector.GoogleCloudResourceDetector())
+  return otel_resources_sdk.get_aggregated_resources(detectors)