flink-init.j2 1.31 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14
#!/bin/bash

# The path where Apache Flink is installed.
INSTALLATION_PATH="{{ installation_path }}"

# The full path of the pid file to use.
PIDFILE="$INSTALLATION_PATH/flink/flink.pid"

# The full path of the lock file to use.
LOCKFILE="$INSTALLATION_PATH/flink/flink-lock"

# The command that will start Apache Flink.
START_COMMAND="$INSTALLATION_PATH/flink/bin/yarn-session.sh -n {{ number_of_taskmanagers }} -tm {{ ram_per_task_manager }}"

15 16 17
# The command that will stop Apache Flink. Note that, HADOOP_HOME home variable should be set outside from this script and
# before Flink's deployment.
STOP_COMMAND="$HADOOP_HOME/bin/yarn application --kill"
18

19
start(){
20
  nohup $START_COMMAND &
21 22 23 24 25 26
  RETVAL=$?
  [ $RETVAL -eq 0 ] && touch $LOCKFILE
  return $RETVAL
}

stop(){
27 28
  id=$($HADOOP_HOME/bin/yarn application --list | grep "Flink session" | cut -f1)
  nohup $STOP_COMMAND $id &
29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64
  RETVAL=$?
  [ $RETVAL -eq 0 ] && $(rm -f $LOCKFILE)
  return $RETVAL
}

restart(){
  stop
  start
}

RETVAL=0

case "$1" in
  start)
    start
    ;;
  stop)
    stop
    ;;
  restart|reload|force-reload)
    restart
    ;;
  condrestart)
    [ -f $LOCKFILE ] && restart || :
    ;;
  status)
    ps -p $(cat $PIDFILE)
    RETVAL=$?
    ;;
  *)
    echo "Usage: $0 {start|stop|status|restart|reload|force-reload|condrestart}"
    RETVAL=1
esac

exit $RETVAL