Added execute_async to HiveServer2DictCursor

balmar · Aug 19, 2024 · aed1aaf · aed1aaf
1 parent b453413
commit aed1aaf
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/impala/hiveserver2.py b/impala/hiveserver2.py
@@ -787,6 +787,15 @@ def execute(self, operation, parameters=None, configuration=None):
             self.fields = [d[0] for d in self.description]
         else:
             self.fields = None
+
+    # https://github.com/cloudera/impyla/issues/292
+    def execute_async(self, operation, parameters=None, configuration=None):
+        super(self.__class__, self).execute_async(operation, parameters,
+                                                configuration)
+        if self.description is not None:
+            self.fields = [d[0] for d in self.description]
+        else:
+            self.fields = None
 
     def __next__(self):
         record = super(self.__class__, self).__next__()