فهرست منبع

HUE-6245 [optimizer] Always run optimization checks via HS2 API

Romain Rigaux 8 سال پیش
والد
کامیت
2d27fe7

+ 2 - 1
desktop/libs/notebook/src/notebook/api.py

@@ -31,6 +31,7 @@ from desktop.lib.django_util import JsonResponse
 from desktop.models import Document2, Document
 
 from notebook.connectors.base import get_api, Notebook, QueryExpired, SessionExpired, QueryError, _get_snippet_name
+from notebook.connectors.hiveserver2 import HS2Api
 from notebook.decorators import api_error_handler, check_document_access_permission, check_document_modify_permission
 from notebook.models import escape_rows, make_notebook
 from notebook.views import upgrade_session_properties
@@ -745,7 +746,7 @@ def statement_risk(request):
   notebook = json.loads(request.POST.get('notebook', '{}'))
   snippet = json.loads(request.POST.get('snippet', '{}'))
 
-  api = get_api(request, snippet)
+  api = HS2Api(request.user, snippet)
 
   response['query_complexity'] = api.statement_risk(notebook, snippet)
   response['status'] = 0

+ 1 - 1
desktop/libs/notebook/src/notebook/templates/editor_components.mako

@@ -1113,7 +1113,7 @@ from notebook.conf import ENABLE_QUERY_BUILDER, ENABLE_QUERY_SCHEDULING, ENABLE_
               <i class="fa fa-external-link-square"></i>
             </a>
           </div>
-          <a class="btn" data-bind="click: function() { getExternalStatement(); }"> <i class="fa fa-lg fa-refresh"/> </a>
+          <a class="btn" data-bind="click: function() { getExternalStatement(); }"><i class="fa fa-lg fa-refresh"></i></a>
           <a class="btn" data-bind="tooltip: { placement: 'bottom', title: 'Save content back to file' }, click: function() { huePubSub.publish('show.saveToFile.modal'); }"><i class="fa fa-save"></i></a>
           <!-- /ko -->
         </div>