#!/bin/sh

# 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.

DESC="Cassandra Stress Test Daemon"
if [ "x$CASSANDRA_INCLUDE" = "x" ]; then 
    for include in "`dirname $0`/cassandra.in.sh" \
                   "$HOME/.cassandra.in.sh" \
                   /usr/share/cassandra/cassandra.in.sh \
                   /usr/local/share/cassandra/cassandra.in.sh \
                   /opt/cassandra/cassandra.in.sh; do
        if [ -r $include ]; then 
            . $include
            break   
        fi      
    done    
elif [ -r $CASSANDRA_INCLUDE ]; then 
    . $CASSANDRA_INCLUDE
fi

if [ -x $JAVA_HOME/bin/java ]; then 
    JAVA=$JAVA_HOME/bin/java
else
    JAVA=`which java`
fi

if [ -x $JAVA_HOME/bin/java ]; then
    JAVA=$JAVA_HOME/bin/java
else
    JAVA=`which java`
fi

if [ "x$JAVA" = "x" ]; then
    echo "Java executable not found (hint: set JAVA_HOME)" >&2
    exit 1
fi

case "$1" in
  start)
    echo "Starting $DESC: "
    $JAVA -server -cp $CLASSPATH \
          -Dcassandra.storagedir="$cassandra_storagedir" \
          org.apache.cassandra.stress.StressServer $@ 1> ./stressd.out.log 2> ./stressd.err.log &
    echo $! > ./stressd.pid
    echo "done."
  ;;
  
  stop)
    PID=`cat ./stressd.pid 2> /dev/null`
    
    if [ "x$PID" = "x" ]; then
      echo "$DESC is not running."
    else
      kill -9 $PID
      rm ./stressd.pid
      echo "$DESC is stopped."
    fi
  ;;

  status)
    PID=`cat ./stressd.pid 2> /dev/null`

    if [ "x$PID" = "x" ]; then
      echo "$DESC is not running."
    else
      echo "$DESC is running with pid $PID."
    fi
  ;;

  *)
    echo "Usage: $0 start|stop|status [-h <host>]"
  ;;
esac

