Create util/trace.py

util/trace.py

@@ -0,0 +1,37 @@
+import os
+from google.cloud import trace_v2
+from google.oauth2 import service_account
+import uuid
+
+with open(os.getenv("GOOGLE_APPLICATION_CREDENTIALS"), "w") as f:
+    f.write(os.getenv("gcp_service_json"))
+    f.close()
+
+def initialize_trace_client():
+    credentials = service_account.Credentials.from_service_account_file(os.getenv("GOOGLE_APPLICATION_CREDENTIALS"))
+    client = trace_v2.TraceServiceClient(credentials=credentials)
+    return client, f"projects/{os.getenv("PROJECT")}"
+
+
+from google.protobuf.timestamp_pb2 import Timestamp
+from datetime import datetime
+
+def create_span(client, parent_span_id=None, span_display_name="Default Span"):
+    trace_id = uuid.uuid4().hex
+    span_id = uuid.uuid4().hex[:16]
+    span_name = f"{os.getenv("PROJECT")}/traces/{trace_id}/spans/{span_id}"
+    now = datetime.utcnow()
+    timestamp = Timestamp()
+    timestamp.FromDatetime(now)
+    
+    span = trace_v2.Span(
+        name=span_name,
+        span_id=span_id,
+        parent_span_id=parent_span_id,
+        display_name=trace_v2.types.TruncatableString(value=span_display_name),
+        start_time=timestamp,
+        end_time=timestamp
+    )
+    
+    client.batch_write_spans(name=os.getenv("PROJECT"), spans=[span])
+    return span_name  # 返り値としてスパンの名前を返す