mirror of
https://github.com/apache/impala.git
synced 2025-12-19 09:58:28 -05:00
c5a0ec8bdf
This puts all of the thrift-generated python code into the impala_thrift_gen package. This is similar to what Impyla does for its thrift-generated python code, except that it uses the impala_thrift_gen package rather than impala._thrift_gen. This is a preparatory patch for fixing the absolute import issues. This patches all of the thrift files to add the python namespace. This has code to apply the patching to the thirdparty thrift files (hive_metastore.thrift, fb303.thrift) to do the same. Putting all the generated python into a package makes it easier to understand where the imports are getting code. When the subsequent change rearranges the shell code, the thrift generated code can stay in a separate directory. This uses isort to sort the imports for the affected Python files with the provided .isort.cfg file. This also adds an impala-isort shell script to make it easy to run. Testing: - Ran a core job Change-Id: Ie2927f22c7257aa38a78084efe5bd76d566493c0 Reviewed-on: http://gerrit.cloudera.org:8080/20169 Tested-by: Impala Public Jenkins <impala-public-jenkins@cloudera.com> Reviewed-by: Riza Suminto <riza.suminto@cloudera.com>
92 lines
3.7 KiB
Python
Executable File
92 lines
3.7 KiB
Python
Executable File
#!/usr/bin/env impala-python
|
|
#
|
|
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
#
|
|
# This script waits for the Hive HiveServer2 service to become available by attempting
|
|
# to create a new session until the session creation succeeds, or a timeout is reached.
|
|
# TODO: Consider combining this with wait-for-metastore.py. A TCLIService client
|
|
# can perhaps also talk to the metastore.
|
|
|
|
from __future__ import absolute_import, division, print_function
|
|
import os
|
|
import time
|
|
import getpass
|
|
from optparse import OptionParser
|
|
from tests.util.thrift_util import create_transport
|
|
|
|
# Imports required for HiveServer2 Client
|
|
from impala_thrift_gen.TCLIService import TCLIService
|
|
from thrift.transport import TTransport, TSocket
|
|
from thrift.protocol import TBinaryProtocol
|
|
|
|
# Disable buffering to get "print" output on the console immediately.
|
|
os.environ["PYTHONUNBUFFERED"] = "1"
|
|
|
|
parser = OptionParser()
|
|
parser.add_option("--hs2_hostport", dest="hs2_hostport",
|
|
default="localhost:11050", help="HiveServer2 hostport to wait for.")
|
|
parser.add_option("--transport", dest="transport", default="buffered",
|
|
help="Transport to use for connecting to HiveServer2. Valid values: "
|
|
"'buffered', 'kerberos', 'plain_sasl'.")
|
|
options, args = parser.parse_args()
|
|
|
|
hs2_host, hs2_port = options.hs2_hostport.split(':')
|
|
|
|
if options.transport == "plain_sasl":
|
|
# Here we supply a bogus username of "foo" and a bogus password of "bar".
|
|
# We just have to supply *something*, else HS2 will block waiting for user
|
|
# input. Any bogus username and password are accepted.
|
|
hs2_transport = create_transport(hs2_host, hs2_port, "hive", options.transport,
|
|
"foo", "bar")
|
|
else:
|
|
hs2_transport = create_transport(hs2_host, hs2_port, "hive", options.transport)
|
|
|
|
protocol = TBinaryProtocol.TBinaryProtocol(hs2_transport)
|
|
hs2_client = TCLIService.Client(protocol)
|
|
|
|
# Try to connect to the HiveServer2 service and create a session
|
|
now = time.time()
|
|
TIMEOUT_SECONDS = 300.0
|
|
while time.time() - now < TIMEOUT_SECONDS:
|
|
try:
|
|
hs2_transport.open()
|
|
open_session_req = TCLIService.TOpenSessionReq()
|
|
open_session_req.username = getpass.getuser()
|
|
resp = hs2_client.OpenSession(open_session_req)
|
|
if resp.status.statusCode == TCLIService.TStatusCode.SUCCESS_STATUS:
|
|
close_session_req = TCLIService.TCloseSessionReq()
|
|
close_session_req.sessionHandle = resp.sessionHandle
|
|
hs2_client.CloseSession(close_session_req)
|
|
print("HiveServer2 service is up at %s." % options.hs2_hostport)
|
|
exit(0)
|
|
except Exception as e:
|
|
if "SASL" in str(e): # Bail out on SASL failures
|
|
print("SASL failure when attempting connection:")
|
|
raise
|
|
if "GSS" in str(e): # Other GSSAPI failures
|
|
print("GSS failure when attempting connection:")
|
|
raise
|
|
print("Waiting for HiveServer2 at %s..." % options.hs2_hostport)
|
|
print(e)
|
|
finally:
|
|
hs2_transport.close()
|
|
time.sleep(0.5)
|
|
|
|
print("HiveServer2 service failed to start within %s seconds." % TIMEOUT_SECONDS)
|
|
exit(1)
|