Pārlūkot izejas kodu

HUE-2212 [metastore] Memory can leak when getting a sample of data of a table

Romain Rigaux 11 gadi atpakaļ
vecāks
revīzija
fb0615d
1 mainītis faili ar 6 papildinājumiem un 0 dzēšanām
  1. 6 0
      apps/beeswax/src/beeswax/server/dbms.py

+ 6 - 0
apps/beeswax/src/beeswax/server/dbms.py

@@ -348,12 +348,18 @@ class HiveServer2Dbms(object):
     handle = self.client.query(query)
     curr = time.time()
     end = curr + timeout_sec
+
     while curr <= end:
       state = self.client.get_state(handle)
       if state not in (QueryHistory.STATE.running, QueryHistory.STATE.submitted):
         return handle
       time.sleep(sleep_interval)
       curr = time.time()
+
+    try:
+      self.cancel_operation(handle)
+    except:
+      self.close_operation(handle)
     return None