Добавлен сбор телеметрии на ВМ2

2026-08-26 15:12:51 +03:00
parent 728b9826a3
commit f989097484
32 changed files with 2029 additions and 134 deletions
@@ -0,0 +1,42 @@
+"""store safe async trace context on safety tasks
+
+Revision ID: 0002_task_trace_context
+Revises: 0001_message_safety_v2
+"""
+
+from alembic import op
+
+revision = "0002_task_trace_context"
+down_revision = "0001_message_safety_v2"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    op.execute(
+        "ALTER TABLE message_safety.safety_tasks "
+        "ADD COLUMN IF NOT EXISTS origin_trace_id bytea NULL"
+    )
+    op.execute(
+        "ALTER TABLE message_safety.safety_tasks "
+        "ADD COLUMN IF NOT EXISTS origin_span_id bytea NULL"
+    )
+    op.execute(
+        "ALTER TABLE message_safety.safety_tasks "
+        "ADD COLUMN IF NOT EXISTS origin_trace_flags integer NULL"
+    )
+    op.execute(
+        "ALTER TABLE message_safety.safety_tasks "
+        "ADD COLUMN IF NOT EXISTS origin_tracestate varchar(512) NULL"
+    )
+
+
+def downgrade() -> None:
+    op.execute(
+        "ALTER TABLE message_safety.safety_tasks DROP COLUMN IF EXISTS origin_tracestate"
+    )
+    op.execute(
+        "ALTER TABLE message_safety.safety_tasks DROP COLUMN IF EXISTS origin_trace_flags"
+    )
+    op.execute("ALTER TABLE message_safety.safety_tasks DROP COLUMN IF EXISTS origin_span_id")
+    op.execute("ALTER TABLE message_safety.safety_tasks DROP COLUMN IF EXISTS origin_trace_id")