#!/bin/bash
#
#
# 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.
#
#

user=""
secure="false"
keytab=""
while getopts ":u:k:s" opt; do
  case $opt in
    u)
      user=$OPTARG;
      ;;
    k)
      keytab=$OPTARG;
      ;;
    s)
      secure="true";
      ;;
    \?)
      echo "Invalid option: -$OPTARG" >&2
      exit 3
      ;;
    :)
      echo "UNKNOWNOption -$OPTARG requires an argument." >&2
      exit 3
      ;;
  esac
done

outfile="/tmp/nagios-hadoop-check.out"
curtime=`date +"%F-%H-%M-%S"`
fname="nagios-hadoop-check-${curtime}"

if [[ "$user" == "" ]]; then
  echo "INVALID: user argument not specified";
  exit 3;
fi
if [[ "$keytab" == "" ]]; then 
  keytab="/homes/$user/$user.headless.keytab"
fi

if [[ "$secure" == "true" ]]; then
  sudo -u $user -i "/usr/kerberos/bin/kinit -kt $keytab $user" > ${outfile} 2>&1
fi

sudo -u $user -i "hadoop dfs -copyFromLocal /etc/passwd ${fname}.input " > ${outfile} 2>&1
if [[ "$?" -ne "0" ]]; then 
  echo "CRITICAL: Error copying file to HDFS. See error output in ${outfile} on nagios server";
  exit 2; 
fi
sudo -u $user -i "hadoop dfs -ls" > ${outfile} 2>&1
if [[ "$?" -ne "0" ]]; then 
  echo "CRITICAL: Error listing HDFS files. See error output in ${outfile} on nagios server";
  exit 2; 
fi
sudo -u $user -i "hadoop jar /usr/share/hadoop/hadoop-examples-*.jar wordcount ${fname}.input ${fname}.out" >> ${outfile} 2>&1
if [[ "$?" -ne "0" ]]; then 
  echo "CRITICAL: Error running M/R job. See error output in ${outfile} on nagios server";
  exit 2; 
fi
sudo -u $user -i "hadoop fs -rmr -skipTrash ${fname}.out" >> ${outfile} 2>&1
if [[ "$?" -ne "0" ]]; then 
  echo "CRITICAL: Error removing M/R job output. See error output in ${outfile} on nagios server";
  exit 2; 
fi
sudo -u $user -i "hadoop fs -rm -skipTrash ${fname}.input" >> ${outfile} 2>&1
if [[ "$?" -ne "0" ]]; then 
  echo "CRITICAL: Error removing M/R job input. See error output in ${outfile} on nagios server";
  exit 2; 
fi

echo "OK: M/R WordCount Job ran successfully"
exit 0;