def __backward_migrations(
self, table_name: str
) -> Sequence[operations.SqlOperation]:
return [
operations.ModifyColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name=table_name,
column=Column(
"title",
String(
Modifiers(low_cardinality=True, materialized="transaction_name")
),
),
),
operations.ModifyColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name=table_name,
column=Column(
"message",
String(
Modifiers(low_cardinality=True, materialized="transaction_name")
),
),
),
]
def __backwards_migrations(
self, table_name: str) -> Sequence[operations.SqlOperation]:
return [
operations.ModifyColumn(
storage_set=StorageSetKey.DISCOVER,
table_name=table_name,
column=Column("title", String(Modifiers(nullable=True))),
),
operations.ModifyColumn(
storage_set=StorageSetKey.DISCOVER,
table_name=table_name,
column=Column("message", String(Modifiers(nullable=True))),
),
]
def __forward_migrations(
self, table_name: str) -> Sequence[operations.SqlOperation]:
return [
operations.ModifyColumn(
storage_set=StorageSetKey.DISCOVER,
table_name=table_name,
column=Column("title", String()),
),
operations.ModifyColumn(
storage_set=StorageSetKey.DISCOVER,
table_name=table_name,
column=Column("message", String()),
),
]
def __backwards_migrations(
self, table_name: str) -> Sequence[operations.Operation]:
status_type = Enum([("success", 0), ("error", 1), ("rate-limited", 2)])
return [
operations.ModifyColumn(
StorageSetKey.QUERYLOG,
table_name,
Column("status", status_type),
),
operations.ModifyColumn(
StorageSetKey.QUERYLOG,
table_name,
Column("clickhouse_queries.status", Array(status_type)),
),
]
def __forward_migrations(
self, table_name: str) -> Sequence[operations.Operation]:
return [
operations.ModifyColumn(
StorageSetKey.QUERYLOG,
table_name,
Column("status", LowCardinality(String())),
),
operations.ModifyColumn(
StorageSetKey.QUERYLOG,
table_name,
Column("clickhouse_queries.status",
Array(LowCardinality(String()))),
),
]
def __backwards_migrations(
self, table_name: str) -> Sequence[operations.Operation]:
return [
operations.ModifyColumn(
storage_set=StorageSetKey.DISCOVER,
table_name=table_name,
column=Column("user", String(Modifiers(low_cardinality=True))),
)
]
def __backward_migrations(
self, table_name: str) -> Sequence[operations.SqlOperation]:
return [
operations.ModifyColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name=table_name,
column=Column("trace_id", UUID()),
),
]
def __forward_migrations(
self, table_name: str) -> Sequence[operations.Operation]:
return [
operations.ModifyColumn(
storage_set=StorageSetKey.EVENTS,
table_name=table_name,
column=Column(
"level",
String(Modifiers(low_cardinality=True, nullable=True))),
)
]
def __forward_migrations(
self, table_name: str
) -> Sequence[operations.SqlOperation]:
return [
operations.ModifyColumn(
storage_set=StorageSetKey.DISCOVER,
table_name=table_name,
column=Column(
"transaction_name", String(Modifiers(low_cardinality=True))
),
),
]
def forwards_local(self) -> Sequence[operations.SqlOperation]:
return [
operations.AddColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column=Column("spans.exclusive_time_32", Array(Float(32))),
after="spans.group",
),
operations.ModifyColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column=Column("spans.exclusive_time_32", Array(Float(32))),
ttl_month=("finish_ts", 1),
),
]
def forwards_local(self) -> Sequence[operations.SqlOperation]:
return [
operations.ModifyColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column=Column("duration", UInt(32)),
),
operations.AddColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column=Column(
"sdk_name", String(Modifiers(low_cardinality=True, default="''"))
),
after="user_email",
),
operations.AddColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column=Column(
"sdk_version", String(Modifiers(low_cardinality=True, default="''"))
),
after="sdk_name",
),
operations.AddColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column=Column(
"transaction_status",
UInt(8, Modifiers(default=str(UNKNOWN_SPAN_STATUS))),
),
after="transaction_op",
),
operations.AddColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column=Column("_tags_flattened", String()),
after="tags",
),
operations.AddColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column=Column("_contexts_flattened", String()),
after="contexts",
),
operations.AddColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column=Column(
"user_hash", UInt(64, Modifiers(materialized="cityHash64(user)"))
),
after="user",
),
# The following columns were originally created as non low cardinality strings
operations.ModifyColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column=Column(
"transaction_name", String(Modifiers(low_cardinality=True))
),
),
operations.ModifyColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column=Column(
"release", String(Modifiers(nullable=True, low_cardinality=True))
),
),
operations.ModifyColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column=Column(
"dist", String(Modifiers(nullable=True, low_cardinality=True))
),
),
operations.ModifyColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column=Column(
"sdk_name", String(Modifiers(low_cardinality=True, default="''"))
),
),
operations.ModifyColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column=Column(
"sdk_version", String(Modifiers(low_cardinality=True, default="''"))
),
),
operations.ModifyColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column=Column(
"environment",
String(Modifiers(nullable=True, low_cardinality=True)),
),
),
operations.AddColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column=Column("message_timestamp", DateTime()),
after="offset",
),
operations.DropColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column_name="_start_date",
),
operations.DropColumn(
storage_set=StorageSetKey.TRANSACTIONS,
table_name="transactions_local",
column_name="_finish_date",
),
]
