ImpalaService-remote 6.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243
  1. #!/usr/bin/env python
  2. #
  3. # Autogenerated by Thrift Compiler (0.13.0)
  4. #
  5. # DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
  6. #
  7. # options string: py:new_style
  8. #
  9. import sys
  10. import pprint
  11. if sys.version_info[0] > 2:
  12. from urllib.parse import urlparse
  13. else:
  14. from urlparse import urlparse
  15. from thrift.transport import TTransport, TSocket, TSSLSocket, THttpClient
  16. from thrift.protocol.TBinaryProtocol import TBinaryProtocol
  17. from ImpalaService import ImpalaService
  18. from ImpalaService.ttypes import *
  19. if len(sys.argv) <= 1 or sys.argv[1] == '--help':
  20. print('')
  21. print('Usage: ' + sys.argv[0] + ' [-h host[:port]] [-u url] [-f[ramed]] [-s[sl]] [-novalidate] [-ca_certs certs] [-keyfile keyfile] [-certfile certfile] function [arg1 [arg2...]]')
  22. print('')
  23. print('Functions:')
  24. print(' TStatus Cancel(QueryHandle query_id)')
  25. print(' TStatus ResetCatalog()')
  26. print(' TStatus ResetTable(TResetTableReq request)')
  27. print(' string GetRuntimeProfile(QueryHandle query_id)')
  28. print(' TInsertResult CloseInsert(QueryHandle handle)')
  29. print(' TPingImpalaServiceResp PingImpalaService()')
  30. print(' TExecSummary GetExecSummary(QueryHandle handle)')
  31. print(' QueryHandle query(Query query)')
  32. print(' QueryHandle executeAndWait(Query query, LogContextId clientCtx)')
  33. print(' QueryExplanation explain(Query query)')
  34. print(' Results fetch(QueryHandle query_id, bool start_over, i32 fetch_size)')
  35. print(' QueryState get_state(QueryHandle handle)')
  36. print(' ResultsMetadata get_results_metadata(QueryHandle handle)')
  37. print(' string echo(string s)')
  38. print(' string dump_config()')
  39. print(' string get_log(LogContextId context)')
  40. print(' get_default_configuration(bool include_hadoop)')
  41. print(' void close(QueryHandle handle)')
  42. print(' void clean(LogContextId log_context)')
  43. print('')
  44. sys.exit(0)
  45. pp = pprint.PrettyPrinter(indent=2)
  46. host = 'localhost'
  47. port = 9090
  48. uri = ''
  49. framed = False
  50. ssl = False
  51. validate = True
  52. ca_certs = None
  53. keyfile = None
  54. certfile = None
  55. http = False
  56. argi = 1
  57. if sys.argv[argi] == '-h':
  58. parts = sys.argv[argi + 1].split(':')
  59. host = parts[0]
  60. if len(parts) > 1:
  61. port = int(parts[1])
  62. argi += 2
  63. if sys.argv[argi] == '-u':
  64. url = urlparse(sys.argv[argi + 1])
  65. parts = url[1].split(':')
  66. host = parts[0]
  67. if len(parts) > 1:
  68. port = int(parts[1])
  69. else:
  70. port = 80
  71. uri = url[2]
  72. if url[4]:
  73. uri += '?%s' % url[4]
  74. http = True
  75. argi += 2
  76. if sys.argv[argi] == '-f' or sys.argv[argi] == '-framed':
  77. framed = True
  78. argi += 1
  79. if sys.argv[argi] == '-s' or sys.argv[argi] == '-ssl':
  80. ssl = True
  81. argi += 1
  82. if sys.argv[argi] == '-novalidate':
  83. validate = False
  84. argi += 1
  85. if sys.argv[argi] == '-ca_certs':
  86. ca_certs = sys.argv[argi+1]
  87. argi += 2
  88. if sys.argv[argi] == '-keyfile':
  89. keyfile = sys.argv[argi+1]
  90. argi += 2
  91. if sys.argv[argi] == '-certfile':
  92. certfile = sys.argv[argi+1]
  93. argi += 2
  94. cmd = sys.argv[argi]
  95. args = sys.argv[argi + 1:]
  96. if http:
  97. transport = THttpClient.THttpClient(host, port, uri)
  98. else:
  99. if ssl:
  100. socket = TSSLSocket.TSSLSocket(host, port, validate=validate, ca_certs=ca_certs, keyfile=keyfile, certfile=certfile)
  101. else:
  102. socket = TSocket.TSocket(host, port)
  103. if framed:
  104. transport = TTransport.TFramedTransport(socket)
  105. else:
  106. transport = TTransport.TBufferedTransport(socket)
  107. protocol = TBinaryProtocol(transport)
  108. client = ImpalaService.Client(protocol)
  109. transport.open()
  110. if cmd == 'Cancel':
  111. if len(args) != 1:
  112. print('Cancel requires 1 args')
  113. sys.exit(1)
  114. pp.pprint(client.Cancel(eval(args[0]),))
  115. elif cmd == 'ResetCatalog':
  116. if len(args) != 0:
  117. print('ResetCatalog requires 0 args')
  118. sys.exit(1)
  119. pp.pprint(client.ResetCatalog())
  120. elif cmd == 'ResetTable':
  121. if len(args) != 1:
  122. print('ResetTable requires 1 args')
  123. sys.exit(1)
  124. pp.pprint(client.ResetTable(eval(args[0]),))
  125. elif cmd == 'GetRuntimeProfile':
  126. if len(args) != 1:
  127. print('GetRuntimeProfile requires 1 args')
  128. sys.exit(1)
  129. pp.pprint(client.GetRuntimeProfile(eval(args[0]),))
  130. elif cmd == 'CloseInsert':
  131. if len(args) != 1:
  132. print('CloseInsert requires 1 args')
  133. sys.exit(1)
  134. pp.pprint(client.CloseInsert(eval(args[0]),))
  135. elif cmd == 'PingImpalaService':
  136. if len(args) != 0:
  137. print('PingImpalaService requires 0 args')
  138. sys.exit(1)
  139. pp.pprint(client.PingImpalaService())
  140. elif cmd == 'GetExecSummary':
  141. if len(args) != 1:
  142. print('GetExecSummary requires 1 args')
  143. sys.exit(1)
  144. pp.pprint(client.GetExecSummary(eval(args[0]),))
  145. elif cmd == 'query':
  146. if len(args) != 1:
  147. print('query requires 1 args')
  148. sys.exit(1)
  149. pp.pprint(client.query(eval(args[0]),))
  150. elif cmd == 'executeAndWait':
  151. if len(args) != 2:
  152. print('executeAndWait requires 2 args')
  153. sys.exit(1)
  154. pp.pprint(client.executeAndWait(eval(args[0]), eval(args[1]),))
  155. elif cmd == 'explain':
  156. if len(args) != 1:
  157. print('explain requires 1 args')
  158. sys.exit(1)
  159. pp.pprint(client.explain(eval(args[0]),))
  160. elif cmd == 'fetch':
  161. if len(args) != 3:
  162. print('fetch requires 3 args')
  163. sys.exit(1)
  164. pp.pprint(client.fetch(eval(args[0]), eval(args[1]), eval(args[2]),))
  165. elif cmd == 'get_state':
  166. if len(args) != 1:
  167. print('get_state requires 1 args')
  168. sys.exit(1)
  169. pp.pprint(client.get_state(eval(args[0]),))
  170. elif cmd == 'get_results_metadata':
  171. if len(args) != 1:
  172. print('get_results_metadata requires 1 args')
  173. sys.exit(1)
  174. pp.pprint(client.get_results_metadata(eval(args[0]),))
  175. elif cmd == 'echo':
  176. if len(args) != 1:
  177. print('echo requires 1 args')
  178. sys.exit(1)
  179. pp.pprint(client.echo(args[0],))
  180. elif cmd == 'dump_config':
  181. if len(args) != 0:
  182. print('dump_config requires 0 args')
  183. sys.exit(1)
  184. pp.pprint(client.dump_config())
  185. elif cmd == 'get_log':
  186. if len(args) != 1:
  187. print('get_log requires 1 args')
  188. sys.exit(1)
  189. pp.pprint(client.get_log(eval(args[0]),))
  190. elif cmd == 'get_default_configuration':
  191. if len(args) != 1:
  192. print('get_default_configuration requires 1 args')
  193. sys.exit(1)
  194. pp.pprint(client.get_default_configuration(eval(args[0]),))
  195. elif cmd == 'close':
  196. if len(args) != 1:
  197. print('close requires 1 args')
  198. sys.exit(1)
  199. pp.pprint(client.close(eval(args[0]),))
  200. elif cmd == 'clean':
  201. if len(args) != 1:
  202. print('clean requires 1 args')
  203. sys.exit(1)
  204. pp.pprint(client.clean(eval(args[0]),))
  205. else:
  206. print('Unrecognized method %s' % cmd)
  207. sys.exit(1)
  208. transport.close()