Selaa lähdekoodia

[flink] Add get_sample_data skeleton API

Romain Rigaux 4 vuotta sitten
vanhempi
commit
b2393bf6e3
1 muutettua tiedostoa jossa 21 lisäystä ja 1 poistoa
  1. 21 1
      desktop/libs/notebook/src/notebook/connectors/flink_sql.py

+ 21 - 1
desktop/libs/notebook/src/notebook/connectors/flink_sql.py

@@ -137,7 +137,8 @@ class FlinkSqlApi(Api):
             'comment': ''
           }
           for col in description
-        ] if has_result_set else [],
+        ]
+        if has_result_set else [],
         'type': 'table'
       }
     }
@@ -228,6 +229,25 @@ class FlinkSqlApi(Api):
     return response
 
 
+  @query_error_handler
+  def get_sample_data(self, snippet, database=None, table=None, column=None, is_async=False, operation=None):
+    if operation == 'hello':
+      snippet['statement'] = "SELECT 'Hello World!'"
+
+    notebook = {}
+    sample = self.execute(notebook, snippet)
+
+    response = {
+      'status': 0,
+      'result': {}
+    }
+
+    response['rows'] = sample['result']['data']
+    response['full_headers'] = sample['result']['meta']
+
+    return response
+
+
   def show_databases(self):
     session = self._get_session()
     session_id = session['id']