Add extra packet types in receive_* methods according to server side …

…possible packet types
mymarilyn · Oct 13, 2023 · a771321 · a771321
1 parent 7bb9a63
commit a771321
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/clickhouse_driver/client.py b/clickhouse_driver/client.py
@@ -698,15 +698,18 @@ def receive_end_of_insert_query(self):
             if packet.type == ServerPacketTypes.END_OF_STREAM:
                 break
 
-            elif packet.type == ServerPacketTypes.EXCEPTION:
-                raise packet.exception
-
             elif packet.type == ServerPacketTypes.LOG:
                 log_block(packet.block)
 
+            elif packet.type == ServerPacketTypes.PROGRESS:
+                continue
+
+            elif packet.type == ServerPacketTypes.EXCEPTION:
+                raise packet.exception
+
             else:
                 message = self.connection.unexpected_packet_message(
-                    'Exception, EndOfStream or Log', packet.type
+                    'EndOfStream, Log, Progress or Exception', packet.type
                 )
                 raise errors.UnexpectedPacketFromServerError(message)
 
@@ -727,12 +730,15 @@ def receive_profile_events(self):
             elif packet.type == ServerPacketTypes.PROGRESS:
                 self.last_query.store_progress(packet.progress)
 
+            elif packet.type == ServerPacketTypes.LOG:
+                log_block(packet.block)
+
             elif packet.type == ServerPacketTypes.EXCEPTION:
                 raise packet.exception
 
             else:
                 message = self.connection.unexpected_packet_message(
-                    'ProfileEvent, Progress or Exception', packet.type
+                    'ProfileEvents, Progress, Log or Exception', packet.type
                 )
                 raise errors.UnexpectedPacketFromServerError(message)