#!/usr/bin/env bash

CURRENT_DIR=`pwd`

# Read the version from version.sbt
SPARK_LUCENERDD_VERSION=`cat version.sbt | awk '{print $5}' | xargs`

# You should have downloaded this spark version under your ${HOME}
SPARK_VERSION="3.2.0"

echo "==============================================="
echo "Loading LuceneRDD with version ${SPARK_LUCENERDD_VERSION}"
echo "==============================================="

echo "==============================================="
echo "SPARK version: ${SPARK_VERSION}"
echo "==============================================="

# Assumes that spark is installed under home directory
HOME_DIR=`echo ~`
#export SPARK_LOCAL_IP=localhost
SPARK_HOME=${HOME_DIR}/spark-${SPARK_VERSION}-bin-hadoop3.2

# spark-lucenerdd assembly JAR
MAIN_JAR=${CURRENT_DIR}/target/scala-2.12/spark-lucenerdd-assembly-${SPARK_LUCENERDD_VERSION}.jar

# Run spark shell locally
${SPARK_HOME}/bin/spark-shell   --jars "${MAIN_JAR}" \
				--conf "spark.executor.memory=1g" \
				--conf "spark.driver.memory=1g" \
				--conf "spark.rdd.compress=true" \
				--conf "spark.serializer=org.apache.spark.serializer.KryoSerializer" \
				--conf "spark.kryo.registrator=org.zouzias.spark.lucenerdd.LuceneRDDKryoRegistrator" \
				--conf spark.executor.extraJavaOptions="-Dlucenerdd.index.store.mode=disk" \
				--conf spark.driver.extraJavaOptions="-Dlucenerdd.index.store.mode=disk" \
				--conf "spark.kryoserializer.buffer=24mb" \
				--master local[*]