packaging/src/docker/start-hive.sh

# 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.

#!/usr/bin/env bash
set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
cd "$SCRIPT_DIR"

MODE="container"
PROFILE=""
SCALE=""
COMPOSE_FILES="docker-compose.yml"

for arg in "$@"; do
  case "$arg" in
    --llap)
      MODE="llap"
      PROFILE="--profile llap"
      SCALE="--scale llapdaemon=2"
      export HIVE_ZOOKEEPER_QUORUM=zookeeper:2181
      export HIVE_LLAP_DAEMON_SERVICE_HOSTS=@llap0
      ;;
--ozone)
      COMPOSE_FILES="docker-compose.yml:storage/ozone/docker-compose.yml"
      # DEFAULT_FS defines the bucket authority
      export DEFAULT_FS="s3a://hive"

      export HIVE_WAREHOUSE_PATH="/warehouse"

      export S3_ENDPOINT_URL="http://s3.ozone:9878"
      export AWS_ACCESS_KEY_ID="ozone"
      export AWS_SECRET_ACCESS_KEY="secret"
      ;;
    *)
      echo "Unknown option: $arg"
      exit 1
      ;;
  esac
done

export HIVE_EXECUTION_MODE="$MODE"
export COMPOSE_FILE="$COMPOSE_FILES"

echo "Starting Hive cluster (mode=$HIVE_EXECUTION_MODE, compose_files=$COMPOSE_FILE)"

docker compose $PROFILE up -d $SCALE