jprdonnelly/impala
1
0
Fork 0
mirror of https://github.com/apache/impala.git synced 2025-12-23 21:08:39 -05:00
Code Issues Projects Releases Wiki Activity

IMP-690: impala-shell calls PingImpalaService thrift API to verify

Browse Source 
the connected server is an impalad.
This commit is contained in:
Alan Choi
2013-01-11 08:45:56 -08:00
committed by Henry Robinson
parent 20f9d46328
commit ff704ce586
4 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
shell/impala_shell.py
View File

@@ -230,15 +230,23 @@ class ImpalaShell(cmd.Cmd):
self.transport.close()
self.transport = None
self.connected = False
try:
self.transport = self.__get_transport()
self.transport.open()
protocol = TBinaryProtocol.TBinaryProtocol(self.transport)
self.imp_service = ImpalaService.Client(protocol)
self.connected = True
try:
self.imp_service.PingImpalaService()
self.connected = True
except Exception, e:
print ("Error: Unable to communicate with impalad service. This service may not "
"be an impalad instance. Check host:port and try again.")
self.transport.close()
raise
except Exception, e:
print "Error connecting: %s, %s" % (type(e),e)
self.connected = False
return self.connected
def __get_transport(self):