GIT_CFG_USER="$(git config user.name)"

echo -n "Enter Github.com Username, press return to use default of '${GIT_CFG_USER}'> "

read -r GIT_USER

if [[ -z "${GIT_USER}" ]]; then

GIT_USER="${GIT_CFG_USER}"

fi

pushd "${IMPALA_HOME}"

git remote add asf-gerrit "ssh://${GIT_USER}@gerrit.cloudera.org:29418/Impala-ASF" 2>/dev/null

git remote set-url asf-gerrit "ssh://${GIT_USER}@gerrit.cloudera.org:29418/Impala-ASF"

echo

echo "asf-gerrit remote:"

git remote -v | grep asf-gerrit

echo

popd

echo

echo "Add this SSH public key to Gerrit: "

cat "${HOME}/.ssh/id_ed25519.pub"

echo

{

BASENAME="${1%.*}"

BASENAME=${BASENAME##*/}

MAKEFILE=$(grep -l "$BASENAME.cc.o:" $IMPALA_HOME/be/src/*/CMakeFiles/*.dir/build.make)

MAKEFILE=${MAKEFILE#$IMPALA_HOME/}

make DEBUG_NOOPT=1 -s -f $MAKEFILE ${MAKEFILE%/*}/$BASENAME.cc.o

local target_test=${1}

if [[ -z ${target_test} ]]; then

echo "Error: undefined target test"

else

mvn \

-Dmaven.surefire.debug="-Xdebug \

-Djava.compiler=NONE" \

-fae \

-Dtest="${target_test}" \

test

fi

local target_test=${1}

if [[ -z ${target_test} ]]; then

echo "Error: undefined target test"

else

mvn test -Dtest="${target_test}"

fi

local log_dir="${1:-$IMPALA_LOGS_DIR/cluster}"

local minidump_file

if [[ "${log_dir}" == *".dmp" ]]; then

minidump_file="${log_dir}"

else

minidump_file=$(grep "Wrote minidump to " "${log_dir}/impalad.ERROR" | rev | cut -d" " -f1 | rev)

fi

if [[ -z "${minidump_file}" ]]; then

echo "Error: minidump file '${minidump_file}' not found"

return 1

fi

echo "Resolving minidump: ${minidump_file}"

resolve_minidumps.py --minidump_file "${minidump_file}" --output_file /tmp/resolve.txt

cat /tmp/resolve.txt | head -n 100

local db_name="${1:-gisdb}"

echo "Setting up PostGIS extension for PostgreSQL in the '${db_name}' database."

sudo apt install -y postgresql-common gnupg2

sudo /usr/share/postgresql-common/pgdg/apt.postgresql.org.sh -y

POSTGRES_VER=$(sudo -u postgres psql -c "select version();" | grep "PostgreSQL" | sed -E 's/^\s*PostgreSQL\s*([0-9]+).*$/\1/')

sudo apt install -y "postgresql-${POSTGRES_VER}-postgis-3"

sudo apt install -y postgis

sudo -u postgres --preserve-env=db_name psql -c "CREATE DATABASE ${db_name};"

sudo -u postgres --preserve-env=db_name psql -c "ALTER DATABASE ${db_name} OWNER TO hiveuser;"

sudo -u postgres --preserve-env=db_name psql -d "${db_name}" -c "CREATE EXTENSION postgis;"

sudo -u postgres --preserve-env=db_name psql -d "${db_name}" -c "SELECT postgis_full_version();"

echo; echo "Adding connection to VSCode settings.json for database '${db_name}'."

if [[ $(grep -c "localhost-${db_name}" "${IMPALA_HOME}/.vscode/settings.json") -gt 0 ]]; then

echo "Connection localhost-${db_name} already exists; no change made."

else

awk -v db_name="${db_name}" '

' "${IMPALA_HOME}/.vscode/settings.json" > "${IMPALA_HOME}/.vscode/settings.json.tmp" && mv "${IMPALA_HOME}/.vscode/settings.json.tmp" "${IMPALA_HOME}/.vscode/settings.json"

fi

# Define colors

local CYAN='\033[0;36m'

local GREEN='\033[0;32m'

local YELLOW='\033[1;33m'

local BOLD='\033[1m'

local NC='\033[0m'

echo

echo

echo -e "${CYAN}${BOLD}Welcome to Impala Devcontainer${NC}"

echo "Here are some helpful commands to get you started:"

echo

echo -e "${GREEN}${BOLD}Dependency Management Commands${NC}"

echo -e " * ${YELLOW}restart-hdfs${NC}: Starts or restarts HDFS"

echo -e " * ${YELLOW}restart-hive${NC}: Starts or restarts Hive"

echo -e " * ${YELLOW}restart-minicluster${NC}: Starts or restarts all supporting services (HMS, HDFS, Kudu, etc)"

echo -e " * ${YELLOW}setup-postgis${NC}: Sets up PostGIS extension for PostgreSQL. Defaults to the 'gisdb' database. Can specify a different database name as the first argument."

echo

echo -e "${GREEN}${BOLD}Build Commands${NC}"

echo -e " * ${YELLOW}build-be${NC}: Build the Impala backend code."

echo -e " * ${YELLOW}build-fe${NC}: Build the Impala frontend code."

echo -e " * ${YELLOW}build-all${NC}: Build the entire Impala project (including frontend, backend, and test code) without clean or running tests."

echo -e " * ${YELLOW}omake${NC}: Build a single '.o' file. Pass only the file name without any extension. For example, 'omake rpc-trace'."

echo

echo -e "${GREEN}${BOLD}Test/Debugging Commands${NC}"

echo -e " * ${YELLOW}mvn-debug-front-end${NC}: Runs JUnit tests in debug mode and waits for a debugger to connect. Use the 'JUnit FE Debug' debug configuration to connect to the waiting tests. Specify the tests to run as a comma separated string of class names or ClassName#testMethodName. Must be run from the 'fe' directory."

echo -e " * ${YELLOW}mvn-run-front-end${NC}: Runs JUnit tests. Specify the tests to run as a comma separated string of class names or ClassName#testMethodName. Must be run from the 'fe' directory."

echo -e " * ${YELLOW}resolve-minidump${NC}: Resolves minidump files to '/tmp/resolve.txt', and prints the first 100 lines. Can be used in multiple ways:"

echo -e " 1. No arguments -- Parses impalad.ERROR log file to locate the minidump file to resolve."

echo -e " 2. Log dir as only argument -- Parses impalad.ERROR log file in the specified directory to locate the minidump file to resolve."

echo -e " 3. Minidump file as only argument -- Resolves the specified minidump file to '/tmp/resolve.txt' and prints the first 100 lines."

echo

echo -e "${GREEN}${BOLD}Git Aliases${NC}"

echo -e " * ${YELLOW}ga${NC}: alias for 'git add'"

echo -e " * ${YELLOW}gb${NC}: alias for 'git branch'"

echo -e " * ${YELLOW}gba${NC}: alias for 'git branch -a'"

echo -e " * ${YELLOW}gbv${NC}: alias for 'git branch -v'"

echo -e " * ${YELLOW}gc${NC}: alias for 'git commit -v'"

echo -e " * ${YELLOW}gd${NC}: alias for 'git diff'"

echo -e " * ${YELLOW}grv${NC}: alias for 'git remote -v'"

echo -e " * ${YELLOW}gs${NC}: alias for 'git status'"

echo

echo -e "${GREEN}${BOLD}Gerrit Commands:${NC}"

echo -e " * ${YELLOW}critique${NC}: Dry run the Jenkins critque Gerrit review."

echo -e " * ${YELLOW}push-asf-gerrit-draft${NC}: Pushes the current branch to Gerrit as a draft patch set."

echo -e " * ${YELLOW}push-asf-gerrit-publish${NC}: Pushes the current branch to Gerrit as a published patch set."

echo -e " * ${YELLOW}setup-asf-gerrit-remote${NC}: Configures the 'asf-gerrit' git remote to point to the Gerrit repository for Impala. The remote is added with the SSH URL, so make sure you have your SSH keys set up properly to push to Gerrit."

echo