mirror/signoz
1
0
Fork 0
mirror of https://github.com/SigNoz/signoz.git synced 2026-05-08 08:00:13 -05:00
Code Issues Packages Projects Releases Wiki Activity

test(integration): add dtype=object to preserve types and insert resource keys (#9897)

Browse Source
This commit is contained in:
Srikanth Chekuri
2025-12-30 00:28:34 +05:30
committed by GitHub
parent ab8b42fbbe
commit 201d5c24a5
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/integration/fixtures/traces.py
+10 -1
View File
@@ -81,7 +81,7 @@ class TracesResourceOrAttributeKeys(ABC):
self.is_column = is_column
def np_arr(self) -> np.array:
return np.array([self.name, self.tag_type, self.datatype, self.is_column])
return np.array([self.name, self.tag_type, self.datatype, self.is_column], dtype=object)
class TracesTagAttributes(ABC):
@@ -636,8 +636,10 @@ def insert_traces(
)
attribute_keys: List[TracesResourceOrAttributeKeys] = []
resource_keys: List[TracesResourceOrAttributeKeys] = []
for trace in traces:
attribute_keys.extend(trace.attribute_keys)
resource_keys.extend(trace.resource_keys)
if len(attribute_keys) > 0:
clickhouse.conn.insert(
@@ -646,6 +648,13 @@ def insert_traces(
data=[attribute_key.np_arr() for attribute_key in attribute_keys],
)
if len(resource_keys) > 0:
clickhouse.conn.insert(
database="signoz_traces",
table="distributed_span_attributes_keys",
data=[resource_key.np_arr() for resource_key in resource_keys],
)
# Insert main traces
clickhouse.conn.insert(
database="signoz_traces",