GoogleCloudLoggingFormatter trace context

Author: iamacarpet

src/Monolog/Formatter/GoogleCloudLoggingFormatter.php

@@ -24,6 +24,10 @@
  */
 final class GoogleCloudLoggingFormatter extends JsonFormatter
 {
+    const CONTEXT_HEADER_FORMAT = '/([0-9a-fA-F]{32})(?:\/(\d+))?(?:;o=(\d+))?/';
+
+    private static ?string $traceID = null;
+
     protected function normalizeRecord(LogRecord $record): array
     {
         $normalized = parent::normalizeRecord($record);
@@ -32,9 +36,31 @@ protected function normalizeRecord(LogRecord $record): array
         $normalized['severity'] = $normalized['level_name'];
         $normalized['time'] = $record->datetime->format(DateTimeInterface::RFC3339_EXTENDED);
 
+        // Tag with Trace ID for request attribution
+        $normalized['logging.googleapis.com/trace'] = $this->getTraceID();
+
         // Remove keys that are not used by GCP
         unset($normalized['level'], $normalized['level_name'], $normalized['datetime']);
 
         return $normalized;
     }
+
+    private function getTraceID(): string
+    {
+        if (empty($this->traceID) && !empty($_SERVER['HTTP_X_CLOUD_TRACE_CONTEXT'])) {
+            $matched = preg_match(
+                self::CONTEXT_HEADER_FORMAT,
+                $_SERVER['HTTP_X_CLOUD_TRACE_CONTEXT'] ?? '',
+                $matches,
+            );
+
+            if (!$matched) {
+                return null;
+            }
+
+            $this->traceID = strtolower($matches[1]);
+        }
+
+        return $this->traceID;
+    }
 }