#!/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="Stress Test Daemon"

if [ "x$CLASSPATH" = "x" ]; then
    # Cassandra class files.
    if [ ! -d `dirname $0`/../../../build/classes/main ]; then
        echo "Unable to locate cassandra class files" >&2
        exit 1
    fi

    # Stress class files.
    if [ ! -d `dirname $0`/../build/classes ]; then
        echo "Unable to locate stress class files" >&2
        exit 1
    fi

    CLASSPATH=`dirname $0`/../../../build/classes/main
    CLASSPATH=$CLASSPATH:`dirname $0`/../../../build/classes/thrift
    CLASSPATH=$CLASSPATH:`dirname $0`/../build/classes
    for jar in `dirname $0`/../../../lib/*.jar; do
        CLASSPATH=$CLASSPATH:$jar
    done
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 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