vdk-impala: enhance memory error handling (#2938)

Add another exception type to be handled by the Impala Memory Handler.

Signed-off-by: Dako Dakov <[email protected]>

projects/vdk-plugins/vdk-impala/src/vdk/plugin/impala/impala_memory_error_handler.py

@@ -37,6 +37,10 @@ def handle_memory_error(self, exception, recovery_cursor):
             exception,
             classname_with_package="impala.error.OperationalError",
             exception_message_matcher_regex=".*Memory limit exceeded:.*",
+        ) or errors.exception_matches(
+            exception,
+            classname_with_package="impala.error.HiveServer2Error",
+            exception_message_matcher_regex=".*Memory limit exceeded:.*",
         ):
             # We are going to try to increase the memory limits and see if the query passes
             # But we won't do anything if the sql statement itself sets a memory limit