Support clickhouse-driver==0.2.7 (#2752)

sentrivana · web-flow · commit e07c0ac6d4bf · 2024-02-21T10:27:12.000Z
diff --git a/sentry_sdk/integrations/clickhouse_driver.py b/sentry_sdk/integrations/clickhouse_driver.py
@@ -59,6 +59,11 @@ def setup_once() -> None:
         clickhouse_driver.client.Client.receive_end_of_query = _wrap_end(
             clickhouse_driver.client.Client.receive_end_of_query
         )
+        if hasattr(clickhouse_driver.client.Client, "receive_end_of_insert_query"):
+            # In 0.2.7, insert queries are handled separately via `receive_end_of_insert_query`
+            clickhouse_driver.client.Client.receive_end_of_insert_query = _wrap_end(
+                clickhouse_driver.client.Client.receive_end_of_insert_query
+            )
         clickhouse_driver.client.Client.receive_result = _wrap_end(
             clickhouse_driver.client.Client.receive_result
         )

Original file line number	Diff line number	Diff line change
`@@ -59,6 +59,11 @@ def setup_once() -> None:`
`59`	`59`	`clickhouse_driver.client.Client.receive_end_of_query = _wrap_end(`
`60`	`60`	`clickhouse_driver.client.Client.receive_end_of_query`
`61`	`61`	`)`
	`62`	`+ if hasattr(clickhouse_driver.client.Client, "receive_end_of_insert_query"):`
	`63`	+ # In 0.2.7, insert queries are handled separately via `receive_end_of_insert_query`
	`64`	`+ clickhouse_driver.client.Client.receive_end_of_insert_query = _wrap_end(`
	`65`	`+ clickhouse_driver.client.Client.receive_end_of_insert_query`
	`66`	`+ )`
`62`	`67`	`clickhouse_driver.client.Client.receive_result = _wrap_end(`
`63`	`68`	`clickhouse_driver.client.Client.receive_result`
`64`	`69`	`)`