Riaknostic section

First commit

rel/files/riak-admin

@@ -65,6 +65,15 @@ ERTS_PATH=$RUNNER_BASE_DIR/erts-$ERTS_VSN/bin
 # Setup command to control the node
 NODETOOL="$ERTS_PATH/escript $ERTS_PATH/nodetool $NAME_ARG $COOKIE_ARG"
 
+# Riaknostic location
+RIAKNOSTIC_LOC="$ERTS_PATH/../lib/riaknostic/riaknostic"
+
+# Setup command to run riaknostic
+RIAKNOSTIC="$ERTS_PATH/escript $RIAKNOSTIC_LOC -etc $RUNNER_ETC_DIR -base $RUNNER_BASE_DIR -user $RUNNER_USER"
+
+# URL for Riaknostic download instructions
+RIAKNOSTIC_URL="http://riaknostic.basho.com/"
+
 # Check the first argument for instructions
 case "$1" in
     join)
@@ -385,6 +394,18 @@ case "$1" in
                                _  -> init:stop(1)                             \
                               end."
 
+        ;;
+    diag)
+        shift
+
+        if [ -f "$RIAKNOSTIC_LOC" ]; then
+            $RIAKNOSTIC "$@"
+        else
+            echo "Riak diagnostics utility is not present!"
+            echo "Visit $RIAKNOSTIC_URL for instructions."
+            exit 1
+        fi
+
         ;;
     *)
         echo "Usage: $SCRIPT { join | leave | backup | restore | test | status | "