#!/usr/bin/env bash

ROOT_DIR=$([[ $PWD == */bin ]] && dirname $PWD || echo $PWD)

DRUID_VERSION="0.9.2-SNAPSHOT"
DRUID_PATH=${ROOT_DIR}/dist/druid-${DRUID_VERSION}
MYSQL_PATH=${ROOT_DIR}/dist/mysql-metadata-storage-${DRUID_VERSION}.tar.gz

[[ -d ${DRUID_PATH} && -f ${MYSQL_PATH} ]] || {
    echo "Druid distribution should be here $DRUID_PATH !!!"
    exit 1
}

echo "Uncompressing latest version of myslq-metadata-storage ..."

tar -xf ${MYSQL_PATH} -C ${DRUID_PATH}/extensions/

echo "Initiating Mysql metadata storage recovery from s3 deep storage ..."

DRUID_CLASSPATH="$DRUID_PATH/lib"

echo "Classpath $DRUID_CLASSPATH ..."

cd ${DRUID_CLASSPATH}

java \
    -Ddruid.metadata.storage.type=mysql \
    -Ddruid.metadata.storage.connector.connectURI=jdbc\:mysql\://localhost\:3306/druid \
    -Ddruid.metadata.storage.connector.user=${DRUID_MYSQL_USER} \
    -Ddruid.metadata.storage.connector.password=${DRUID_MYSQL_PSWD} \
    -Ddruid.extensions.loadList=[\"mysql-metadata-storage\",\"druid-s3-extensions\"] \
    -Ddruid.storage.type=s3 \
    -Ddruid.s3.accessKey=${HADOOP_AWS_ACCESS_KEY_ID} \
    -Ddruid.s3.secretKey=${HADOOP_AWS_SECRET_ACCESS_KEY} \
    -Ddruid.storage.bucket=${S3_BUCKET} \
    -Ddruid.storage.baseKey=${S3_PATH} \
    -Ddruid.storage.maxListingLength=1000 \
    -cp . io.druid.cli.Main tools insert-segment-to-db --workingDir ${S3_PATH} --updateDescriptor false