9 from thrift.Thrift
import TType, TMessageType, TFrozenDict, TException, TApplicationException
10 from thrift.protocol.TProtocol
import TProtocolException
11 from thrift.TRecursive
import fix_spec
16 from thrift.Thrift
import TProcessor
17 from thrift.transport
import TTransport
155 - include_system_columns
238 def set_cur_session(self, parent_session, leaf_session, start_time_str, label, for_running_query_kernel):
245 - for_running_query_kernel
257 - for_running_query_kernel
347 def sql_execute(self, session, query, column_format, nonce, first_n, at_most_n):
360 def sql_execute_df(self, session, query, device_type, device_id, first_n, transport_method):
432 def render_vega(self, session, widget_id, vega_json, compression_level, nonce):
489 - custom_expression_ids
512 def create_dashboard(self, session, dashboard_name, dashboard_state, image_hash, dashboard_metadata):
524 def replace_dashboard(self, session, dashboard_id, dashboard_name, dashboard_owner, dashboard_state, image_hash, dashboard_metadata):
567 def share_dashboard(self, session, dashboard_id, groups, objects, permissions, grant_role):
660 - assign_render_groups
676 def load_table(self, session, table_name, rows, column_names):
719 def import_geo_table(self, session, table_name, file_name, copy_params, row_desc, create_params):
789 def start_query(self, leaf_session, parent_session, query_ra, start_time_str, just_explain, outer_fragment_indices):
797 - outer_fragment_indices
819 - is_final_subquery_result
916 def has_role(self, session, granteeName, roleName):
1027 class Client(Iface):
1030 if oprot
is not None:
1046 self._oprot.writeMessageBegin(
'connect', TMessageType.CALL, self.
_seqid)
1049 args.passwd = passwd
1050 args.dbname = dbname
1052 self._oprot.writeMessageEnd()
1053 self._oprot.trans.flush()
1057 (fname, mtype, rseqid) = iprot.readMessageBegin()
1058 if mtype == TMessageType.EXCEPTION:
1059 x = TApplicationException()
1061 iprot.readMessageEnd()
1065 iprot.readMessageEnd()
1066 if result.success
is not None:
1067 return result.success
1068 if result.e
is not None:
1070 raise TApplicationException(TApplicationException.MISSING_RESULT,
"connect failed: unknown result")
1083 self._oprot.writeMessageBegin(
'krb5_connect', TMessageType.CALL, self.
_seqid)
1085 args.inputToken = inputToken
1086 args.dbname = dbname
1088 self._oprot.writeMessageEnd()
1089 self._oprot.trans.flush()
1093 (fname, mtype, rseqid) = iprot.readMessageBegin()
1094 if mtype == TMessageType.EXCEPTION:
1095 x = TApplicationException()
1097 iprot.readMessageEnd()
1101 iprot.readMessageEnd()
1102 if result.success
is not None:
1103 return result.success
1104 if result.e
is not None:
1106 raise TApplicationException(TApplicationException.MISSING_RESULT,
"krb5_connect failed: unknown result")
1118 self._oprot.writeMessageBegin(
'disconnect', TMessageType.CALL, self.
_seqid)
1120 args.session = session
1122 self._oprot.writeMessageEnd()
1123 self._oprot.trans.flush()
1127 (fname, mtype, rseqid) = iprot.readMessageBegin()
1128 if mtype == TMessageType.EXCEPTION:
1129 x = TApplicationException()
1131 iprot.readMessageEnd()
1135 iprot.readMessageEnd()
1136 if result.e
is not None:
1151 self._oprot.writeMessageBegin(
'switch_database', TMessageType.CALL, self.
_seqid)
1153 args.session = session
1154 args.dbname = dbname
1156 self._oprot.writeMessageEnd()
1157 self._oprot.trans.flush()
1161 (fname, mtype, rseqid) = iprot.readMessageBegin()
1162 if mtype == TMessageType.EXCEPTION:
1163 x = TApplicationException()
1165 iprot.readMessageEnd()
1169 iprot.readMessageEnd()
1170 if result.e
is not None:
1184 self._oprot.writeMessageBegin(
'clone_session', TMessageType.CALL, self.
_seqid)
1186 args.session = session
1188 self._oprot.writeMessageEnd()
1189 self._oprot.trans.flush()
1193 (fname, mtype, rseqid) = iprot.readMessageBegin()
1194 if mtype == TMessageType.EXCEPTION:
1195 x = TApplicationException()
1197 iprot.readMessageEnd()
1201 iprot.readMessageEnd()
1202 if result.success
is not None:
1203 return result.success
1204 if result.e
is not None:
1206 raise TApplicationException(TApplicationException.MISSING_RESULT,
"clone_session failed: unknown result")
1218 self._oprot.writeMessageBegin(
'get_server_status', TMessageType.CALL, self.
_seqid)
1220 args.session = session
1222 self._oprot.writeMessageEnd()
1223 self._oprot.trans.flush()
1227 (fname, mtype, rseqid) = iprot.readMessageBegin()
1228 if mtype == TMessageType.EXCEPTION:
1229 x = TApplicationException()
1231 iprot.readMessageEnd()
1235 iprot.readMessageEnd()
1236 if result.success
is not None:
1237 return result.success
1238 if result.e
is not None:
1240 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_server_status failed: unknown result")
1252 self._oprot.writeMessageBegin(
'get_status', TMessageType.CALL, self.
_seqid)
1254 args.session = session
1256 self._oprot.writeMessageEnd()
1257 self._oprot.trans.flush()
1261 (fname, mtype, rseqid) = iprot.readMessageBegin()
1262 if mtype == TMessageType.EXCEPTION:
1263 x = TApplicationException()
1265 iprot.readMessageEnd()
1269 iprot.readMessageEnd()
1270 if result.success
is not None:
1271 return result.success
1272 if result.e
is not None:
1274 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_status failed: unknown result")
1286 self._oprot.writeMessageBegin(
'get_hardware_info', TMessageType.CALL, self.
_seqid)
1288 args.session = session
1290 self._oprot.writeMessageEnd()
1291 self._oprot.trans.flush()
1295 (fname, mtype, rseqid) = iprot.readMessageBegin()
1296 if mtype == TMessageType.EXCEPTION:
1297 x = TApplicationException()
1299 iprot.readMessageEnd()
1303 iprot.readMessageEnd()
1304 if result.success
is not None:
1305 return result.success
1306 if result.e
is not None:
1308 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_hardware_info failed: unknown result")
1320 self._oprot.writeMessageBegin(
'get_tables', TMessageType.CALL, self.
_seqid)
1322 args.session = session
1324 self._oprot.writeMessageEnd()
1325 self._oprot.trans.flush()
1329 (fname, mtype, rseqid) = iprot.readMessageBegin()
1330 if mtype == TMessageType.EXCEPTION:
1331 x = TApplicationException()
1333 iprot.readMessageEnd()
1337 iprot.readMessageEnd()
1338 if result.success
is not None:
1339 return result.success
1340 if result.e
is not None:
1342 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_tables failed: unknown result")
1355 self._oprot.writeMessageBegin(
'get_tables_for_database', TMessageType.CALL, self.
_seqid)
1357 args.session = session
1358 args.database_name = database_name
1360 self._oprot.writeMessageEnd()
1361 self._oprot.trans.flush()
1365 (fname, mtype, rseqid) = iprot.readMessageBegin()
1366 if mtype == TMessageType.EXCEPTION:
1367 x = TApplicationException()
1369 iprot.readMessageEnd()
1373 iprot.readMessageEnd()
1374 if result.success
is not None:
1375 return result.success
1376 if result.e
is not None:
1378 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_tables_for_database failed: unknown result")
1390 self._oprot.writeMessageBegin(
'get_physical_tables', TMessageType.CALL, self.
_seqid)
1392 args.session = session
1394 self._oprot.writeMessageEnd()
1395 self._oprot.trans.flush()
1399 (fname, mtype, rseqid) = iprot.readMessageBegin()
1400 if mtype == TMessageType.EXCEPTION:
1401 x = TApplicationException()
1403 iprot.readMessageEnd()
1407 iprot.readMessageEnd()
1408 if result.success
is not None:
1409 return result.success
1410 if result.e
is not None:
1412 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_physical_tables failed: unknown result")
1424 self._oprot.writeMessageBegin(
'get_views', TMessageType.CALL, self.
_seqid)
1426 args.session = session
1428 self._oprot.writeMessageEnd()
1429 self._oprot.trans.flush()
1433 (fname, mtype, rseqid) = iprot.readMessageBegin()
1434 if mtype == TMessageType.EXCEPTION:
1435 x = TApplicationException()
1437 iprot.readMessageEnd()
1441 iprot.readMessageEnd()
1442 if result.success
is not None:
1443 return result.success
1444 if result.e
is not None:
1446 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_views failed: unknown result")
1458 self._oprot.writeMessageBegin(
'get_tables_meta', TMessageType.CALL, self.
_seqid)
1460 args.session = session
1462 self._oprot.writeMessageEnd()
1463 self._oprot.trans.flush()
1467 (fname, mtype, rseqid) = iprot.readMessageBegin()
1468 if mtype == TMessageType.EXCEPTION:
1469 x = TApplicationException()
1471 iprot.readMessageEnd()
1475 iprot.readMessageEnd()
1476 if result.success
is not None:
1477 return result.success
1478 if result.e
is not None:
1480 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_tables_meta failed: unknown result")
1493 self._oprot.writeMessageBegin(
'get_table_details', TMessageType.CALL, self.
_seqid)
1495 args.session = session
1496 args.table_name = table_name
1498 self._oprot.writeMessageEnd()
1499 self._oprot.trans.flush()
1503 (fname, mtype, rseqid) = iprot.readMessageBegin()
1504 if mtype == TMessageType.EXCEPTION:
1505 x = TApplicationException()
1507 iprot.readMessageEnd()
1511 iprot.readMessageEnd()
1512 if result.success
is not None:
1513 return result.success
1514 if result.e
is not None:
1516 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_table_details failed: unknown result")
1530 self._oprot.writeMessageBegin(
'get_table_details_for_database', TMessageType.CALL, self.
_seqid)
1532 args.session = session
1533 args.table_name = table_name
1534 args.database_name = database_name
1536 self._oprot.writeMessageEnd()
1537 self._oprot.trans.flush()
1541 (fname, mtype, rseqid) = iprot.readMessageBegin()
1542 if mtype == TMessageType.EXCEPTION:
1543 x = TApplicationException()
1545 iprot.readMessageEnd()
1549 iprot.readMessageEnd()
1550 if result.success
is not None:
1551 return result.success
1552 if result.e
is not None:
1554 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_table_details_for_database failed: unknown result")
1561 - include_system_columns
1568 self._oprot.writeMessageBegin(
'get_internal_table_details', TMessageType.CALL, self.
_seqid)
1570 args.session = session
1571 args.table_name = table_name
1572 args.include_system_columns = include_system_columns
1574 self._oprot.writeMessageEnd()
1575 self._oprot.trans.flush()
1579 (fname, mtype, rseqid) = iprot.readMessageBegin()
1580 if mtype == TMessageType.EXCEPTION:
1581 x = TApplicationException()
1583 iprot.readMessageEnd()
1587 iprot.readMessageEnd()
1588 if result.success
is not None:
1589 return result.success
1590 if result.e
is not None:
1592 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_internal_table_details failed: unknown result")
1606 self._oprot.writeMessageBegin(
'get_internal_table_details_for_database', TMessageType.CALL, self.
_seqid)
1608 args.session = session
1609 args.table_name = table_name
1610 args.database_name = database_name
1612 self._oprot.writeMessageEnd()
1613 self._oprot.trans.flush()
1617 (fname, mtype, rseqid) = iprot.readMessageBegin()
1618 if mtype == TMessageType.EXCEPTION:
1619 x = TApplicationException()
1621 iprot.readMessageEnd()
1625 iprot.readMessageEnd()
1626 if result.success
is not None:
1627 return result.success
1628 if result.e
is not None:
1630 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_internal_table_details_for_database failed: unknown result")
1642 self._oprot.writeMessageBegin(
'get_users', TMessageType.CALL, self.
_seqid)
1644 args.session = session
1646 self._oprot.writeMessageEnd()
1647 self._oprot.trans.flush()
1651 (fname, mtype, rseqid) = iprot.readMessageBegin()
1652 if mtype == TMessageType.EXCEPTION:
1653 x = TApplicationException()
1655 iprot.readMessageEnd()
1659 iprot.readMessageEnd()
1660 if result.success
is not None:
1661 return result.success
1662 if result.e
is not None:
1664 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_users failed: unknown result")
1676 self._oprot.writeMessageBegin(
'get_databases', TMessageType.CALL, self.
_seqid)
1678 args.session = session
1680 self._oprot.writeMessageEnd()
1681 self._oprot.trans.flush()
1685 (fname, mtype, rseqid) = iprot.readMessageBegin()
1686 if mtype == TMessageType.EXCEPTION:
1687 x = TApplicationException()
1689 iprot.readMessageEnd()
1693 iprot.readMessageEnd()
1694 if result.success
is not None:
1695 return result.success
1696 if result.e
is not None:
1698 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_databases failed: unknown result")
1705 self._oprot.writeMessageBegin(
'get_version', TMessageType.CALL, self.
_seqid)
1708 self._oprot.writeMessageEnd()
1709 self._oprot.trans.flush()
1713 (fname, mtype, rseqid) = iprot.readMessageBegin()
1714 if mtype == TMessageType.EXCEPTION:
1715 x = TApplicationException()
1717 iprot.readMessageEnd()
1721 iprot.readMessageEnd()
1722 if result.success
is not None:
1723 return result.success
1724 if result.e
is not None:
1726 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_version failed: unknown result")
1738 self._oprot.writeMessageBegin(
'start_heap_profile', TMessageType.CALL, self.
_seqid)
1740 args.session = session
1742 self._oprot.writeMessageEnd()
1743 self._oprot.trans.flush()
1747 (fname, mtype, rseqid) = iprot.readMessageBegin()
1748 if mtype == TMessageType.EXCEPTION:
1749 x = TApplicationException()
1751 iprot.readMessageEnd()
1755 iprot.readMessageEnd()
1756 if result.e
is not None:
1770 self._oprot.writeMessageBegin(
'stop_heap_profile', TMessageType.CALL, self.
_seqid)
1772 args.session = session
1774 self._oprot.writeMessageEnd()
1775 self._oprot.trans.flush()
1779 (fname, mtype, rseqid) = iprot.readMessageBegin()
1780 if mtype == TMessageType.EXCEPTION:
1781 x = TApplicationException()
1783 iprot.readMessageEnd()
1787 iprot.readMessageEnd()
1788 if result.e
is not None:
1802 self._oprot.writeMessageBegin(
'get_heap_profile', TMessageType.CALL, self.
_seqid)
1804 args.session = session
1806 self._oprot.writeMessageEnd()
1807 self._oprot.trans.flush()
1811 (fname, mtype, rseqid) = iprot.readMessageBegin()
1812 if mtype == TMessageType.EXCEPTION:
1813 x = TApplicationException()
1815 iprot.readMessageEnd()
1819 iprot.readMessageEnd()
1820 if result.success
is not None:
1821 return result.success
1822 if result.e
is not None:
1824 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_heap_profile failed: unknown result")
1837 self._oprot.writeMessageBegin(
'get_memory', TMessageType.CALL, self.
_seqid)
1839 args.session = session
1840 args.memory_level = memory_level
1842 self._oprot.writeMessageEnd()
1843 self._oprot.trans.flush()
1847 (fname, mtype, rseqid) = iprot.readMessageBegin()
1848 if mtype == TMessageType.EXCEPTION:
1849 x = TApplicationException()
1851 iprot.readMessageEnd()
1855 iprot.readMessageEnd()
1856 if result.success
is not None:
1857 return result.success
1858 if result.e
is not None:
1860 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_memory failed: unknown result")
1872 self._oprot.writeMessageBegin(
'clear_cpu_memory', TMessageType.CALL, self.
_seqid)
1874 args.session = session
1876 self._oprot.writeMessageEnd()
1877 self._oprot.trans.flush()
1881 (fname, mtype, rseqid) = iprot.readMessageBegin()
1882 if mtype == TMessageType.EXCEPTION:
1883 x = TApplicationException()
1885 iprot.readMessageEnd()
1889 iprot.readMessageEnd()
1890 if result.e
is not None:
1904 self._oprot.writeMessageBegin(
'clear_gpu_memory', TMessageType.CALL, self.
_seqid)
1906 args.session = session
1908 self._oprot.writeMessageEnd()
1909 self._oprot.trans.flush()
1913 (fname, mtype, rseqid) = iprot.readMessageBegin()
1914 if mtype == TMessageType.EXCEPTION:
1915 x = TApplicationException()
1917 iprot.readMessageEnd()
1921 iprot.readMessageEnd()
1922 if result.e
is not None:
1926 def set_cur_session(self, parent_session, leaf_session, start_time_str, label, for_running_query_kernel):
1933 - for_running_query_kernel
1936 self.
send_set_cur_session(parent_session, leaf_session, start_time_str, label, for_running_query_kernel)
1940 self._oprot.writeMessageBegin(
'set_cur_session', TMessageType.CALL, self.
_seqid)
1942 args.parent_session = parent_session
1943 args.leaf_session = leaf_session
1944 args.start_time_str = start_time_str
1946 args.for_running_query_kernel = for_running_query_kernel
1948 self._oprot.writeMessageEnd()
1949 self._oprot.trans.flush()
1953 (fname, mtype, rseqid) = iprot.readMessageBegin()
1954 if mtype == TMessageType.EXCEPTION:
1955 x = TApplicationException()
1957 iprot.readMessageEnd()
1961 iprot.readMessageEnd()
1962 if result.e
is not None:
1973 - for_running_query_kernel
1980 self._oprot.writeMessageBegin(
'invalidate_cur_session', TMessageType.CALL, self.
_seqid)
1982 args.parent_session = parent_session
1983 args.leaf_session = leaf_session
1984 args.start_time_str = start_time_str
1986 args.for_running_query_kernel = for_running_query_kernel
1988 self._oprot.writeMessageEnd()
1989 self._oprot.trans.flush()
1993 (fname, mtype, rseqid) = iprot.readMessageBegin()
1994 if mtype == TMessageType.EXCEPTION:
1995 x = TApplicationException()
1997 iprot.readMessageEnd()
2001 iprot.readMessageEnd()
2002 if result.e
is not None:
2019 self._oprot.writeMessageBegin(
'set_table_epoch', TMessageType.CALL, self.
_seqid)
2021 args.session = session
2023 args.table_id = table_id
2024 args.new_epoch = new_epoch
2026 self._oprot.writeMessageEnd()
2027 self._oprot.trans.flush()
2031 (fname, mtype, rseqid) = iprot.readMessageBegin()
2032 if mtype == TMessageType.EXCEPTION:
2033 x = TApplicationException()
2035 iprot.readMessageEnd()
2039 iprot.readMessageEnd()
2040 if result.e
is not None:
2056 self._oprot.writeMessageBegin(
'set_table_epoch_by_name', TMessageType.CALL, self.
_seqid)
2058 args.session = session
2059 args.table_name = table_name
2060 args.new_epoch = new_epoch
2062 self._oprot.writeMessageEnd()
2063 self._oprot.trans.flush()
2067 (fname, mtype, rseqid) = iprot.readMessageBegin()
2068 if mtype == TMessageType.EXCEPTION:
2069 x = TApplicationException()
2071 iprot.readMessageEnd()
2075 iprot.readMessageEnd()
2076 if result.e
is not None:
2092 self._oprot.writeMessageBegin(
'get_table_epoch', TMessageType.CALL, self.
_seqid)
2094 args.session = session
2096 args.table_id = table_id
2098 self._oprot.writeMessageEnd()
2099 self._oprot.trans.flush()
2103 (fname, mtype, rseqid) = iprot.readMessageBegin()
2104 if mtype == TMessageType.EXCEPTION:
2105 x = TApplicationException()
2107 iprot.readMessageEnd()
2111 iprot.readMessageEnd()
2112 if result.success
is not None:
2113 return result.success
2114 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_table_epoch failed: unknown result")
2127 self._oprot.writeMessageBegin(
'get_table_epoch_by_name', TMessageType.CALL, self.
_seqid)
2129 args.session = session
2130 args.table_name = table_name
2132 self._oprot.writeMessageEnd()
2133 self._oprot.trans.flush()
2137 (fname, mtype, rseqid) = iprot.readMessageBegin()
2138 if mtype == TMessageType.EXCEPTION:
2139 x = TApplicationException()
2141 iprot.readMessageEnd()
2145 iprot.readMessageEnd()
2146 if result.success
is not None:
2147 return result.success
2148 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_table_epoch_by_name failed: unknown result")
2162 self._oprot.writeMessageBegin(
'get_table_epochs', TMessageType.CALL, self.
_seqid)
2164 args.session = session
2166 args.table_id = table_id
2168 self._oprot.writeMessageEnd()
2169 self._oprot.trans.flush()
2173 (fname, mtype, rseqid) = iprot.readMessageBegin()
2174 if mtype == TMessageType.EXCEPTION:
2175 x = TApplicationException()
2177 iprot.readMessageEnd()
2181 iprot.readMessageEnd()
2182 if result.success
is not None:
2183 return result.success
2184 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_table_epochs failed: unknown result")
2198 self._oprot.writeMessageBegin(
'set_table_epochs', TMessageType.CALL, self.
_seqid)
2200 args.session = session
2202 args.table_epochs = table_epochs
2204 self._oprot.writeMessageEnd()
2205 self._oprot.trans.flush()
2209 (fname, mtype, rseqid) = iprot.readMessageBegin()
2210 if mtype == TMessageType.EXCEPTION:
2211 x = TApplicationException()
2213 iprot.readMessageEnd()
2217 iprot.readMessageEnd()
2230 self._oprot.writeMessageBegin(
'get_session_info', TMessageType.CALL, self.
_seqid)
2232 args.session = session
2234 self._oprot.writeMessageEnd()
2235 self._oprot.trans.flush()
2239 (fname, mtype, rseqid) = iprot.readMessageBegin()
2240 if mtype == TMessageType.EXCEPTION:
2241 x = TApplicationException()
2243 iprot.readMessageEnd()
2247 iprot.readMessageEnd()
2248 if result.success
is not None:
2249 return result.success
2250 if result.e
is not None:
2252 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_session_info failed: unknown result")
2264 self._oprot.writeMessageBegin(
'get_queries_info', TMessageType.CALL, self.
_seqid)
2266 args.session = session
2268 self._oprot.writeMessageEnd()
2269 self._oprot.trans.flush()
2273 (fname, mtype, rseqid) = iprot.readMessageBegin()
2274 if mtype == TMessageType.EXCEPTION:
2275 x = TApplicationException()
2277 iprot.readMessageEnd()
2281 iprot.readMessageEnd()
2282 if result.success
is not None:
2283 return result.success
2284 if result.e
is not None:
2286 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_queries_info failed: unknown result")
2299 self._oprot.writeMessageBegin(
'set_leaf_info', TMessageType.CALL, self.
_seqid)
2301 args.session = session
2302 args.leaf_info = leaf_info
2304 self._oprot.writeMessageEnd()
2305 self._oprot.trans.flush()
2309 (fname, mtype, rseqid) = iprot.readMessageBegin()
2310 if mtype == TMessageType.EXCEPTION:
2311 x = TApplicationException()
2313 iprot.readMessageEnd()
2317 iprot.readMessageEnd()
2318 if result.e
is not None:
2322 def sql_execute(self, session, query, column_format, nonce, first_n, at_most_n):
2333 self.
send_sql_execute(session, query, column_format, nonce, first_n, at_most_n)
2337 self._oprot.writeMessageBegin(
'sql_execute', TMessageType.CALL, self.
_seqid)
2339 args.session = session
2341 args.column_format = column_format
2343 args.first_n = first_n
2344 args.at_most_n = at_most_n
2346 self._oprot.writeMessageEnd()
2347 self._oprot.trans.flush()
2351 (fname, mtype, rseqid) = iprot.readMessageBegin()
2352 if mtype == TMessageType.EXCEPTION:
2353 x = TApplicationException()
2355 iprot.readMessageEnd()
2359 iprot.readMessageEnd()
2360 if result.success
is not None:
2361 return result.success
2362 if result.e
is not None:
2364 raise TApplicationException(TApplicationException.MISSING_RESULT,
"sql_execute failed: unknown result")
2366 def sql_execute_df(self, session, query, device_type, device_id, first_n, transport_method):
2377 self.
send_sql_execute_df(session, query, device_type, device_id, first_n, transport_method)
2381 self._oprot.writeMessageBegin(
'sql_execute_df', TMessageType.CALL, self.
_seqid)
2383 args.session = session
2385 args.device_type = device_type
2386 args.device_id = device_id
2387 args.first_n = first_n
2388 args.transport_method = transport_method
2390 self._oprot.writeMessageEnd()
2391 self._oprot.trans.flush()
2395 (fname, mtype, rseqid) = iprot.readMessageBegin()
2396 if mtype == TMessageType.EXCEPTION:
2397 x = TApplicationException()
2399 iprot.readMessageEnd()
2403 iprot.readMessageEnd()
2404 if result.success
is not None:
2405 return result.success
2406 if result.e
is not None:
2408 raise TApplicationException(TApplicationException.MISSING_RESULT,
"sql_execute_df failed: unknown result")
2423 self._oprot.writeMessageBegin(
'sql_execute_gdf', TMessageType.CALL, self.
_seqid)
2425 args.session = session
2427 args.device_id = device_id
2428 args.first_n = first_n
2430 self._oprot.writeMessageEnd()
2431 self._oprot.trans.flush()
2435 (fname, mtype, rseqid) = iprot.readMessageBegin()
2436 if mtype == TMessageType.EXCEPTION:
2437 x = TApplicationException()
2439 iprot.readMessageEnd()
2443 iprot.readMessageEnd()
2444 if result.success
is not None:
2445 return result.success
2446 if result.e
is not None:
2448 raise TApplicationException(TApplicationException.MISSING_RESULT,
"sql_execute_gdf failed: unknown result")
2463 self._oprot.writeMessageBegin(
'deallocate_df', TMessageType.CALL, self.
_seqid)
2465 args.session = session
2467 args.device_type = device_type
2468 args.device_id = device_id
2470 self._oprot.writeMessageEnd()
2471 self._oprot.trans.flush()
2475 (fname, mtype, rseqid) = iprot.readMessageBegin()
2476 if mtype == TMessageType.EXCEPTION:
2477 x = TApplicationException()
2479 iprot.readMessageEnd()
2483 iprot.readMessageEnd()
2484 if result.e
is not None:
2499 self._oprot.writeMessageBegin(
'interrupt', TMessageType.CALL, self.
_seqid)
2501 args.query_session = query_session
2502 args.interrupt_session = interrupt_session
2504 self._oprot.writeMessageEnd()
2505 self._oprot.trans.flush()
2509 (fname, mtype, rseqid) = iprot.readMessageBegin()
2510 if mtype == TMessageType.EXCEPTION:
2511 x = TApplicationException()
2513 iprot.readMessageEnd()
2517 iprot.readMessageEnd()
2518 if result.e
is not None:
2533 self._oprot.writeMessageBegin(
'sql_validate', TMessageType.CALL, self.
_seqid)
2535 args.session = session
2538 self._oprot.writeMessageEnd()
2539 self._oprot.trans.flush()
2543 (fname, mtype, rseqid) = iprot.readMessageBegin()
2544 if mtype == TMessageType.EXCEPTION:
2545 x = TApplicationException()
2547 iprot.readMessageEnd()
2551 iprot.readMessageEnd()
2552 if result.success
is not None:
2553 return result.success
2554 if result.e
is not None:
2556 raise TApplicationException(TApplicationException.MISSING_RESULT,
"sql_validate failed: unknown result")
2570 self._oprot.writeMessageBegin(
'get_completion_hints', TMessageType.CALL, self.
_seqid)
2572 args.session = session
2574 args.cursor = cursor
2576 self._oprot.writeMessageEnd()
2577 self._oprot.trans.flush()
2581 (fname, mtype, rseqid) = iprot.readMessageBegin()
2582 if mtype == TMessageType.EXCEPTION:
2583 x = TApplicationException()
2585 iprot.readMessageEnd()
2589 iprot.readMessageEnd()
2590 if result.success
is not None:
2591 return result.success
2592 if result.e
is not None:
2594 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_completion_hints failed: unknown result")
2607 self._oprot.writeMessageBegin(
'set_execution_mode', TMessageType.CALL, self.
_seqid)
2609 args.session = session
2612 self._oprot.writeMessageEnd()
2613 self._oprot.trans.flush()
2617 (fname, mtype, rseqid) = iprot.readMessageBegin()
2618 if mtype == TMessageType.EXCEPTION:
2619 x = TApplicationException()
2621 iprot.readMessageEnd()
2625 iprot.readMessageEnd()
2626 if result.e
is not None:
2630 def render_vega(self, session, widget_id, vega_json, compression_level, nonce):
2640 self.
send_render_vega(session, widget_id, vega_json, compression_level, nonce)
2644 self._oprot.writeMessageBegin(
'render_vega', TMessageType.CALL, self.
_seqid)
2646 args.session = session
2647 args.widget_id = widget_id
2648 args.vega_json = vega_json
2649 args.compression_level = compression_level
2652 self._oprot.writeMessageEnd()
2653 self._oprot.trans.flush()
2657 (fname, mtype, rseqid) = iprot.readMessageBegin()
2658 if mtype == TMessageType.EXCEPTION:
2659 x = TApplicationException()
2661 iprot.readMessageEnd()
2665 iprot.readMessageEnd()
2666 if result.success
is not None:
2667 return result.success
2668 if result.e
is not None:
2670 raise TApplicationException(TApplicationException.MISSING_RESULT,
"render_vega failed: unknown result")
2688 self._oprot.writeMessageBegin(
'get_result_row_for_pixel', TMessageType.CALL, self.
_seqid)
2690 args.session = session
2691 args.widget_id = widget_id
2693 args.table_col_names = table_col_names
2694 args.column_format = column_format
2695 args.pixelRadius = pixelRadius
2698 self._oprot.writeMessageEnd()
2699 self._oprot.trans.flush()
2703 (fname, mtype, rseqid) = iprot.readMessageBegin()
2704 if mtype == TMessageType.EXCEPTION:
2705 x = TApplicationException()
2707 iprot.readMessageEnd()
2711 iprot.readMessageEnd()
2712 if result.success
is not None:
2713 return result.success
2714 if result.e
is not None:
2716 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_result_row_for_pixel failed: unknown result")
2729 self._oprot.writeMessageBegin(
'create_custom_expression', TMessageType.CALL, self.
_seqid)
2731 args.session = session
2732 args.custom_expression = custom_expression
2734 self._oprot.writeMessageEnd()
2735 self._oprot.trans.flush()
2739 (fname, mtype, rseqid) = iprot.readMessageBegin()
2740 if mtype == TMessageType.EXCEPTION:
2741 x = TApplicationException()
2743 iprot.readMessageEnd()
2747 iprot.readMessageEnd()
2748 if result.success
is not None:
2749 return result.success
2750 if result.e
is not None:
2752 raise TApplicationException(TApplicationException.MISSING_RESULT,
"create_custom_expression failed: unknown result")
2764 self._oprot.writeMessageBegin(
'get_custom_expressions', TMessageType.CALL, self.
_seqid)
2766 args.session = session
2768 self._oprot.writeMessageEnd()
2769 self._oprot.trans.flush()
2773 (fname, mtype, rseqid) = iprot.readMessageBegin()
2774 if mtype == TMessageType.EXCEPTION:
2775 x = TApplicationException()
2777 iprot.readMessageEnd()
2781 iprot.readMessageEnd()
2782 if result.success
is not None:
2783 return result.success
2784 if result.e
is not None:
2786 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_custom_expressions failed: unknown result")
2800 self._oprot.writeMessageBegin(
'update_custom_expression', TMessageType.CALL, self.
_seqid)
2802 args.session = session
2804 args.expression_json = expression_json
2806 self._oprot.writeMessageEnd()
2807 self._oprot.trans.flush()
2811 (fname, mtype, rseqid) = iprot.readMessageBegin()
2812 if mtype == TMessageType.EXCEPTION:
2813 x = TApplicationException()
2815 iprot.readMessageEnd()
2819 iprot.readMessageEnd()
2820 if result.e
is not None:
2828 - custom_expression_ids
2836 self._oprot.writeMessageBegin(
'delete_custom_expressions', TMessageType.CALL, self.
_seqid)
2838 args.session = session
2839 args.custom_expression_ids = custom_expression_ids
2840 args.do_soft_delete = do_soft_delete
2842 self._oprot.writeMessageEnd()
2843 self._oprot.trans.flush()
2847 (fname, mtype, rseqid) = iprot.readMessageBegin()
2848 if mtype == TMessageType.EXCEPTION:
2849 x = TApplicationException()
2851 iprot.readMessageEnd()
2855 iprot.readMessageEnd()
2856 if result.e
is not None:
2871 self._oprot.writeMessageBegin(
'get_dashboard', TMessageType.CALL, self.
_seqid)
2873 args.session = session
2874 args.dashboard_id = dashboard_id
2876 self._oprot.writeMessageEnd()
2877 self._oprot.trans.flush()
2881 (fname, mtype, rseqid) = iprot.readMessageBegin()
2882 if mtype == TMessageType.EXCEPTION:
2883 x = TApplicationException()
2885 iprot.readMessageEnd()
2889 iprot.readMessageEnd()
2890 if result.success
is not None:
2891 return result.success
2892 if result.e
is not None:
2894 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_dashboard failed: unknown result")
2906 self._oprot.writeMessageBegin(
'get_dashboards', TMessageType.CALL, self.
_seqid)
2908 args.session = session
2910 self._oprot.writeMessageEnd()
2911 self._oprot.trans.flush()
2915 (fname, mtype, rseqid) = iprot.readMessageBegin()
2916 if mtype == TMessageType.EXCEPTION:
2917 x = TApplicationException()
2919 iprot.readMessageEnd()
2923 iprot.readMessageEnd()
2924 if result.success
is not None:
2925 return result.success
2926 if result.e
is not None:
2928 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_dashboards failed: unknown result")
2930 def create_dashboard(self, session, dashboard_name, dashboard_state, image_hash, dashboard_metadata):
2937 - dashboard_metadata
2940 self.
send_create_dashboard(session, dashboard_name, dashboard_state, image_hash, dashboard_metadata)
2944 self._oprot.writeMessageBegin(
'create_dashboard', TMessageType.CALL, self.
_seqid)
2946 args.session = session
2947 args.dashboard_name = dashboard_name
2948 args.dashboard_state = dashboard_state
2949 args.image_hash = image_hash
2950 args.dashboard_metadata = dashboard_metadata
2952 self._oprot.writeMessageEnd()
2953 self._oprot.trans.flush()
2957 (fname, mtype, rseqid) = iprot.readMessageBegin()
2958 if mtype == TMessageType.EXCEPTION:
2959 x = TApplicationException()
2961 iprot.readMessageEnd()
2965 iprot.readMessageEnd()
2966 if result.success
is not None:
2967 return result.success
2968 if result.e
is not None:
2970 raise TApplicationException(TApplicationException.MISSING_RESULT,
"create_dashboard failed: unknown result")
2972 def replace_dashboard(self, session, dashboard_id, dashboard_name, dashboard_owner, dashboard_state, image_hash, dashboard_metadata):
2981 - dashboard_metadata
2984 self.
send_replace_dashboard(session, dashboard_id, dashboard_name, dashboard_owner, dashboard_state, image_hash, dashboard_metadata)
2987 def send_replace_dashboard(self, session, dashboard_id, dashboard_name, dashboard_owner, dashboard_state, image_hash, dashboard_metadata):
2988 self._oprot.writeMessageBegin(
'replace_dashboard', TMessageType.CALL, self.
_seqid)
2990 args.session = session
2991 args.dashboard_id = dashboard_id
2992 args.dashboard_name = dashboard_name
2993 args.dashboard_owner = dashboard_owner
2994 args.dashboard_state = dashboard_state
2995 args.image_hash = image_hash
2996 args.dashboard_metadata = dashboard_metadata
2998 self._oprot.writeMessageEnd()
2999 self._oprot.trans.flush()
3003 (fname, mtype, rseqid) = iprot.readMessageBegin()
3004 if mtype == TMessageType.EXCEPTION:
3005 x = TApplicationException()
3007 iprot.readMessageEnd()
3011 iprot.readMessageEnd()
3012 if result.e
is not None:
3027 self._oprot.writeMessageBegin(
'delete_dashboard', TMessageType.CALL, self.
_seqid)
3029 args.session = session
3030 args.dashboard_id = dashboard_id
3032 self._oprot.writeMessageEnd()
3033 self._oprot.trans.flush()
3037 (fname, mtype, rseqid) = iprot.readMessageBegin()
3038 if mtype == TMessageType.EXCEPTION:
3039 x = TApplicationException()
3041 iprot.readMessageEnd()
3045 iprot.readMessageEnd()
3046 if result.e
is not None:
3063 self._oprot.writeMessageBegin(
'share_dashboards', TMessageType.CALL, self.
_seqid)
3065 args.session = session
3066 args.dashboard_ids = dashboard_ids
3067 args.groups = groups
3068 args.permissions = permissions
3070 self._oprot.writeMessageEnd()
3071 self._oprot.trans.flush()
3075 (fname, mtype, rseqid) = iprot.readMessageBegin()
3076 if mtype == TMessageType.EXCEPTION:
3077 x = TApplicationException()
3079 iprot.readMessageEnd()
3083 iprot.readMessageEnd()
3084 if result.e
is not None:
3099 self._oprot.writeMessageBegin(
'delete_dashboards', TMessageType.CALL, self.
_seqid)
3101 args.session = session
3102 args.dashboard_ids = dashboard_ids
3104 self._oprot.writeMessageEnd()
3105 self._oprot.trans.flush()
3109 (fname, mtype, rseqid) = iprot.readMessageBegin()
3110 if mtype == TMessageType.EXCEPTION:
3111 x = TApplicationException()
3113 iprot.readMessageEnd()
3117 iprot.readMessageEnd()
3118 if result.e
is not None:
3122 def share_dashboard(self, session, dashboard_id, groups, objects, permissions, grant_role):
3137 self._oprot.writeMessageBegin(
'share_dashboard', TMessageType.CALL, self.
_seqid)
3139 args.session = session
3140 args.dashboard_id = dashboard_id
3141 args.groups = groups
3142 args.objects = objects
3143 args.permissions = permissions
3144 args.grant_role = grant_role
3146 self._oprot.writeMessageEnd()
3147 self._oprot.trans.flush()
3151 (fname, mtype, rseqid) = iprot.readMessageBegin()
3152 if mtype == TMessageType.EXCEPTION:
3153 x = TApplicationException()
3155 iprot.readMessageEnd()
3159 iprot.readMessageEnd()
3160 if result.e
is not None:
3178 self._oprot.writeMessageBegin(
'unshare_dashboard', TMessageType.CALL, self.
_seqid)
3180 args.session = session
3181 args.dashboard_id = dashboard_id
3182 args.groups = groups
3183 args.objects = objects
3184 args.permissions = permissions
3186 self._oprot.writeMessageEnd()
3187 self._oprot.trans.flush()
3191 (fname, mtype, rseqid) = iprot.readMessageBegin()
3192 if mtype == TMessageType.EXCEPTION:
3193 x = TApplicationException()
3195 iprot.readMessageEnd()
3199 iprot.readMessageEnd()
3200 if result.e
is not None:
3217 self._oprot.writeMessageBegin(
'unshare_dashboards', TMessageType.CALL, self.
_seqid)
3219 args.session = session
3220 args.dashboard_ids = dashboard_ids
3221 args.groups = groups
3222 args.permissions = permissions
3224 self._oprot.writeMessageEnd()
3225 self._oprot.trans.flush()
3229 (fname, mtype, rseqid) = iprot.readMessageBegin()
3230 if mtype == TMessageType.EXCEPTION:
3231 x = TApplicationException()
3233 iprot.readMessageEnd()
3237 iprot.readMessageEnd()
3238 if result.e
is not None:
3253 self._oprot.writeMessageBegin(
'get_dashboard_grantees', TMessageType.CALL, self.
_seqid)
3255 args.session = session
3256 args.dashboard_id = dashboard_id
3258 self._oprot.writeMessageEnd()
3259 self._oprot.trans.flush()
3263 (fname, mtype, rseqid) = iprot.readMessageBegin()
3264 if mtype == TMessageType.EXCEPTION:
3265 x = TApplicationException()
3267 iprot.readMessageEnd()
3271 iprot.readMessageEnd()
3272 if result.success
is not None:
3273 return result.success
3274 if result.e
is not None:
3276 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_dashboard_grantees failed: unknown result")
3289 self._oprot.writeMessageBegin(
'get_link_view', TMessageType.CALL, self.
_seqid)
3291 args.session = session
3294 self._oprot.writeMessageEnd()
3295 self._oprot.trans.flush()
3299 (fname, mtype, rseqid) = iprot.readMessageBegin()
3300 if mtype == TMessageType.EXCEPTION:
3301 x = TApplicationException()
3303 iprot.readMessageEnd()
3307 iprot.readMessageEnd()
3308 if result.success
is not None:
3309 return result.success
3310 if result.e
is not None:
3312 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_link_view failed: unknown result")
3326 self._oprot.writeMessageBegin(
'create_link', TMessageType.CALL, self.
_seqid)
3328 args.session = session
3329 args.view_state = view_state
3330 args.view_metadata = view_metadata
3332 self._oprot.writeMessageEnd()
3333 self._oprot.trans.flush()
3337 (fname, mtype, rseqid) = iprot.readMessageBegin()
3338 if mtype == TMessageType.EXCEPTION:
3339 x = TApplicationException()
3341 iprot.readMessageEnd()
3345 iprot.readMessageEnd()
3346 if result.success
is not None:
3347 return result.success
3348 if result.e
is not None:
3350 raise TApplicationException(TApplicationException.MISSING_RESULT,
"create_link failed: unknown result")
3365 self._oprot.writeMessageBegin(
'load_table_binary', TMessageType.CALL, self.
_seqid)
3367 args.session = session
3368 args.table_name = table_name
3370 args.column_names = column_names
3372 self._oprot.writeMessageEnd()
3373 self._oprot.trans.flush()
3377 (fname, mtype, rseqid) = iprot.readMessageBegin()
3378 if mtype == TMessageType.EXCEPTION:
3379 x = TApplicationException()
3381 iprot.readMessageEnd()
3385 iprot.readMessageEnd()
3386 if result.e
is not None:
3403 self._oprot.writeMessageBegin(
'load_table_binary_columnar', TMessageType.CALL, self.
_seqid)
3405 args.session = session
3406 args.table_name = table_name
3408 args.column_names = column_names
3410 self._oprot.writeMessageEnd()
3411 self._oprot.trans.flush()
3415 (fname, mtype, rseqid) = iprot.readMessageBegin()
3416 if mtype == TMessageType.EXCEPTION:
3417 x = TApplicationException()
3419 iprot.readMessageEnd()
3423 iprot.readMessageEnd()
3424 if result.e
is not None:
3435 - assign_render_groups
3442 self._oprot.writeMessageBegin(
'load_table_binary_columnar_polys', TMessageType.CALL, self.
_seqid)
3444 args.session = session
3445 args.table_name = table_name
3447 args.column_names = column_names
3448 args.assign_render_groups = assign_render_groups
3450 self._oprot.writeMessageEnd()
3451 self._oprot.trans.flush()
3455 (fname, mtype, rseqid) = iprot.readMessageBegin()
3456 if mtype == TMessageType.EXCEPTION:
3457 x = TApplicationException()
3459 iprot.readMessageEnd()
3463 iprot.readMessageEnd()
3464 if result.e
is not None:
3481 self._oprot.writeMessageBegin(
'load_table_binary_arrow', TMessageType.CALL, self.
_seqid)
3483 args.session = session
3484 args.table_name = table_name
3485 args.arrow_stream = arrow_stream
3486 args.use_column_names = use_column_names
3488 self._oprot.writeMessageEnd()
3489 self._oprot.trans.flush()
3493 (fname, mtype, rseqid) = iprot.readMessageBegin()
3494 if mtype == TMessageType.EXCEPTION:
3495 x = TApplicationException()
3497 iprot.readMessageEnd()
3501 iprot.readMessageEnd()
3502 if result.e
is not None:
3519 self._oprot.writeMessageBegin(
'load_table', TMessageType.CALL, self.
_seqid)
3521 args.session = session
3522 args.table_name = table_name
3524 args.column_names = column_names
3526 self._oprot.writeMessageEnd()
3527 self._oprot.trans.flush()
3531 (fname, mtype, rseqid) = iprot.readMessageBegin()
3532 if mtype == TMessageType.EXCEPTION:
3533 x = TApplicationException()
3535 iprot.readMessageEnd()
3539 iprot.readMessageEnd()
3540 if result.e
is not None:
3556 self._oprot.writeMessageBegin(
'detect_column_types', TMessageType.CALL, self.
_seqid)
3558 args.session = session
3559 args.file_name = file_name
3560 args.copy_params = copy_params
3562 self._oprot.writeMessageEnd()
3563 self._oprot.trans.flush()
3567 (fname, mtype, rseqid) = iprot.readMessageBegin()
3568 if mtype == TMessageType.EXCEPTION:
3569 x = TApplicationException()
3571 iprot.readMessageEnd()
3575 iprot.readMessageEnd()
3576 if result.success
is not None:
3577 return result.success
3578 if result.e
is not None:
3580 raise TApplicationException(TApplicationException.MISSING_RESULT,
"detect_column_types failed: unknown result")
3595 self._oprot.writeMessageBegin(
'create_table', TMessageType.CALL, self.
_seqid)
3597 args.session = session
3598 args.table_name = table_name
3599 args.row_desc = row_desc
3600 args.create_params = create_params
3602 self._oprot.writeMessageEnd()
3603 self._oprot.trans.flush()
3607 (fname, mtype, rseqid) = iprot.readMessageBegin()
3608 if mtype == TMessageType.EXCEPTION:
3609 x = TApplicationException()
3611 iprot.readMessageEnd()
3615 iprot.readMessageEnd()
3616 if result.e
is not None:
3633 self._oprot.writeMessageBegin(
'import_table', TMessageType.CALL, self.
_seqid)
3635 args.session = session
3636 args.table_name = table_name
3637 args.file_name = file_name
3638 args.copy_params = copy_params
3640 self._oprot.writeMessageEnd()
3641 self._oprot.trans.flush()
3645 (fname, mtype, rseqid) = iprot.readMessageBegin()
3646 if mtype == TMessageType.EXCEPTION:
3647 x = TApplicationException()
3649 iprot.readMessageEnd()
3653 iprot.readMessageEnd()
3654 if result.e
is not None:
3658 def import_geo_table(self, session, table_name, file_name, copy_params, row_desc, create_params):
3673 self._oprot.writeMessageBegin(
'import_geo_table', TMessageType.CALL, self.
_seqid)
3675 args.session = session
3676 args.table_name = table_name
3677 args.file_name = file_name
3678 args.copy_params = copy_params
3679 args.row_desc = row_desc
3680 args.create_params = create_params
3682 self._oprot.writeMessageEnd()
3683 self._oprot.trans.flush()
3687 (fname, mtype, rseqid) = iprot.readMessageBegin()
3688 if mtype == TMessageType.EXCEPTION:
3689 x = TApplicationException()
3691 iprot.readMessageEnd()
3695 iprot.readMessageEnd()
3696 if result.e
is not None:
3711 self._oprot.writeMessageBegin(
'import_table_status', TMessageType.CALL, self.
_seqid)
3713 args.session = session
3714 args.import_id = import_id
3716 self._oprot.writeMessageEnd()
3717 self._oprot.trans.flush()
3721 (fname, mtype, rseqid) = iprot.readMessageBegin()
3722 if mtype == TMessageType.EXCEPTION:
3723 x = TApplicationException()
3725 iprot.readMessageEnd()
3729 iprot.readMessageEnd()
3730 if result.success
is not None:
3731 return result.success
3732 if result.e
is not None:
3734 raise TApplicationException(TApplicationException.MISSING_RESULT,
"import_table_status failed: unknown result")
3748 self._oprot.writeMessageBegin(
'get_first_geo_file_in_archive', TMessageType.CALL, self.
_seqid)
3750 args.session = session
3751 args.archive_path = archive_path
3752 args.copy_params = copy_params
3754 self._oprot.writeMessageEnd()
3755 self._oprot.trans.flush()
3759 (fname, mtype, rseqid) = iprot.readMessageBegin()
3760 if mtype == TMessageType.EXCEPTION:
3761 x = TApplicationException()
3763 iprot.readMessageEnd()
3767 iprot.readMessageEnd()
3768 if result.success
is not None:
3769 return result.success
3770 if result.e
is not None:
3772 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_first_geo_file_in_archive failed: unknown result")
3786 self._oprot.writeMessageBegin(
'get_all_files_in_archive', TMessageType.CALL, self.
_seqid)
3788 args.session = session
3789 args.archive_path = archive_path
3790 args.copy_params = copy_params
3792 self._oprot.writeMessageEnd()
3793 self._oprot.trans.flush()
3797 (fname, mtype, rseqid) = iprot.readMessageBegin()
3798 if mtype == TMessageType.EXCEPTION:
3799 x = TApplicationException()
3801 iprot.readMessageEnd()
3805 iprot.readMessageEnd()
3806 if result.success
is not None:
3807 return result.success
3808 if result.e
is not None:
3810 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_all_files_in_archive failed: unknown result")
3824 self._oprot.writeMessageBegin(
'get_layers_in_geo_file', TMessageType.CALL, self.
_seqid)
3826 args.session = session
3827 args.file_name = file_name
3828 args.copy_params = copy_params
3830 self._oprot.writeMessageEnd()
3831 self._oprot.trans.flush()
3835 (fname, mtype, rseqid) = iprot.readMessageBegin()
3836 if mtype == TMessageType.EXCEPTION:
3837 x = TApplicationException()
3839 iprot.readMessageEnd()
3843 iprot.readMessageEnd()
3844 if result.success
is not None:
3845 return result.success
3846 if result.e
is not None:
3848 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_layers_in_geo_file failed: unknown result")
3861 self._oprot.writeMessageBegin(
'query_get_outer_fragment_count', TMessageType.CALL, self.
_seqid)
3863 args.session = session
3866 self._oprot.writeMessageEnd()
3867 self._oprot.trans.flush()
3871 (fname, mtype, rseqid) = iprot.readMessageBegin()
3872 if mtype == TMessageType.EXCEPTION:
3873 x = TApplicationException()
3875 iprot.readMessageEnd()
3879 iprot.readMessageEnd()
3880 if result.success
is not None:
3881 return result.success
3882 if result.e
is not None:
3884 raise TApplicationException(TApplicationException.MISSING_RESULT,
"query_get_outer_fragment_count failed: unknown result")
3897 self._oprot.writeMessageBegin(
'check_table_consistency', TMessageType.CALL, self.
_seqid)
3899 args.session = session
3900 args.table_id = table_id
3902 self._oprot.writeMessageEnd()
3903 self._oprot.trans.flush()
3907 (fname, mtype, rseqid) = iprot.readMessageBegin()
3908 if mtype == TMessageType.EXCEPTION:
3909 x = TApplicationException()
3911 iprot.readMessageEnd()
3915 iprot.readMessageEnd()
3916 if result.success
is not None:
3917 return result.success
3918 if result.e
is not None:
3920 raise TApplicationException(TApplicationException.MISSING_RESULT,
"check_table_consistency failed: unknown result")
3922 def start_query(self, leaf_session, parent_session, query_ra, start_time_str, just_explain, outer_fragment_indices):
3930 - outer_fragment_indices
3933 self.
send_start_query(leaf_session, parent_session, query_ra, start_time_str, just_explain, outer_fragment_indices)
3936 def send_start_query(self, leaf_session, parent_session, query_ra, start_time_str, just_explain, outer_fragment_indices):
3937 self._oprot.writeMessageBegin(
'start_query', TMessageType.CALL, self.
_seqid)
3939 args.leaf_session = leaf_session
3940 args.parent_session = parent_session
3941 args.query_ra = query_ra
3942 args.start_time_str = start_time_str
3943 args.just_explain = just_explain
3944 args.outer_fragment_indices = outer_fragment_indices
3946 self._oprot.writeMessageEnd()
3947 self._oprot.trans.flush()
3951 (fname, mtype, rseqid) = iprot.readMessageBegin()
3952 if mtype == TMessageType.EXCEPTION:
3953 x = TApplicationException()
3955 iprot.readMessageEnd()
3959 iprot.readMessageEnd()
3960 if result.success
is not None:
3961 return result.success
3962 if result.e
is not None:
3964 raise TApplicationException(TApplicationException.MISSING_RESULT,
"start_query failed: unknown result")
3978 self._oprot.writeMessageBegin(
'execute_query_step', TMessageType.CALL, self.
_seqid)
3980 args.pending_query = pending_query
3981 args.subquery_id = subquery_id
3982 args.start_time_str = start_time_str
3984 self._oprot.writeMessageEnd()
3985 self._oprot.trans.flush()
3989 (fname, mtype, rseqid) = iprot.readMessageBegin()
3990 if mtype == TMessageType.EXCEPTION:
3991 x = TApplicationException()
3993 iprot.readMessageEnd()
3997 iprot.readMessageEnd()
3998 if result.success
is not None:
3999 return result.success
4000 if result.e
is not None:
4002 raise TApplicationException(TApplicationException.MISSING_RESULT,
"execute_query_step failed: unknown result")
4011 - is_final_subquery_result
4018 self._oprot.writeMessageBegin(
'broadcast_serialized_rows', TMessageType.CALL, self.
_seqid)
4020 args.serialized_rows = serialized_rows
4021 args.row_desc = row_desc
4022 args.query_id = query_id
4023 args.subquery_id = subquery_id
4024 args.is_final_subquery_result = is_final_subquery_result
4026 self._oprot.writeMessageEnd()
4027 self._oprot.trans.flush()
4031 (fname, mtype, rseqid) = iprot.readMessageBegin()
4032 if mtype == TMessageType.EXCEPTION:
4033 x = TApplicationException()
4035 iprot.readMessageEnd()
4039 iprot.readMessageEnd()
4040 if result.e
is not None:
4057 self._oprot.writeMessageBegin(
'start_render_query', TMessageType.CALL, self.
_seqid)
4059 args.session = session
4060 args.widget_id = widget_id
4061 args.node_idx = node_idx
4062 args.vega_json = vega_json
4064 self._oprot.writeMessageEnd()
4065 self._oprot.trans.flush()
4069 (fname, mtype, rseqid) = iprot.readMessageBegin()
4070 if mtype == TMessageType.EXCEPTION:
4071 x = TApplicationException()
4073 iprot.readMessageEnd()
4077 iprot.readMessageEnd()
4078 if result.success
is not None:
4079 return result.success
4080 if result.e
is not None:
4082 raise TApplicationException(TApplicationException.MISSING_RESULT,
"start_render_query failed: unknown result")
4095 self._oprot.writeMessageBegin(
'execute_next_render_step', TMessageType.CALL, self.
_seqid)
4097 args.pending_render = pending_render
4098 args.merged_data = merged_data
4100 self._oprot.writeMessageEnd()
4101 self._oprot.trans.flush()
4105 (fname, mtype, rseqid) = iprot.readMessageBegin()
4106 if mtype == TMessageType.EXCEPTION:
4107 x = TApplicationException()
4109 iprot.readMessageEnd()
4113 iprot.readMessageEnd()
4114 if result.success
is not None:
4115 return result.success
4116 if result.e
is not None:
4118 raise TApplicationException(TApplicationException.MISSING_RESULT,
"execute_next_render_step failed: unknown result")
4131 self._oprot.writeMessageBegin(
'insert_data', TMessageType.CALL, self.
_seqid)
4133 args.session = session
4134 args.insert_data = insert_data
4136 self._oprot.writeMessageEnd()
4137 self._oprot.trans.flush()
4141 (fname, mtype, rseqid) = iprot.readMessageBegin()
4142 if mtype == TMessageType.EXCEPTION:
4143 x = TApplicationException()
4145 iprot.readMessageEnd()
4149 iprot.readMessageEnd()
4150 if result.e
is not None:
4165 self._oprot.writeMessageBegin(
'insert_chunks', TMessageType.CALL, self.
_seqid)
4167 args.session = session
4168 args.insert_chunks = insert_chunks
4170 self._oprot.writeMessageEnd()
4171 self._oprot.trans.flush()
4175 (fname, mtype, rseqid) = iprot.readMessageBegin()
4176 if mtype == TMessageType.EXCEPTION:
4177 x = TApplicationException()
4179 iprot.readMessageEnd()
4183 iprot.readMessageEnd()
4184 if result.e
is not None:
4199 self._oprot.writeMessageBegin(
'checkpoint', TMessageType.CALL, self.
_seqid)
4201 args.session = session
4202 args.table_id = table_id
4204 self._oprot.writeMessageEnd()
4205 self._oprot.trans.flush()
4209 (fname, mtype, rseqid) = iprot.readMessageBegin()
4210 if mtype == TMessageType.EXCEPTION:
4211 x = TApplicationException()
4213 iprot.readMessageEnd()
4217 iprot.readMessageEnd()
4218 if result.e
is not None:
4232 self._oprot.writeMessageBegin(
'get_roles', TMessageType.CALL, self.
_seqid)
4234 args.session = session
4236 self._oprot.writeMessageEnd()
4237 self._oprot.trans.flush()
4241 (fname, mtype, rseqid) = iprot.readMessageBegin()
4242 if mtype == TMessageType.EXCEPTION:
4243 x = TApplicationException()
4245 iprot.readMessageEnd()
4249 iprot.readMessageEnd()
4250 if result.success
is not None:
4251 return result.success
4252 if result.e
is not None:
4254 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_roles failed: unknown result")
4267 self._oprot.writeMessageBegin(
'get_db_objects_for_grantee', TMessageType.CALL, self.
_seqid)
4269 args.session = session
4270 args.roleName = roleName
4272 self._oprot.writeMessageEnd()
4273 self._oprot.trans.flush()
4277 (fname, mtype, rseqid) = iprot.readMessageBegin()
4278 if mtype == TMessageType.EXCEPTION:
4279 x = TApplicationException()
4281 iprot.readMessageEnd()
4285 iprot.readMessageEnd()
4286 if result.success
is not None:
4287 return result.success
4288 if result.e
is not None:
4290 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_db_objects_for_grantee failed: unknown result")
4304 self._oprot.writeMessageBegin(
'get_db_object_privs', TMessageType.CALL, self.
_seqid)
4306 args.session = session
4307 args.objectName = objectName
4310 self._oprot.writeMessageEnd()
4311 self._oprot.trans.flush()
4315 (fname, mtype, rseqid) = iprot.readMessageBegin()
4316 if mtype == TMessageType.EXCEPTION:
4317 x = TApplicationException()
4319 iprot.readMessageEnd()
4323 iprot.readMessageEnd()
4324 if result.success
is not None:
4325 return result.success
4326 if result.e
is not None:
4328 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_db_object_privs failed: unknown result")
4341 self._oprot.writeMessageBegin(
'get_all_roles_for_user', TMessageType.CALL, self.
_seqid)
4343 args.session = session
4344 args.userName = userName
4346 self._oprot.writeMessageEnd()
4347 self._oprot.trans.flush()
4351 (fname, mtype, rseqid) = iprot.readMessageBegin()
4352 if mtype == TMessageType.EXCEPTION:
4353 x = TApplicationException()
4355 iprot.readMessageEnd()
4359 iprot.readMessageEnd()
4360 if result.success
is not None:
4361 return result.success
4362 if result.e
is not None:
4364 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_all_roles_for_user failed: unknown result")
4377 self._oprot.writeMessageBegin(
'get_all_effective_roles_for_user', TMessageType.CALL, self.
_seqid)
4379 args.session = session
4380 args.userName = userName
4382 self._oprot.writeMessageEnd()
4383 self._oprot.trans.flush()
4387 (fname, mtype, rseqid) = iprot.readMessageBegin()
4388 if mtype == TMessageType.EXCEPTION:
4389 x = TApplicationException()
4391 iprot.readMessageEnd()
4395 iprot.readMessageEnd()
4396 if result.success
is not None:
4397 return result.success
4398 if result.e
is not None:
4400 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_all_effective_roles_for_user failed: unknown result")
4414 self._oprot.writeMessageBegin(
'has_role', TMessageType.CALL, self.
_seqid)
4416 args.session = session
4417 args.granteeName = granteeName
4418 args.roleName = roleName
4420 self._oprot.writeMessageEnd()
4421 self._oprot.trans.flush()
4425 (fname, mtype, rseqid) = iprot.readMessageBegin()
4426 if mtype == TMessageType.EXCEPTION:
4427 x = TApplicationException()
4429 iprot.readMessageEnd()
4433 iprot.readMessageEnd()
4434 if result.success
is not None:
4435 return result.success
4436 if result.e
is not None:
4438 raise TApplicationException(TApplicationException.MISSING_RESULT,
"has_role failed: unknown result")
4454 self._oprot.writeMessageBegin(
'has_object_privilege', TMessageType.CALL, self.
_seqid)
4456 args.session = session
4457 args.granteeName = granteeName
4458 args.ObjectName = ObjectName
4459 args.objectType = objectType
4460 args.permissions = permissions
4462 self._oprot.writeMessageEnd()
4463 self._oprot.trans.flush()
4467 (fname, mtype, rseqid) = iprot.readMessageBegin()
4468 if mtype == TMessageType.EXCEPTION:
4469 x = TApplicationException()
4471 iprot.readMessageEnd()
4475 iprot.readMessageEnd()
4476 if result.success
is not None:
4477 return result.success
4478 if result.e
is not None:
4480 raise TApplicationException(TApplicationException.MISSING_RESULT,
"has_object_privilege failed: unknown result")
4494 self._oprot.writeMessageBegin(
'set_license_key', TMessageType.CALL, self.
_seqid)
4496 args.session = session
4500 self._oprot.writeMessageEnd()
4501 self._oprot.trans.flush()
4505 (fname, mtype, rseqid) = iprot.readMessageBegin()
4506 if mtype == TMessageType.EXCEPTION:
4507 x = TApplicationException()
4509 iprot.readMessageEnd()
4513 iprot.readMessageEnd()
4514 if result.success
is not None:
4515 return result.success
4516 if result.e
is not None:
4518 raise TApplicationException(TApplicationException.MISSING_RESULT,
"set_license_key failed: unknown result")
4531 self._oprot.writeMessageBegin(
'get_license_claims', TMessageType.CALL, self.
_seqid)
4533 args.session = session
4536 self._oprot.writeMessageEnd()
4537 self._oprot.trans.flush()
4541 (fname, mtype, rseqid) = iprot.readMessageBegin()
4542 if mtype == TMessageType.EXCEPTION:
4543 x = TApplicationException()
4545 iprot.readMessageEnd()
4549 iprot.readMessageEnd()
4550 if result.success
is not None:
4551 return result.success
4552 if result.e
is not None:
4554 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_license_claims failed: unknown result")
4566 self._oprot.writeMessageBegin(
'get_device_parameters', TMessageType.CALL, self.
_seqid)
4568 args.session = session
4570 self._oprot.writeMessageEnd()
4571 self._oprot.trans.flush()
4575 (fname, mtype, rseqid) = iprot.readMessageBegin()
4576 if mtype == TMessageType.EXCEPTION:
4577 x = TApplicationException()
4579 iprot.readMessageEnd()
4583 iprot.readMessageEnd()
4584 if result.success
is not None:
4585 return result.success
4586 if result.e
is not None:
4588 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_device_parameters failed: unknown result")
4603 self._oprot.writeMessageBegin(
'register_runtime_extension_functions', TMessageType.CALL, self.
_seqid)
4605 args.session = session
4608 args.device_ir_map = device_ir_map
4610 self._oprot.writeMessageEnd()
4611 self._oprot.trans.flush()
4615 (fname, mtype, rseqid) = iprot.readMessageBegin()
4616 if mtype == TMessageType.EXCEPTION:
4617 x = TApplicationException()
4619 iprot.readMessageEnd()
4623 iprot.readMessageEnd()
4624 if result.e
is not None:
4638 self._oprot.writeMessageBegin(
'get_table_function_names', TMessageType.CALL, self.
_seqid)
4640 args.session = session
4642 self._oprot.writeMessageEnd()
4643 self._oprot.trans.flush()
4647 (fname, mtype, rseqid) = iprot.readMessageBegin()
4648 if mtype == TMessageType.EXCEPTION:
4649 x = TApplicationException()
4651 iprot.readMessageEnd()
4655 iprot.readMessageEnd()
4656 if result.success
is not None:
4657 return result.success
4658 if result.e
is not None:
4660 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_table_function_names failed: unknown result")
4672 self._oprot.writeMessageBegin(
'get_runtime_table_function_names', TMessageType.CALL, self.
_seqid)
4674 args.session = session
4676 self._oprot.writeMessageEnd()
4677 self._oprot.trans.flush()
4681 (fname, mtype, rseqid) = iprot.readMessageBegin()
4682 if mtype == TMessageType.EXCEPTION:
4683 x = TApplicationException()
4685 iprot.readMessageEnd()
4689 iprot.readMessageEnd()
4690 if result.success
is not None:
4691 return result.success
4692 if result.e
is not None:
4694 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_runtime_table_function_names failed: unknown result")
4707 self._oprot.writeMessageBegin(
'get_table_function_details', TMessageType.CALL, self.
_seqid)
4709 args.session = session
4710 args.udtf_names = udtf_names
4712 self._oprot.writeMessageEnd()
4713 self._oprot.trans.flush()
4717 (fname, mtype, rseqid) = iprot.readMessageBegin()
4718 if mtype == TMessageType.EXCEPTION:
4719 x = TApplicationException()
4721 iprot.readMessageEnd()
4725 iprot.readMessageEnd()
4726 if result.success
is not None:
4727 return result.success
4728 if result.e
is not None:
4730 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_table_function_details failed: unknown result")
4742 self._oprot.writeMessageBegin(
'get_function_names', TMessageType.CALL, self.
_seqid)
4744 args.session = session
4746 self._oprot.writeMessageEnd()
4747 self._oprot.trans.flush()
4751 (fname, mtype, rseqid) = iprot.readMessageBegin()
4752 if mtype == TMessageType.EXCEPTION:
4753 x = TApplicationException()
4755 iprot.readMessageEnd()
4759 iprot.readMessageEnd()
4760 if result.success
is not None:
4761 return result.success
4762 if result.e
is not None:
4764 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_function_names failed: unknown result")
4776 self._oprot.writeMessageBegin(
'get_runtime_function_names', TMessageType.CALL, self.
_seqid)
4778 args.session = session
4780 self._oprot.writeMessageEnd()
4781 self._oprot.trans.flush()
4785 (fname, mtype, rseqid) = iprot.readMessageBegin()
4786 if mtype == TMessageType.EXCEPTION:
4787 x = TApplicationException()
4789 iprot.readMessageEnd()
4793 iprot.readMessageEnd()
4794 if result.success
is not None:
4795 return result.success
4796 if result.e
is not None:
4798 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_runtime_function_names failed: unknown result")
4811 self._oprot.writeMessageBegin(
'get_function_details', TMessageType.CALL, self.
_seqid)
4813 args.session = session
4814 args.udf_names = udf_names
4816 self._oprot.writeMessageEnd()
4817 self._oprot.trans.flush()
4821 (fname, mtype, rseqid) = iprot.readMessageBegin()
4822 if mtype == TMessageType.EXCEPTION:
4823 x = TApplicationException()
4825 iprot.readMessageEnd()
4829 iprot.readMessageEnd()
4830 if result.success
is not None:
4831 return result.success
4832 if result.e
is not None:
4834 raise TApplicationException(TApplicationException.MISSING_RESULT,
"get_function_details failed: unknown result")
4841 self.
_processMap[
"connect"] = Processor.process_connect
4842 self.
_processMap[
"krb5_connect"] = Processor.process_krb5_connect
4843 self.
_processMap[
"disconnect"] = Processor.process_disconnect
4844 self.
_processMap[
"switch_database"] = Processor.process_switch_database
4845 self.
_processMap[
"clone_session"] = Processor.process_clone_session
4846 self.
_processMap[
"get_server_status"] = Processor.process_get_server_status
4847 self.
_processMap[
"get_status"] = Processor.process_get_status
4848 self.
_processMap[
"get_hardware_info"] = Processor.process_get_hardware_info
4849 self.
_processMap[
"get_tables"] = Processor.process_get_tables
4850 self.
_processMap[
"get_tables_for_database"] = Processor.process_get_tables_for_database
4851 self.
_processMap[
"get_physical_tables"] = Processor.process_get_physical_tables
4852 self.
_processMap[
"get_views"] = Processor.process_get_views
4853 self.
_processMap[
"get_tables_meta"] = Processor.process_get_tables_meta
4854 self.
_processMap[
"get_table_details"] = Processor.process_get_table_details
4855 self.
_processMap[
"get_table_details_for_database"] = Processor.process_get_table_details_for_database
4856 self.
_processMap[
"get_internal_table_details"] = Processor.process_get_internal_table_details
4857 self.
_processMap[
"get_internal_table_details_for_database"] = Processor.process_get_internal_table_details_for_database
4858 self.
_processMap[
"get_users"] = Processor.process_get_users
4859 self.
_processMap[
"get_databases"] = Processor.process_get_databases
4860 self.
_processMap[
"get_version"] = Processor.process_get_version
4861 self.
_processMap[
"start_heap_profile"] = Processor.process_start_heap_profile
4862 self.
_processMap[
"stop_heap_profile"] = Processor.process_stop_heap_profile
4863 self.
_processMap[
"get_heap_profile"] = Processor.process_get_heap_profile
4864 self.
_processMap[
"get_memory"] = Processor.process_get_memory
4865 self.
_processMap[
"clear_cpu_memory"] = Processor.process_clear_cpu_memory
4866 self.
_processMap[
"clear_gpu_memory"] = Processor.process_clear_gpu_memory
4867 self.
_processMap[
"set_cur_session"] = Processor.process_set_cur_session
4868 self.
_processMap[
"invalidate_cur_session"] = Processor.process_invalidate_cur_session
4869 self.
_processMap[
"set_table_epoch"] = Processor.process_set_table_epoch
4870 self.
_processMap[
"set_table_epoch_by_name"] = Processor.process_set_table_epoch_by_name
4871 self.
_processMap[
"get_table_epoch"] = Processor.process_get_table_epoch
4872 self.
_processMap[
"get_table_epoch_by_name"] = Processor.process_get_table_epoch_by_name
4873 self.
_processMap[
"get_table_epochs"] = Processor.process_get_table_epochs
4874 self.
_processMap[
"set_table_epochs"] = Processor.process_set_table_epochs
4875 self.
_processMap[
"get_session_info"] = Processor.process_get_session_info
4876 self.
_processMap[
"get_queries_info"] = Processor.process_get_queries_info
4877 self.
_processMap[
"set_leaf_info"] = Processor.process_set_leaf_info
4878 self.
_processMap[
"sql_execute"] = Processor.process_sql_execute
4879 self.
_processMap[
"sql_execute_df"] = Processor.process_sql_execute_df
4880 self.
_processMap[
"sql_execute_gdf"] = Processor.process_sql_execute_gdf
4881 self.
_processMap[
"deallocate_df"] = Processor.process_deallocate_df
4882 self.
_processMap[
"interrupt"] = Processor.process_interrupt
4883 self.
_processMap[
"sql_validate"] = Processor.process_sql_validate
4884 self.
_processMap[
"get_completion_hints"] = Processor.process_get_completion_hints
4885 self.
_processMap[
"set_execution_mode"] = Processor.process_set_execution_mode
4886 self.
_processMap[
"render_vega"] = Processor.process_render_vega
4887 self.
_processMap[
"get_result_row_for_pixel"] = Processor.process_get_result_row_for_pixel
4888 self.
_processMap[
"create_custom_expression"] = Processor.process_create_custom_expression
4889 self.
_processMap[
"get_custom_expressions"] = Processor.process_get_custom_expressions
4890 self.
_processMap[
"update_custom_expression"] = Processor.process_update_custom_expression
4891 self.
_processMap[
"delete_custom_expressions"] = Processor.process_delete_custom_expressions
4892 self.
_processMap[
"get_dashboard"] = Processor.process_get_dashboard
4893 self.
_processMap[
"get_dashboards"] = Processor.process_get_dashboards
4894 self.
_processMap[
"create_dashboard"] = Processor.process_create_dashboard
4895 self.
_processMap[
"replace_dashboard"] = Processor.process_replace_dashboard
4896 self.
_processMap[
"delete_dashboard"] = Processor.process_delete_dashboard
4897 self.
_processMap[
"share_dashboards"] = Processor.process_share_dashboards
4898 self.
_processMap[
"delete_dashboards"] = Processor.process_delete_dashboards
4899 self.
_processMap[
"share_dashboard"] = Processor.process_share_dashboard
4900 self.
_processMap[
"unshare_dashboard"] = Processor.process_unshare_dashboard
4901 self.
_processMap[
"unshare_dashboards"] = Processor.process_unshare_dashboards
4902 self.
_processMap[
"get_dashboard_grantees"] = Processor.process_get_dashboard_grantees
4903 self.
_processMap[
"get_link_view"] = Processor.process_get_link_view
4904 self.
_processMap[
"create_link"] = Processor.process_create_link
4905 self.
_processMap[
"load_table_binary"] = Processor.process_load_table_binary
4906 self.
_processMap[
"load_table_binary_columnar"] = Processor.process_load_table_binary_columnar
4907 self.
_processMap[
"load_table_binary_columnar_polys"] = Processor.process_load_table_binary_columnar_polys
4908 self.
_processMap[
"load_table_binary_arrow"] = Processor.process_load_table_binary_arrow
4909 self.
_processMap[
"load_table"] = Processor.process_load_table
4910 self.
_processMap[
"detect_column_types"] = Processor.process_detect_column_types
4911 self.
_processMap[
"create_table"] = Processor.process_create_table
4912 self.
_processMap[
"import_table"] = Processor.process_import_table
4913 self.
_processMap[
"import_geo_table"] = Processor.process_import_geo_table
4914 self.
_processMap[
"import_table_status"] = Processor.process_import_table_status
4915 self.
_processMap[
"get_first_geo_file_in_archive"] = Processor.process_get_first_geo_file_in_archive
4916 self.
_processMap[
"get_all_files_in_archive"] = Processor.process_get_all_files_in_archive
4917 self.
_processMap[
"get_layers_in_geo_file"] = Processor.process_get_layers_in_geo_file
4918 self.
_processMap[
"query_get_outer_fragment_count"] = Processor.process_query_get_outer_fragment_count
4919 self.
_processMap[
"check_table_consistency"] = Processor.process_check_table_consistency
4920 self.
_processMap[
"start_query"] = Processor.process_start_query
4921 self.
_processMap[
"execute_query_step"] = Processor.process_execute_query_step
4922 self.
_processMap[
"broadcast_serialized_rows"] = Processor.process_broadcast_serialized_rows
4923 self.
_processMap[
"start_render_query"] = Processor.process_start_render_query
4924 self.
_processMap[
"execute_next_render_step"] = Processor.process_execute_next_render_step
4925 self.
_processMap[
"insert_data"] = Processor.process_insert_data
4926 self.
_processMap[
"insert_chunks"] = Processor.process_insert_chunks
4927 self.
_processMap[
"checkpoint"] = Processor.process_checkpoint
4928 self.
_processMap[
"get_roles"] = Processor.process_get_roles
4929 self.
_processMap[
"get_db_objects_for_grantee"] = Processor.process_get_db_objects_for_grantee
4930 self.
_processMap[
"get_db_object_privs"] = Processor.process_get_db_object_privs
4931 self.
_processMap[
"get_all_roles_for_user"] = Processor.process_get_all_roles_for_user
4932 self.
_processMap[
"get_all_effective_roles_for_user"] = Processor.process_get_all_effective_roles_for_user
4933 self.
_processMap[
"has_role"] = Processor.process_has_role
4934 self.
_processMap[
"has_object_privilege"] = Processor.process_has_object_privilege
4935 self.
_processMap[
"set_license_key"] = Processor.process_set_license_key
4936 self.
_processMap[
"get_license_claims"] = Processor.process_get_license_claims
4937 self.
_processMap[
"get_device_parameters"] = Processor.process_get_device_parameters
4938 self.
_processMap[
"register_runtime_extension_functions"] = Processor.process_register_runtime_extension_functions
4939 self.
_processMap[
"get_table_function_names"] = Processor.process_get_table_function_names
4940 self.
_processMap[
"get_runtime_table_function_names"] = Processor.process_get_runtime_table_function_names
4941 self.
_processMap[
"get_table_function_details"] = Processor.process_get_table_function_details
4942 self.
_processMap[
"get_function_names"] = Processor.process_get_function_names
4943 self.
_processMap[
"get_runtime_function_names"] = Processor.process_get_runtime_function_names
4944 self.
_processMap[
"get_function_details"] = Processor.process_get_function_details
4951 (name, type, seqid) = iprot.readMessageBegin()
4955 iprot.skip(TType.STRUCT)
4956 iprot.readMessageEnd()
4957 x = TApplicationException(TApplicationException.UNKNOWN_METHOD,
'Unknown function %s' % (name))
4958 oprot.writeMessageBegin(name, TMessageType.EXCEPTION, seqid)
4960 oprot.writeMessageEnd()
4970 iprot.readMessageEnd()
4973 result.success = self._handler.connect(args.user, args.passwd, args.dbname)
4974 msg_type = TMessageType.REPLY
4975 except TTransport.TTransportException:
4977 except TDBException
as e:
4978 msg_type = TMessageType.REPLY
4980 except TApplicationException
as ex:
4981 logging.exception(
'TApplication exception in handler')
4982 msg_type = TMessageType.EXCEPTION
4985 logging.exception(
'Unexpected exception in handler')
4986 msg_type = TMessageType.EXCEPTION
4987 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
4988 oprot.writeMessageBegin(
"connect", msg_type, seqid)
4990 oprot.writeMessageEnd()
4996 iprot.readMessageEnd()
4999 result.success = self._handler.krb5_connect(args.inputToken, args.dbname)
5000 msg_type = TMessageType.REPLY
5001 except TTransport.TTransportException:
5003 except TDBException
as e:
5004 msg_type = TMessageType.REPLY
5006 except TApplicationException
as ex:
5007 logging.exception(
'TApplication exception in handler')
5008 msg_type = TMessageType.EXCEPTION
5011 logging.exception(
'Unexpected exception in handler')
5012 msg_type = TMessageType.EXCEPTION
5013 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5014 oprot.writeMessageBegin(
"krb5_connect", msg_type, seqid)
5016 oprot.writeMessageEnd()
5022 iprot.readMessageEnd()
5025 self._handler.disconnect(args.session)
5026 msg_type = TMessageType.REPLY
5027 except TTransport.TTransportException:
5029 except TDBException
as e:
5030 msg_type = TMessageType.REPLY
5032 except TApplicationException
as ex:
5033 logging.exception(
'TApplication exception in handler')
5034 msg_type = TMessageType.EXCEPTION
5037 logging.exception(
'Unexpected exception in handler')
5038 msg_type = TMessageType.EXCEPTION
5039 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5040 oprot.writeMessageBegin(
"disconnect", msg_type, seqid)
5042 oprot.writeMessageEnd()
5048 iprot.readMessageEnd()
5051 self._handler.switch_database(args.session, args.dbname)
5052 msg_type = TMessageType.REPLY
5053 except TTransport.TTransportException:
5055 except TDBException
as e:
5056 msg_type = TMessageType.REPLY
5058 except TApplicationException
as ex:
5059 logging.exception(
'TApplication exception in handler')
5060 msg_type = TMessageType.EXCEPTION
5063 logging.exception(
'Unexpected exception in handler')
5064 msg_type = TMessageType.EXCEPTION
5065 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5066 oprot.writeMessageBegin(
"switch_database", msg_type, seqid)
5068 oprot.writeMessageEnd()
5074 iprot.readMessageEnd()
5077 result.success = self._handler.clone_session(args.session)
5078 msg_type = TMessageType.REPLY
5079 except TTransport.TTransportException:
5081 except TDBException
as e:
5082 msg_type = TMessageType.REPLY
5084 except TApplicationException
as ex:
5085 logging.exception(
'TApplication exception in handler')
5086 msg_type = TMessageType.EXCEPTION
5089 logging.exception(
'Unexpected exception in handler')
5090 msg_type = TMessageType.EXCEPTION
5091 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5092 oprot.writeMessageBegin(
"clone_session", msg_type, seqid)
5094 oprot.writeMessageEnd()
5100 iprot.readMessageEnd()
5103 result.success = self._handler.get_server_status(args.session)
5104 msg_type = TMessageType.REPLY
5105 except TTransport.TTransportException:
5107 except TDBException
as e:
5108 msg_type = TMessageType.REPLY
5110 except TApplicationException
as ex:
5111 logging.exception(
'TApplication exception in handler')
5112 msg_type = TMessageType.EXCEPTION
5115 logging.exception(
'Unexpected exception in handler')
5116 msg_type = TMessageType.EXCEPTION
5117 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5118 oprot.writeMessageBegin(
"get_server_status", msg_type, seqid)
5120 oprot.writeMessageEnd()
5126 iprot.readMessageEnd()
5129 result.success = self._handler.get_status(args.session)
5130 msg_type = TMessageType.REPLY
5131 except TTransport.TTransportException:
5133 except TDBException
as e:
5134 msg_type = TMessageType.REPLY
5136 except TApplicationException
as ex:
5137 logging.exception(
'TApplication exception in handler')
5138 msg_type = TMessageType.EXCEPTION
5141 logging.exception(
'Unexpected exception in handler')
5142 msg_type = TMessageType.EXCEPTION
5143 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5144 oprot.writeMessageBegin(
"get_status", msg_type, seqid)
5146 oprot.writeMessageEnd()
5152 iprot.readMessageEnd()
5155 result.success = self._handler.get_hardware_info(args.session)
5156 msg_type = TMessageType.REPLY
5157 except TTransport.TTransportException:
5159 except TDBException
as e:
5160 msg_type = TMessageType.REPLY
5162 except TApplicationException
as ex:
5163 logging.exception(
'TApplication exception in handler')
5164 msg_type = TMessageType.EXCEPTION
5167 logging.exception(
'Unexpected exception in handler')
5168 msg_type = TMessageType.EXCEPTION
5169 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5170 oprot.writeMessageBegin(
"get_hardware_info", msg_type, seqid)
5172 oprot.writeMessageEnd()
5178 iprot.readMessageEnd()
5181 result.success = self._handler.get_tables(args.session)
5182 msg_type = TMessageType.REPLY
5183 except TTransport.TTransportException:
5185 except TDBException
as e:
5186 msg_type = TMessageType.REPLY
5188 except TApplicationException
as ex:
5189 logging.exception(
'TApplication exception in handler')
5190 msg_type = TMessageType.EXCEPTION
5193 logging.exception(
'Unexpected exception in handler')
5194 msg_type = TMessageType.EXCEPTION
5195 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5196 oprot.writeMessageBegin(
"get_tables", msg_type, seqid)
5198 oprot.writeMessageEnd()
5204 iprot.readMessageEnd()
5207 result.success = self._handler.get_tables_for_database(args.session, args.database_name)
5208 msg_type = TMessageType.REPLY
5209 except TTransport.TTransportException:
5211 except TDBException
as e:
5212 msg_type = TMessageType.REPLY
5214 except TApplicationException
as ex:
5215 logging.exception(
'TApplication exception in handler')
5216 msg_type = TMessageType.EXCEPTION
5219 logging.exception(
'Unexpected exception in handler')
5220 msg_type = TMessageType.EXCEPTION
5221 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5222 oprot.writeMessageBegin(
"get_tables_for_database", msg_type, seqid)
5224 oprot.writeMessageEnd()
5230 iprot.readMessageEnd()
5233 result.success = self._handler.get_physical_tables(args.session)
5234 msg_type = TMessageType.REPLY
5235 except TTransport.TTransportException:
5237 except TDBException
as e:
5238 msg_type = TMessageType.REPLY
5240 except TApplicationException
as ex:
5241 logging.exception(
'TApplication exception in handler')
5242 msg_type = TMessageType.EXCEPTION
5245 logging.exception(
'Unexpected exception in handler')
5246 msg_type = TMessageType.EXCEPTION
5247 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5248 oprot.writeMessageBegin(
"get_physical_tables", msg_type, seqid)
5250 oprot.writeMessageEnd()
5256 iprot.readMessageEnd()
5259 result.success = self._handler.get_views(args.session)
5260 msg_type = TMessageType.REPLY
5261 except TTransport.TTransportException:
5263 except TDBException
as e:
5264 msg_type = TMessageType.REPLY
5266 except TApplicationException
as ex:
5267 logging.exception(
'TApplication exception in handler')
5268 msg_type = TMessageType.EXCEPTION
5271 logging.exception(
'Unexpected exception in handler')
5272 msg_type = TMessageType.EXCEPTION
5273 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5274 oprot.writeMessageBegin(
"get_views", msg_type, seqid)
5276 oprot.writeMessageEnd()
5282 iprot.readMessageEnd()
5285 result.success = self._handler.get_tables_meta(args.session)
5286 msg_type = TMessageType.REPLY
5287 except TTransport.TTransportException:
5289 except TDBException
as e:
5290 msg_type = TMessageType.REPLY
5292 except TApplicationException
as ex:
5293 logging.exception(
'TApplication exception in handler')
5294 msg_type = TMessageType.EXCEPTION
5297 logging.exception(
'Unexpected exception in handler')
5298 msg_type = TMessageType.EXCEPTION
5299 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5300 oprot.writeMessageBegin(
"get_tables_meta", msg_type, seqid)
5302 oprot.writeMessageEnd()
5308 iprot.readMessageEnd()
5311 result.success = self._handler.get_table_details(args.session, args.table_name)
5312 msg_type = TMessageType.REPLY
5313 except TTransport.TTransportException:
5315 except TDBException
as e:
5316 msg_type = TMessageType.REPLY
5318 except TApplicationException
as ex:
5319 logging.exception(
'TApplication exception in handler')
5320 msg_type = TMessageType.EXCEPTION
5323 logging.exception(
'Unexpected exception in handler')
5324 msg_type = TMessageType.EXCEPTION
5325 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5326 oprot.writeMessageBegin(
"get_table_details", msg_type, seqid)
5328 oprot.writeMessageEnd()
5334 iprot.readMessageEnd()
5337 result.success = self._handler.get_table_details_for_database(args.session, args.table_name, args.database_name)
5338 msg_type = TMessageType.REPLY
5339 except TTransport.TTransportException:
5341 except TDBException
as e:
5342 msg_type = TMessageType.REPLY
5344 except TApplicationException
as ex:
5345 logging.exception(
'TApplication exception in handler')
5346 msg_type = TMessageType.EXCEPTION
5349 logging.exception(
'Unexpected exception in handler')
5350 msg_type = TMessageType.EXCEPTION
5351 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5352 oprot.writeMessageBegin(
"get_table_details_for_database", msg_type, seqid)
5354 oprot.writeMessageEnd()
5360 iprot.readMessageEnd()
5363 result.success = self._handler.get_internal_table_details(args.session, args.table_name, args.include_system_columns)
5364 msg_type = TMessageType.REPLY
5365 except TTransport.TTransportException:
5367 except TDBException
as e:
5368 msg_type = TMessageType.REPLY
5370 except TApplicationException
as ex:
5371 logging.exception(
'TApplication exception in handler')
5372 msg_type = TMessageType.EXCEPTION
5375 logging.exception(
'Unexpected exception in handler')
5376 msg_type = TMessageType.EXCEPTION
5377 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5378 oprot.writeMessageBegin(
"get_internal_table_details", msg_type, seqid)
5380 oprot.writeMessageEnd()
5386 iprot.readMessageEnd()
5389 result.success = self._handler.get_internal_table_details_for_database(args.session, args.table_name, args.database_name)
5390 msg_type = TMessageType.REPLY
5391 except TTransport.TTransportException:
5393 except TDBException
as e:
5394 msg_type = TMessageType.REPLY
5396 except TApplicationException
as ex:
5397 logging.exception(
'TApplication exception in handler')
5398 msg_type = TMessageType.EXCEPTION
5401 logging.exception(
'Unexpected exception in handler')
5402 msg_type = TMessageType.EXCEPTION
5403 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5404 oprot.writeMessageBegin(
"get_internal_table_details_for_database", msg_type, seqid)
5406 oprot.writeMessageEnd()
5412 iprot.readMessageEnd()
5415 result.success = self._handler.get_users(args.session)
5416 msg_type = TMessageType.REPLY
5417 except TTransport.TTransportException:
5419 except TDBException
as e:
5420 msg_type = TMessageType.REPLY
5422 except TApplicationException
as ex:
5423 logging.exception(
'TApplication exception in handler')
5424 msg_type = TMessageType.EXCEPTION
5427 logging.exception(
'Unexpected exception in handler')
5428 msg_type = TMessageType.EXCEPTION
5429 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5430 oprot.writeMessageBegin(
"get_users", msg_type, seqid)
5432 oprot.writeMessageEnd()
5438 iprot.readMessageEnd()
5441 result.success = self._handler.get_databases(args.session)
5442 msg_type = TMessageType.REPLY
5443 except TTransport.TTransportException:
5445 except TDBException
as e:
5446 msg_type = TMessageType.REPLY
5448 except TApplicationException
as ex:
5449 logging.exception(
'TApplication exception in handler')
5450 msg_type = TMessageType.EXCEPTION
5453 logging.exception(
'Unexpected exception in handler')
5454 msg_type = TMessageType.EXCEPTION
5455 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5456 oprot.writeMessageBegin(
"get_databases", msg_type, seqid)
5458 oprot.writeMessageEnd()
5464 iprot.readMessageEnd()
5467 result.success = self._handler.get_version()
5468 msg_type = TMessageType.REPLY
5469 except TTransport.TTransportException:
5471 except TDBException
as e:
5472 msg_type = TMessageType.REPLY
5474 except TApplicationException
as ex:
5475 logging.exception(
'TApplication exception in handler')
5476 msg_type = TMessageType.EXCEPTION
5479 logging.exception(
'Unexpected exception in handler')
5480 msg_type = TMessageType.EXCEPTION
5481 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5482 oprot.writeMessageBegin(
"get_version", msg_type, seqid)
5484 oprot.writeMessageEnd()
5490 iprot.readMessageEnd()
5493 self._handler.start_heap_profile(args.session)
5494 msg_type = TMessageType.REPLY
5495 except TTransport.TTransportException:
5497 except TDBException
as e:
5498 msg_type = TMessageType.REPLY
5500 except TApplicationException
as ex:
5501 logging.exception(
'TApplication exception in handler')
5502 msg_type = TMessageType.EXCEPTION
5505 logging.exception(
'Unexpected exception in handler')
5506 msg_type = TMessageType.EXCEPTION
5507 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5508 oprot.writeMessageBegin(
"start_heap_profile", msg_type, seqid)
5510 oprot.writeMessageEnd()
5516 iprot.readMessageEnd()
5519 self._handler.stop_heap_profile(args.session)
5520 msg_type = TMessageType.REPLY
5521 except TTransport.TTransportException:
5523 except TDBException
as e:
5524 msg_type = TMessageType.REPLY
5526 except TApplicationException
as ex:
5527 logging.exception(
'TApplication exception in handler')
5528 msg_type = TMessageType.EXCEPTION
5531 logging.exception(
'Unexpected exception in handler')
5532 msg_type = TMessageType.EXCEPTION
5533 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5534 oprot.writeMessageBegin(
"stop_heap_profile", msg_type, seqid)
5536 oprot.writeMessageEnd()
5542 iprot.readMessageEnd()
5545 result.success = self._handler.get_heap_profile(args.session)
5546 msg_type = TMessageType.REPLY
5547 except TTransport.TTransportException:
5549 except TDBException
as e:
5550 msg_type = TMessageType.REPLY
5552 except TApplicationException
as ex:
5553 logging.exception(
'TApplication exception in handler')
5554 msg_type = TMessageType.EXCEPTION
5557 logging.exception(
'Unexpected exception in handler')
5558 msg_type = TMessageType.EXCEPTION
5559 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5560 oprot.writeMessageBegin(
"get_heap_profile", msg_type, seqid)
5562 oprot.writeMessageEnd()
5568 iprot.readMessageEnd()
5571 result.success = self._handler.get_memory(args.session, args.memory_level)
5572 msg_type = TMessageType.REPLY
5573 except TTransport.TTransportException:
5575 except TDBException
as e:
5576 msg_type = TMessageType.REPLY
5578 except TApplicationException
as ex:
5579 logging.exception(
'TApplication exception in handler')
5580 msg_type = TMessageType.EXCEPTION
5583 logging.exception(
'Unexpected exception in handler')
5584 msg_type = TMessageType.EXCEPTION
5585 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5586 oprot.writeMessageBegin(
"get_memory", msg_type, seqid)
5588 oprot.writeMessageEnd()
5594 iprot.readMessageEnd()
5597 self._handler.clear_cpu_memory(args.session)
5598 msg_type = TMessageType.REPLY
5599 except TTransport.TTransportException:
5601 except TDBException
as e:
5602 msg_type = TMessageType.REPLY
5604 except TApplicationException
as ex:
5605 logging.exception(
'TApplication exception in handler')
5606 msg_type = TMessageType.EXCEPTION
5609 logging.exception(
'Unexpected exception in handler')
5610 msg_type = TMessageType.EXCEPTION
5611 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5612 oprot.writeMessageBegin(
"clear_cpu_memory", msg_type, seqid)
5614 oprot.writeMessageEnd()
5620 iprot.readMessageEnd()
5623 self._handler.clear_gpu_memory(args.session)
5624 msg_type = TMessageType.REPLY
5625 except TTransport.TTransportException:
5627 except TDBException
as e:
5628 msg_type = TMessageType.REPLY
5630 except TApplicationException
as ex:
5631 logging.exception(
'TApplication exception in handler')
5632 msg_type = TMessageType.EXCEPTION
5635 logging.exception(
'Unexpected exception in handler')
5636 msg_type = TMessageType.EXCEPTION
5637 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5638 oprot.writeMessageBegin(
"clear_gpu_memory", msg_type, seqid)
5640 oprot.writeMessageEnd()
5646 iprot.readMessageEnd()
5649 self._handler.set_cur_session(args.parent_session, args.leaf_session, args.start_time_str, args.label, args.for_running_query_kernel)
5650 msg_type = TMessageType.REPLY
5651 except TTransport.TTransportException:
5653 except TDBException
as e:
5654 msg_type = TMessageType.REPLY
5656 except TApplicationException
as ex:
5657 logging.exception(
'TApplication exception in handler')
5658 msg_type = TMessageType.EXCEPTION
5661 logging.exception(
'Unexpected exception in handler')
5662 msg_type = TMessageType.EXCEPTION
5663 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5664 oprot.writeMessageBegin(
"set_cur_session", msg_type, seqid)
5666 oprot.writeMessageEnd()
5672 iprot.readMessageEnd()
5675 self._handler.invalidate_cur_session(args.parent_session, args.leaf_session, args.start_time_str, args.label, args.for_running_query_kernel)
5676 msg_type = TMessageType.REPLY
5677 except TTransport.TTransportException:
5679 except TDBException
as e:
5680 msg_type = TMessageType.REPLY
5682 except TApplicationException
as ex:
5683 logging.exception(
'TApplication exception in handler')
5684 msg_type = TMessageType.EXCEPTION
5687 logging.exception(
'Unexpected exception in handler')
5688 msg_type = TMessageType.EXCEPTION
5689 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5690 oprot.writeMessageBegin(
"invalidate_cur_session", msg_type, seqid)
5692 oprot.writeMessageEnd()
5698 iprot.readMessageEnd()
5701 self._handler.set_table_epoch(args.session, args.db_id, args.table_id, args.new_epoch)
5702 msg_type = TMessageType.REPLY
5703 except TTransport.TTransportException:
5705 except TDBException
as e:
5706 msg_type = TMessageType.REPLY
5708 except TApplicationException
as ex:
5709 logging.exception(
'TApplication exception in handler')
5710 msg_type = TMessageType.EXCEPTION
5713 logging.exception(
'Unexpected exception in handler')
5714 msg_type = TMessageType.EXCEPTION
5715 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5716 oprot.writeMessageBegin(
"set_table_epoch", msg_type, seqid)
5718 oprot.writeMessageEnd()
5724 iprot.readMessageEnd()
5727 self._handler.set_table_epoch_by_name(args.session, args.table_name, args.new_epoch)
5728 msg_type = TMessageType.REPLY
5729 except TTransport.TTransportException:
5731 except TDBException
as e:
5732 msg_type = TMessageType.REPLY
5734 except TApplicationException
as ex:
5735 logging.exception(
'TApplication exception in handler')
5736 msg_type = TMessageType.EXCEPTION
5739 logging.exception(
'Unexpected exception in handler')
5740 msg_type = TMessageType.EXCEPTION
5741 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5742 oprot.writeMessageBegin(
"set_table_epoch_by_name", msg_type, seqid)
5744 oprot.writeMessageEnd()
5750 iprot.readMessageEnd()
5753 result.success = self._handler.get_table_epoch(args.session, args.db_id, args.table_id)
5754 msg_type = TMessageType.REPLY
5755 except TTransport.TTransportException:
5757 except TApplicationException
as ex:
5758 logging.exception(
'TApplication exception in handler')
5759 msg_type = TMessageType.EXCEPTION
5762 logging.exception(
'Unexpected exception in handler')
5763 msg_type = TMessageType.EXCEPTION
5764 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5765 oprot.writeMessageBegin(
"get_table_epoch", msg_type, seqid)
5767 oprot.writeMessageEnd()
5773 iprot.readMessageEnd()
5776 result.success = self._handler.get_table_epoch_by_name(args.session, args.table_name)
5777 msg_type = TMessageType.REPLY
5778 except TTransport.TTransportException:
5780 except TApplicationException
as ex:
5781 logging.exception(
'TApplication exception in handler')
5782 msg_type = TMessageType.EXCEPTION
5785 logging.exception(
'Unexpected exception in handler')
5786 msg_type = TMessageType.EXCEPTION
5787 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5788 oprot.writeMessageBegin(
"get_table_epoch_by_name", msg_type, seqid)
5790 oprot.writeMessageEnd()
5796 iprot.readMessageEnd()
5799 result.success = self._handler.get_table_epochs(args.session, args.db_id, args.table_id)
5800 msg_type = TMessageType.REPLY
5801 except TTransport.TTransportException:
5803 except TApplicationException
as ex:
5804 logging.exception(
'TApplication exception in handler')
5805 msg_type = TMessageType.EXCEPTION
5808 logging.exception(
'Unexpected exception in handler')
5809 msg_type = TMessageType.EXCEPTION
5810 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5811 oprot.writeMessageBegin(
"get_table_epochs", msg_type, seqid)
5813 oprot.writeMessageEnd()
5819 iprot.readMessageEnd()
5822 self._handler.set_table_epochs(args.session, args.db_id, args.table_epochs)
5823 msg_type = TMessageType.REPLY
5824 except TTransport.TTransportException:
5826 except TApplicationException
as ex:
5827 logging.exception(
'TApplication exception in handler')
5828 msg_type = TMessageType.EXCEPTION
5831 logging.exception(
'Unexpected exception in handler')
5832 msg_type = TMessageType.EXCEPTION
5833 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5834 oprot.writeMessageBegin(
"set_table_epochs", msg_type, seqid)
5836 oprot.writeMessageEnd()
5842 iprot.readMessageEnd()
5845 result.success = self._handler.get_session_info(args.session)
5846 msg_type = TMessageType.REPLY
5847 except TTransport.TTransportException:
5849 except TDBException
as e:
5850 msg_type = TMessageType.REPLY
5852 except TApplicationException
as ex:
5853 logging.exception(
'TApplication exception in handler')
5854 msg_type = TMessageType.EXCEPTION
5857 logging.exception(
'Unexpected exception in handler')
5858 msg_type = TMessageType.EXCEPTION
5859 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5860 oprot.writeMessageBegin(
"get_session_info", msg_type, seqid)
5862 oprot.writeMessageEnd()
5868 iprot.readMessageEnd()
5871 result.success = self._handler.get_queries_info(args.session)
5872 msg_type = TMessageType.REPLY
5873 except TTransport.TTransportException:
5875 except TDBException
as e:
5876 msg_type = TMessageType.REPLY
5878 except TApplicationException
as ex:
5879 logging.exception(
'TApplication exception in handler')
5880 msg_type = TMessageType.EXCEPTION
5883 logging.exception(
'Unexpected exception in handler')
5884 msg_type = TMessageType.EXCEPTION
5885 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5886 oprot.writeMessageBegin(
"get_queries_info", msg_type, seqid)
5888 oprot.writeMessageEnd()
5894 iprot.readMessageEnd()
5897 self._handler.set_leaf_info(args.session, args.leaf_info)
5898 msg_type = TMessageType.REPLY
5899 except TTransport.TTransportException:
5901 except TDBException
as e:
5902 msg_type = TMessageType.REPLY
5904 except TApplicationException
as ex:
5905 logging.exception(
'TApplication exception in handler')
5906 msg_type = TMessageType.EXCEPTION
5909 logging.exception(
'Unexpected exception in handler')
5910 msg_type = TMessageType.EXCEPTION
5911 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5912 oprot.writeMessageBegin(
"set_leaf_info", msg_type, seqid)
5914 oprot.writeMessageEnd()
5920 iprot.readMessageEnd()
5923 result.success = self._handler.sql_execute(args.session, args.query, args.column_format, args.nonce, args.first_n, args.at_most_n)
5924 msg_type = TMessageType.REPLY
5925 except TTransport.TTransportException:
5927 except TDBException
as e:
5928 msg_type = TMessageType.REPLY
5930 except TApplicationException
as ex:
5931 logging.exception(
'TApplication exception in handler')
5932 msg_type = TMessageType.EXCEPTION
5935 logging.exception(
'Unexpected exception in handler')
5936 msg_type = TMessageType.EXCEPTION
5937 result = TApplicationException(TApplicationException.INTERNAL_ERROR,
'Internal error')
5938 oprot.writeMessageBegin(
"sql_execute", msg_type, seqid)
5940 oprot.writeMessageEnd()
5946 iprot.readMessageEnd()
5949 result.success = self._handler.sql_execute_df(args.session, args.query, args.device_type, args.device_id, args.first_n, args.transport_method)
5950 msg_type = TMessageType.REPLY
5951 except TTransport.TTransportException:
5953 except TDBException
as e: