eagle-assembly/src/main/bin/eagle-topology.sh

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

source $(dirname $0)/eagle-env.sh

TOPOLOGY_NAME_SET=0

function print_help() {
	echo "Usage: $0 options {start | stop | status}"
	echo "Options:                       Description:"
	echo "  --main     <main class>        Default is org.apache.eagle.security.auditlog.HdfsAuditLogProcessorMain"
	echo "  --topology <topology name>     Default is sandbox-hdfsAuditLog-topology"
	echo "  --config   <file path>         Default is $EAGLE_HOME/conf/sandbox-hdfsAuditLog-application.conf"
	echo "  --storm-ui <storm ui url>      Execute through storm UI API, default: http://localhost:8744"

	echo "Command Examples:"
	echo "  $0 --main <mainClass> [--topology <topologyName>] --config <filePath> start"
	echo "  $0 --topology <topologyName> stop"
	echo "  $0 --topology <topologyName> status"
}

function env_check(){
    which storm >/dev/null 2>&1
    if [ $? != 0 ];then
        echo "Error: storm is not installed"
        exit 1
    fi
}

#### parameters are in pair plus command
#if [ `expr $# % 2` != 1 ]
#then
#    print_help
#    exit 1
#fi

cmd=""
while [  $# -gt 0  ]; do
case $1 in
    "start")
        cmd=$1
        shift
        ;;
    "stop")
        cmd=$1
        shift
        ;;
    "status")
        cmd=$1
        shift
        ;;
    --main)
        if [ $# -lt 3 ]; then
            print_help
            exit 1
        fi
        mainClass=$2
        shift 2
        ;;
    --topology)
        if [ $# -lt 3 ]; then
            print_help
            exit 1
        fi
        TOPOLOGY_NAME_SET=1
        topologyName=$2
        shift 2
        ;;
    --config)
        if [ $# -lt 3 ]; then
            print_help
            exit 1
        fi
        configFile=$2
        shift 2
        ;;
	--storm-ui)
		# TODO: configure through arguments
		storm_ui="http://localhost:8744"
		shift 1
		;;
    *)
        echo "Internal Error: option processing error: $1" 1>&2
        exit 1
        ;;
    esac
done


jarName=$(ls ${EAGLE_HOME}/lib/topology/eagle-topology-*-assembly.jar)

if [ -z "$jarName" ]; then
    echo "Error: jar file is not found"
    exit 1
fi

if [ -z "$mainClass" ]; then
    mainClass="org.apache.eagle.security.auditlog.HdfsAuditLogProcessorMain"
fi

if [ -z "$topologyName" ]; then
    topologyName=sandbox-hdfsAuditLog-topology
fi

if [ -z "$configFile" ]; then
    configFile="$EAGLE_HOME/conf/sandbox-hdfsAuditLog-application.conf"
fi

case $cmd in
"start")
    env_check
    echo "Starting eagle topology ..."
    echo "jarName="$jarName "mainClass="$mainClass "configFile="$configFile
    if [ $TOPOLOGY_NAME_SET = 1 ]; then
	    storm jar -c nimbus.host=$EAGLE_NIMBUS_HOST ${jarName} $mainClass -D config.file=$configFile -D envContextConfig.topologyName=$topologyName
	else
	    storm jar -c nimbus.host=$EAGLE_NIMBUS_HOST ${jarName} $mainClass -D config.file=$configFile
	fi
	if [ $? = 0 ]; then
		echo "Starting is completed"
		exit 0
	else
		echo "Failure"
		exit 1
	fi
	;;
"stop")
    env_check
    echo "Stopping eagle topology ..."
    storm kill -c nimbus.host=$EAGLE_NIMBUS_HOST $topologyName
    if [ $? = 0 ]; then
    	echo "Stopping is completed"
	    exit 0
    else
    	echo "Failure"
	exit 1
    fi
	;;
"status")
	if [ -z "$storm_ui" ];then
	    env_check
	    echo "Checking topology $topologyName status ..."
	    output=`storm list  -c nimbus.host=$EAGLE_NIMBUS_HOST | grep $topologyName`
	    if [ $? != 0 ];then
	        echo "Topology is not alive: $topologyName is not found"
	        exit 1
	    fi

	    echo $output | grep ACTIVE > /dev/null 2>&1

	    if [ $? = 0 ];then
	        echo "Topology is alive: $output"
	        exit 0
	    else
	        echo "Topology is not alive: $output"
	        exit 1
	    fi
    else
	    echo "Checking topology $topologyName status through storm UI API on $storm_ui"
	    curl -XGET $storm_ui/api/v1/topology/summary | grep $topologyName | grep ACTIVE >/dev/null 2>&1
	    if [ $? == 0 ];then
	        echo "$topologyName is running"
	        exit 0
	    else
	        echo "$topologyName is dead"
	        exit 1
	    fi
    fi
    ;;
*)
	print_help
	exit 1
esac

exit 0