mirror of
https://github.com/apache/impala.git
synced 2025-12-19 18:12:08 -05:00
b0f1d49042
This patch adds a new MetaProvider called MultiMetaProvider, which is capable of handling multiple MetaProviders at once, prioritizing one primary provider over multiple secondary providers. The primary provider handles some methods exclusively for deterministic behavior. In database listings, if one database name occurs multiple times the contained tables are merged under that database name; if the two separate databases contain a table with the same name, the query analyzation fails with an error. This change also modifies the local catalog implementation's initialization. If catalogd is deployed, then it instantiates the CatalogdMetaProvider and checks if the catalog configuration directory is set as a backend flag. If it's set, then it tries to load every configuration from the folder, and tries to instantiate the IcebergMetaProvider from those configs. If the instantiation fails, an error is reported to the logs, but the startup is not interrupted. Tests: - E2E tests for multi-catalog behavior - Unit test for ConfigLoader Change-Id: Ifbdd0f7085345e7954d9f6f264202699182dd1e1 Reviewed-on: http://gerrit.cloudera.org:8080/22878 Tested-by: Impala Public Jenkins <impala-public-jenkins@cloudera.com> Reviewed-by: Zoltan Borok-Nagy <boroknagyz@cloudera.com>
100 lines
3.6 KiB
Python
100 lines
3.6 KiB
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.
|
|
|
|
from __future__ import absolute_import, division, print_function
|
|
import logging
|
|
import os
|
|
import subprocess
|
|
import signal
|
|
import socket
|
|
import sys
|
|
import time
|
|
|
|
IMPALA_HOME = os.environ['IMPALA_HOME']
|
|
IMPALA_CLUSTER_LOGS = os.environ['IMPALA_CLUSTER_LOGS_DIR']
|
|
LOG = logging.getLogger('impala_test_suite')
|
|
|
|
|
|
class IcebergRestServer(object):
|
|
"""
|
|
Utility class for starting and stopping our minimal Iceberg REST server.
|
|
"""
|
|
|
|
DEFAULT_REST_SERVER_PORT = 9084
|
|
DEFAULT_CATALOG_LOCATION = "/test-warehouse/iceberg_test/hadoop_catalog"
|
|
LOG_PATTERN = "%s/iceberg-rest-server.%d.%s"
|
|
|
|
def __init__(self, port=DEFAULT_REST_SERVER_PORT,
|
|
catalog_location=DEFAULT_CATALOG_LOCATION):
|
|
self.port = port
|
|
self.catalog_location = catalog_location
|
|
self.process = None
|
|
self.stdout = None
|
|
self.stderr = None
|
|
|
|
def start_rest_server(self, timeout_s=60):
|
|
start_time = time.strftime("%Y%m%d-%H%M%S")
|
|
log_pattern = self.LOG_PATTERN % (IMPALA_CLUSTER_LOGS, self.port, start_time)
|
|
self.stdout = open(log_pattern + '.stdout', 'w')
|
|
self.stderr = open(log_pattern + '.stderr', 'w')
|
|
self.process = subprocess.Popen(
|
|
['testdata/bin/run-iceberg-rest-server.sh', "--port",
|
|
str(self.port), "--catalog-location", self.catalog_location],
|
|
stdout=self.stdout, stderr=self.stderr,
|
|
preexec_fn=os.setsid, cwd=IMPALA_HOME)
|
|
self._wait_for_rest_server_to_start(timeout_s)
|
|
|
|
def stop_rest_server(self, timeout_s=60):
|
|
try:
|
|
if self.process:
|
|
os.killpg(self.process.pid, signal.SIGTERM)
|
|
self._wait_for_rest_server_to_be_killed(timeout_s)
|
|
except Exception as e:
|
|
LOG.error("An error occurred while stopping the Iceberg REST server: %s", e)
|
|
finally:
|
|
if self.stdout:
|
|
self.stdout.close()
|
|
if self.stderr:
|
|
self.stderr.close()
|
|
|
|
def _wait_for_rest_server_to_start(self, timeout_s):
|
|
sleep_interval_s = 0.5
|
|
start_time = time.time()
|
|
while time.time() - start_time < timeout_s:
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
if s.connect_ex(('localhost', self.port)) == 0:
|
|
LOG.info("Iceberg REST server is available.")
|
|
return
|
|
s.close()
|
|
time.sleep(sleep_interval_s)
|
|
raise Exception("Webserver did not become available within {} "
|
|
"seconds.".format(timeout_s))
|
|
|
|
def _wait_for_rest_server_to_be_killed(self, timeout_s):
|
|
sleep_interval_s = 0.5
|
|
start_time = time.time()
|
|
while time.time() - start_time < timeout_s:
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
if s.connect_ex(('localhost', self.port)) != 0:
|
|
LOG.info("Iceberg REST server has stopped.")
|
|
return
|
|
s.close()
|
|
time.sleep(sleep_interval_s)
|
|
# Let's not throw an exception as this is typically invoked during cleanup, and we
|
|
# want the rest of the cleanup code to be executed.
|
|
LOG.info("Iceberg REST server hasn't stopped in time.")
|