浏览代码

HUE-8728 [jb] Fix optimzer_api test.

jdesjean 6 年之前
父节点
当前提交
2feb5533fa
共有 1 个文件被更改,包括 1 次插入2 次删除
  1. 1 2
      desktop/libs/metadata/src/metadata/optimizer_api.py

+ 1 - 2
desktop/libs/metadata/src/metadata/optimizer_api.py

@@ -26,7 +26,6 @@ from django.views.decorators.http import require_POST
 
 from desktop.lib.django_util import JsonResponse
 from desktop.lib.i18n import force_unicode
-from desktop.lib.thrift_util import unpack_guid
 from desktop.models import Document2
 from libsentry.privilege_checker import MissingSentryPrivilegeException
 from notebook.api import _get_statement
@@ -312,7 +311,7 @@ def _convert_queries(queries_data):
     try:
       snippet = query_data['snippets'][0]
       if 'guid' in snippet['result']['handle']: # Not failed query
-        original_query_id = unpack_guid(base64.decodestring(snippet['result']['handle']['guid']))
+        original_query_id = '%s:%s' % struct.unpack(b"QQ", base64.decodestring(snippet['result']['handle']['guid'])) # unpack_guid uses '%016x:%016x' while optmizer api uses '%s:%s'.
         execution_time = snippet['result']['executionTime'] * 100 if snippet['status'] in ('available', 'expired') else -1
         statement = _clean_query(_get_statement(query_data))
         queries.append((original_query_id, execution_time, statement, snippet.get('database', 'default').strip()))