diff --git a/sentry_sdk/integrations/clickhouse_driver.py b/sentry_sdk/integrations/clickhouse_driver.py index 730ecf3bf2..e6b3009548 100644 --- a/sentry_sdk/integrations/clickhouse_driver.py +++ b/sentry_sdk/integrations/clickhouse_driver.py @@ -90,6 +90,7 @@ def _inner(*args: "P.args", **kwargs: "P.kwargs") -> "T": attributes={ "sentry.op": OP.DB, "sentry.origin": ClickhouseDriverIntegration.origin, + SPANDATA.DB_QUERY_TEXT: str(query), }, ) else: diff --git a/tests/integrations/clickhouse_driver/test_clickhouse_driver.py b/tests/integrations/clickhouse_driver/test_clickhouse_driver.py index 33b62c7d89..57b19badcc 100644 --- a/tests/integrations/clickhouse_driver/test_clickhouse_driver.py +++ b/tests/integrations/clickhouse_driver/test_clickhouse_driver.py @@ -234,7 +234,6 @@ def test_clickhouse_client_spans( sentry_init, capture_events, capture_items, - capture_envelopes, span_streaming, ): sentry_init( @@ -278,6 +277,7 @@ def test_clickhouse_client_spans( "db.namespace": "", "db.user": "default", "sentry.op": "db", + "db.query.text": "DROP TABLE IF EXISTS test", "sentry.origin": "auto.db.clickhouse_driver", "server.address": "localhost", "server.port": 9000, @@ -293,6 +293,7 @@ def test_clickhouse_client_spans( "db.namespace": "", "db.user": "default", "sentry.op": "db", + "db.query.text": "CREATE TABLE test (x Int32) ENGINE = Memory", "sentry.origin": "auto.db.clickhouse_driver", "server.address": "localhost", "server.port": 9000, @@ -308,6 +309,7 @@ def test_clickhouse_client_spans( "db.namespace": "", "db.user": "default", "sentry.op": "db", + "db.query.text": "INSERT INTO test (x) VALUES", "sentry.origin": "auto.db.clickhouse_driver", "server.address": "localhost", "server.port": 9000, @@ -323,6 +325,7 @@ def test_clickhouse_client_spans( "db.namespace": "", "db.user": "default", "sentry.op": "db", + "db.query.text": "INSERT INTO test (x) VALUES", "sentry.origin": "auto.db.clickhouse_driver", "server.address": "localhost", "server.port": 9000, @@ -338,6 +341,7 @@ def test_clickhouse_client_spans( "db.namespace": "", "db.user": "default", "sentry.op": "db", + "db.query.text": "SELECT sum(x) FROM test WHERE x > 150", "sentry.origin": "auto.db.clickhouse_driver", "server.address": "localhost", "server.port": 9000, @@ -563,6 +567,7 @@ def test_clickhouse_client_spans_with_pii( "db.system.name": "clickhouse", "db.namespace": "", "db.user": "default", + "db.query.text": "DROP TABLE IF EXISTS test", "server.address": "localhost", "server.port": 9000, "thread.id": mock.ANY, @@ -577,6 +582,7 @@ def test_clickhouse_client_spans_with_pii( "db.system.name": "clickhouse", "db.namespace": "", "db.user": "default", + "db.query.text": "CREATE TABLE test (x Int32) ENGINE = Memory", "server.address": "localhost", "server.port": 9000, "thread.id": mock.ANY, @@ -591,6 +597,7 @@ def test_clickhouse_client_spans_with_pii( "db.system.name": "clickhouse", "db.namespace": "", "db.user": "default", + "db.query.text": "INSERT INTO test (x) VALUES", "server.address": "localhost", "server.port": 9000, "thread.id": mock.ANY, @@ -605,6 +612,7 @@ def test_clickhouse_client_spans_with_pii( "db.system.name": "clickhouse", "db.namespace": "", "db.user": "default", + "db.query.text": "INSERT INTO test (x) VALUES", "server.address": "localhost", "server.port": 9000, }, @@ -617,6 +625,7 @@ def test_clickhouse_client_spans_with_pii( "db.system.name": "clickhouse", "db.namespace": "", "db.user": "default", + "db.query.text": "SELECT sum(x) FROM test WHERE x > 150", "server.address": "localhost", "server.port": 9000, }, @@ -1012,6 +1021,7 @@ def test_clickhouse_dbapi_spans( "db.driver.name": "clickhouse-driver", "db.namespace": "", "db.user": "default", + "db.query.text": "DROP TABLE IF EXISTS test", "sentry.op": "db", "sentry.origin": "auto.db.clickhouse_driver", "server.address": "localhost", @@ -1027,6 +1037,7 @@ def test_clickhouse_dbapi_spans( "db.driver.name": "clickhouse-driver", "db.namespace": "", "db.user": "default", + "db.query.text": "CREATE TABLE test (x Int32) ENGINE = Memory", "sentry.op": "db", "sentry.origin": "auto.db.clickhouse_driver", "server.address": "localhost", @@ -1042,6 +1053,7 @@ def test_clickhouse_dbapi_spans( "db.driver.name": "clickhouse-driver", "db.namespace": "", "db.user": "default", + "db.query.text": "INSERT INTO test (x) VALUES", "sentry.op": "db", "sentry.origin": "auto.db.clickhouse_driver", "server.address": "localhost", @@ -1057,6 +1069,7 @@ def test_clickhouse_dbapi_spans( "db.driver.name": "clickhouse-driver", "db.namespace": "", "db.user": "default", + "db.query.text": "INSERT INTO test (x) VALUES", "sentry.op": "db", "sentry.origin": "auto.db.clickhouse_driver", "server.address": "localhost", @@ -1072,6 +1085,7 @@ def test_clickhouse_dbapi_spans( "db.driver.name": "clickhouse-driver", "db.namespace": "", "db.user": "default", + "db.query.text": "SELECT sum(x) FROM test WHERE x > 150", "sentry.op": "db", "sentry.origin": "auto.db.clickhouse_driver", "server.address": "localhost", @@ -1266,6 +1280,7 @@ def test_clickhouse_dbapi_spans_with_pii( "db.system.name": "clickhouse", "db.namespace": "", "db.user": "default", + "db.query.text": "DROP TABLE IF EXISTS test", "sentry.op": "db", "sentry.origin": "auto.db.clickhouse_driver", "server.address": "localhost", @@ -1280,6 +1295,7 @@ def test_clickhouse_dbapi_spans_with_pii( "db.system.name": "clickhouse", "db.namespace": "", "db.user": "default", + "db.query.text": "CREATE TABLE test (x Int32) ENGINE = Memory", "sentry.op": "db", "sentry.origin": "auto.db.clickhouse_driver", "server.address": "localhost", @@ -1294,6 +1310,7 @@ def test_clickhouse_dbapi_spans_with_pii( "db.system.name": "clickhouse", "db.namespace": "", "db.user": "default", + "db.query.text": "INSERT INTO test (x) VALUES", "sentry.op": "db", "sentry.origin": "auto.db.clickhouse_driver", "server.address": "localhost", @@ -1308,6 +1325,7 @@ def test_clickhouse_dbapi_spans_with_pii( "db.system.name": "clickhouse", "db.namespace": "", "db.user": "default", + "db.query.text": "INSERT INTO test (x) VALUES", "sentry.op": "db", "sentry.origin": "auto.db.clickhouse_driver", "server.address": "localhost", @@ -1322,6 +1340,7 @@ def test_clickhouse_dbapi_spans_with_pii( "db.system.name": "clickhouse", "db.namespace": "", "db.user": "default", + "db.query.text": "SELECT sum(x) FROM test WHERE x > 150", "sentry.op": "db", "sentry.origin": "auto.db.clickhouse_driver", "server.address": "localhost",