Support clickhouse-driver==0.2.7 (#2752)

getsentry · Feb 21, 2024 · e07c0ac · e07c0ac
1 parent e24508f
commit e07c0ac
Showing 1 changed file with 5 additions and 0 deletions.
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
  )