fixes

qubole · Mar 12, 2018 · 510a91f · 510a91f
1 parent d5fa01a
commit 510a91f
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/hadoop/util.sh b/hadoop/util.sh
@@ -1,6 +1,8 @@
 #!/bin/bash
 
 source /usr/lib/hustler/bin/qubole-bash-lib.sh
+export PROFILE_FILE=${PROFILE_FILE:-/etc/profile}
+export HADOOP_ETC_DIR=${HADOOP_ETC_DIR:-/etc/hadoop}
 
 function restart_master_services() {
   monit restart resourcemanager
@@ -14,12 +16,12 @@ function restart_worker_services() {
 function use_java8() {
  export JAVA_HOME=/usr/lib/jvm/java-1.8.0_60
  export PATH=$JAVA_HOME/bin:$PATH
- echo "export JAVA_HOME=/usr/lib/jvm/java-1.8.0_60" >> /etc/profile
- echo "export PATH=$JAVA_HOME/bin:$PATH" >> /etc/profile
+ echo "export JAVA_HOME=/usr/lib/jvm/java-1.8.0_60" >> "$PROFILE_FILE"
+ echo "export PATH=$JAVA_HOME/bin:$PATH" >> "$PROFILE_FILE"
 
- sed -i 's/java-1.7.0/java-1.8.0_60/' /etc/hadoop/hadoop-env.sh
- sed -i 's/java-1.7.0/java-1.8.0_60/' /etc/hadoop/mapred-env.sh
- sed -i 's/java-1.7.0/java-1.8.0_60/' /etc/hadoop/yarn-env.sh
+ sed -i 's/java-1.7.0/java-1.8.0_60/' "$HADOOP_ETC_DIR/hadoop-env.sh"
+ sed -i 's/java-1.7.0/java-1.8.0_60/' "$HADOOP_ETC_DIR/mapred-env.sh"
+ sed -i 's/java-1.7.0/java-1.8.0_60/' "$HADOOP_ETC_DIR/yarn-env.sh"
 
  is_master=$(nodeinfo is_master)
  if [[ "$is_master" == "1" ]]; then