瀏覽代碼

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

Romain Rigaux 11 年之前
父節點
當前提交
fb0615d
共有 1 個文件被更改,包括 6 次插入0 次删除
  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