Browse Source

HUE-9077 [ksql] Properly handle SELECT that fails

Romain 6 years ago
parent
commit
9f1559e6f5
1 changed files with 2 additions and 0 deletions
  1. 2 0
      desktop/libs/kafka/src/kafka/ksql_client.py

+ 2 - 0
desktop/libs/kafka/src/kafka/ksql_client.py

@@ -131,6 +131,8 @@ class KSqlApi(object):
         # data.append([line[col] for col in columns])
         # data.append([line[col] for col in columns])
         if is_select and line: # Empty first 2 lines?
         if is_select and line: # Empty first 2 lines?
           data_line = json.loads(line)
           data_line = json.loads(line)
+          if data_line.get('@type') == 'statement_error':
+            raise KSqlApiException(data_line['message'])
           if data_line['row']: # If limit not reached
           if data_line['row']: # If limit not reached
             data.append(data_line['row']['columns'])
             data.append(data_line['row']['columns'])
         else:
         else: