Skip to content

Instantly share code, notes, and snippets.

@spuder
Created November 20, 2015 17:03
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save spuder/6e7a69f7f9d98c0548c7 to your computer and use it in GitHub Desktop.
Save spuder/6e7a69f7f9d98c0548c7 to your computer and use it in GitHub Desktop.
Elasticsearch init scirpts
#!/usr/bin/env bash
#
# elasticsearch
#
### BEGIN INIT INFO
# Provides: elasticsearch
# Required-Start: $network $remote_fs $named
# Required-Stop: $network $remote_fs $named
# Default-Start: 2 3 4 5
# Default-Stop: 0 1 6
# Short-Description: Starts elasticsearch
# Description: Starts elasticsearch using start-stop-daemon
### END INIT INFO
# Source networking configuration
#
if [ -f /etc/sysconfig/network ]; then source /etc/sysconfig/network; fi
# Set limits for environments ignoring `/etc/security/limits.d`
#
ulimit -n 64000
ulimit -l unlimited
# Exit if networking is not up
#
[ "$NETWORKING" = "no" ] && exit
PIDFILE='/usr/local/var/run/default_ubuntu_1404.pid'
ES_INCLUDE='/usr/local/etc/elasticsearch/elasticsearch.in.sh'
CHECK_PID_RUNNING=$(ps ax | grep 'java' | grep -e "es.pidfile=$PIDFILE" | sed 's/^\s*\([0-9]*\)\s.*/\1/')
start() {
if [ -f $PIDFILE ]; then
# PIDFILE EXISTS -- ES RUNNING?
echo -e "PID file found in $PIDFILE"
es_pid="$(cat $PIDFILE)"
pid_running="$( ps ax | grep 'java' | grep $es_pid )"
if [ ! -z "$pid_running" ] ; then
# EXIT IF ES IS ALREADY RUNNING
echo -e "\033[31;1mPID $es_pid still alive, Elasticsearch already running...\033[0m"
return 1
fi
fi
echo -en "\033[1mStarting Elasticsearch...\033[0m"
touch $PIDFILE && chown elasticsearch $PIDFILE
ES_INCLUDE=$ES_INCLUDE start-stop-daemon --start --background --quiet --pidfile $PIDFILE --chuid elasticsearch --exec /usr/local/bin/elasticsearch -- -p $PIDFILE
if [ $? ]; then
echo -e " \e[32m[OK]\e[0m"
else
echo -e " \e[31m[FAILURE]\e[0m"
fi
return $?
}
stop() {
if [[ -f $PIDFILE ]]; then
echo -n -e "\033[1mStopping elasticsearch...\033[0m"
# REMOVE PIDFILE AND EXIT IF PROCESS NOT RUNNING
if [ ! $CHECK_PID_RUNNING ]; then
echo -e "\033[1mPID file found, but no matching process running?\033[0m"
echo "Removing PID file..."
rm $PIDFILE
exit 0
fi
# KILL PROCESS
pid=$(cat $PIDFILE)
su elasticsearch -m -c "kill $(cat $PIDFILE)"
r=$?
# Check for process
timeout=0
while [ $(ps -p $pid | wc -l ) -gt '1' ]; do
echo -n '.'
(( timeout ++))
if [ $timeout -gt '30' ]; then return; fi
sleep 1
done
# Check for pidfile
timeout=0
while [ -f $PIDFILE ]; do
echo -n '.'
(( timeout++ ))
if [ $timeout -gt '15' ]; then return; fi
sleep 1
done
echo;
return $r
else
echo -e "\033[1mNo PID file found -- elasticsearch not running?\033[0m"
fi
}
restart() {
stop
timeout=30
while ps aux | grep 'java' | grep -e "es.pidfile"; do
echo -n '.'
(( timeout-- ))
if [ $timeout -lt '1' ]; then return; fi
sleep 1
done;
start
}
status() {
# GOT PIDFILE?
[ -f $PIDFILE ] && pid=$(cat $PIDFILE)
# RUNNING
if [[ $pid && -d "/proc/$pid" ]]; then
version=$(curl -s 'http://localhost:9200' | ruby -rubygems -e 'require "json"; print JSON.parse(STDIN.read)["version"]["number"]')
echo -e "\033[1;37;46melasticsearch $version running with PID $pid\033[0m"
# VERBOSE
if [[ $pid && $1 == '-v' || $1 == '--verbose' ]]; then
curl -s 'http://localhost:9200/_cluster/nodes/default-ubuntu-1404?os&process&jvm&network&transport&settings&pretty' | \
ruby -rubygems -e '
begin
require "json"; h = JSON.parse(STDIN.read); id, node = h["nodes"].first;
def e(name, value); puts %Q|\e[1;36m#{(name.to_s+":").ljust(20)}\e[0m #{value || "N/A" rescue "N/A"}|; end
e "HTTP Address", node["http_address"]
e "Node Name", node["name"]
e "Cluster Name", h["cluster_name"]
e "Started", Time.at(node["jvm"]["start_time"].to_i/1000)
e "JVM", "#{node["jvm"]["vm_name"]} (#{node["jvm"]["version"]})"
e "Memory Total", node["os"]["mem"]["total"]
e "Open Files", node["process"]["max_file_descriptors"]
e "Configuration", node["settings"]["config"]
rescue
puts "Metadata cannot be retrieved."
end
'
fi
# INCORRECT PID?
if [ $pid != $CHECK_PID_RUNNING ]; then
echo -e "\033[1;31;40m[!] Incorrect PID found in $PIDFILE: $pid\033[0m"
return 1
fi
return 0
fi
# NOT RUNNING
if [[ ! $pid || ! -d "/proc/$pid" ]]; then
echo -e "\033[1;33;40melasticsearch not running\033[0m"
return 3
fi
# STALE PID FOUND
if [[ ! -d "/proc/$pid" && -f $PIDFILE ]]; then
echo -e "\033[1;31;40m[!] Stale PID found in $PIDFILE\033[0m"
return 1
fi
}
case "$1" in
start)
start
;;
stop)
stop
;;
restart)
restart
;;
status)
status $2
;;
*)
echo $"Usage: $0 {start|stop|restart|status [-v]|}"
exit 1
esac
exit $?
#!/bin/sh
#
# /etc/init.d/elasticsearch -- startup script for Elasticsearch
#
# Written by Miquel van Smoorenburg <miquels@cistron.nl>.
# Modified for Debian GNU/Linux by Ian Murdock <imurdock@gnu.ai.mit.edu>.
# Modified for Tomcat by Stefan Gybas <sgybas@debian.org>.
# Modified for Tomcat6 by Thierry Carrez <thierry.carrez@ubuntu.com>.
# Additional improvements by Jason Brittain <jason.brittain@mulesoft.com>.
# Modified by Nicolas Huray for Elasticsearch <nicolas.huray@gmail.com>.
#
### BEGIN INIT INFO
# Provides: elasticsearch
# Required-Start: $network $remote_fs $named
# Required-Stop: $network $remote_fs $named
# Default-Start: 2 3 4 5
# Default-Stop: 0 1 6
# Short-Description: Starts elasticsearch
# Description: Starts elasticsearch using start-stop-daemon
### END INIT INFO
PATH=/bin:/usr/bin:/sbin:/usr/sbin
NAME=elasticsearch
DESC="Elasticsearch Server"
DEFAULT=/etc/default/$NAME
if [ `id -u` -ne 0 ]; then
echo "You need root privileges to run this script"
exit 1
fi
. /lib/lsb/init-functions
if [ -r /etc/default/rcS ]; then
. /etc/default/rcS
fi
# The following variables can be overwritten in $DEFAULT
# Run Elasticsearch as this user ID and group ID
ES_USER=elasticsearch
ES_GROUP=elasticsearch
# The first existing directory is used for JAVA_HOME (if JAVA_HOME is not defined in $DEFAULT)
JDK_DIRS="/usr/lib/jvm/java-8-oracle/ /usr/lib/jvm/j2sdk1.8-oracle/ /usr/lib/jvm/jdk-7-oracle-x64 /usr/lib/jvm/java-7-oracle /usr/lib/jvm/j2sdk1.7-oracle/ /usr/lib/jvm/java-7-openjdk /usr/lib/jvm/java-7-openjdk-amd64/ /usr/lib/jvm/java-7-openjdk-armhf /usr/lib/jvm/java-7-openjdk-i386/ /usr/lib/jvm/default-java"
# Look for the right JVM to use
for jdir in $JDK_DIRS; do
if [ -r "$jdir/bin/java" -a -z "${JAVA_HOME}" ]; then
JAVA_HOME="$jdir"
fi
done
export JAVA_HOME
# Directory where the Elasticsearch binary distribution resides
ES_HOME=/usr/share/$NAME
# Heap size defaults to 256m min, 1g max
# Set ES_HEAP_SIZE to 50% of available RAM, but no more than 31g
#ES_HEAP_SIZE=2g
# Heap new generation
#ES_HEAP_NEWSIZE=
# max direct memory
#ES_DIRECT_SIZE=
# Additional Java OPTS
#ES_JAVA_OPTS=
# Maximum number of open files
MAX_OPEN_FILES=65535
# Maximum amount of locked memory
#MAX_LOCKED_MEMORY=
# Elasticsearch log directory
LOG_DIR=/var/log/$NAME
# Elasticsearch data directory
DATA_DIR=/var/lib/$NAME
# Elasticsearch configuration directory
CONF_DIR=/etc/$NAME
# Maximum number of VMA (Virtual Memory Areas) a process can own
MAX_MAP_COUNT=262144
# Path to the GC log file
#ES_GC_LOG_FILE=/var/log/elasticsearch/gc.log
# Elasticsearch PID file directory
PID_DIR="/var/run/elasticsearch"
# End of variables that can be overwritten in $DEFAULT
# overwrite settings from default file
if [ -f "$DEFAULT" ]; then
. "$DEFAULT"
fi
# CONF_FILE setting was removed
if [ ! -z "$CONF_FILE" ]; then
echo "CONF_FILE setting is no longer supported. elasticsearch.yml must be placed in the config directory and cannot be renamed."
exit 1
fi
# Define other required variables
PID_FILE="$PID_DIR/$NAME.pid"
DAEMON=$ES_HOME/bin/elasticsearch
DAEMON_OPTS="-d -p $PID_FILE --default.path.home=$ES_HOME --default.path.logs=$LOG_DIR --default.path.data=$DATA_DIR --default.path.conf=$CONF_DIR"
export ES_HEAP_SIZE
export ES_HEAP_NEWSIZE
export ES_DIRECT_SIZE
export ES_JAVA_OPTS
export ES_GC_LOG_FILE
# Check DAEMON exists
test -x $DAEMON || exit 0
checkJava() {
if [ -x "$JAVA_HOME/bin/java" ]; then
JAVA="$JAVA_HOME/bin/java"
else
JAVA=`which java`
fi
if [ ! -x "$JAVA" ]; then
echo "Could not find any executable java binary. Please install java in your PATH or set JAVA_HOME"
exit 1
fi
}
case "$1" in
start)
checkJava
if [ -n "$MAX_LOCKED_MEMORY" -a -z "$ES_HEAP_SIZE" ]; then
log_failure_msg "MAX_LOCKED_MEMORY is set - ES_HEAP_SIZE must also be set"
exit 1
fi
log_daemon_msg "Starting $DESC"
pid=`pidofproc -p $PID_FILE elasticsearch`
if [ -n "$pid" ] ; then
log_begin_msg "Already running."
log_end_msg 0
exit 0
fi
# Prepare environment
mkdir -p "$LOG_DIR" "$DATA_DIR" && chown "$ES_USER":"$ES_GROUP" "$LOG_DIR" "$DATA_DIR"
# Ensure that the PID_DIR exists (it is cleaned at OS startup time)
if [ -n "$PID_DIR" ] && [ ! -e "$PID_DIR" ]; then
mkdir -p "$PID_DIR" && chown "$ES_USER":"$ES_GROUP" "$PID_DIR"
fi
if [ -n "$PID_FILE" ] && [ ! -e "$PID_FILE" ]; then
touch "$PID_FILE" && chown "$ES_USER":"$ES_GROUP" "$PID_FILE"
fi
if [ -n "$MAX_OPEN_FILES" ]; then
ulimit -n $MAX_OPEN_FILES
fi
if [ -n "$MAX_LOCKED_MEMORY" ]; then
ulimit -l $MAX_LOCKED_MEMORY
fi
if [ -n "$MAX_MAP_COUNT" -a -f /proc/sys/vm/max_map_count ]; then
sysctl -q -w vm.max_map_count=$MAX_MAP_COUNT
fi
# Start Daemon
start-stop-daemon -d $ES_HOME --start -b --user "$ES_USER" -c "$ES_USER" --pidfile "$PID_FILE" --exec $DAEMON -- $DAEMON_OPTS
return=$?
if [ $return -eq 0 ]; then
i=0
timeout=10
# Wait for the process to be properly started before exiting
until { cat "$PID_FILE" | xargs kill -0; } >/dev/null 2>&1
do
sleep 1
i=$(($i + 1))
if [ $i -gt $timeout ]; then
log_end_msg 1
exit 1
fi
done
fi
log_end_msg $return
exit $return
;;
stop)
log_daemon_msg "Stopping $DESC"
if [ -f "$PID_FILE" ]; then
start-stop-daemon --stop --pidfile "$PID_FILE" \
--user "$ES_USER" \
--quiet \
--retry forever/TERM/20 > /dev/null
if [ $? -eq 1 ]; then
log_progress_msg "$DESC is not running but pid file exists, cleaning up"
elif [ $? -eq 3 ]; then
PID="`cat $PID_FILE`"
log_failure_msg "Failed to stop $DESC (pid $PID)"
exit 1
fi
rm -f "$PID_FILE"
else
log_progress_msg "(not running)"
fi
log_end_msg 0
;;
status)
status_of_proc -p $PID_FILE elasticsearch elasticsearch && exit 0 || exit $?
;;
restart|force-reload)
if [ -f "$PID_FILE" ]; then
$0 stop
sleep 1
fi
$0 start
;;
*)
log_success_msg "Usage: $0 {start|stop|restart|force-reload|status}"
exit 1
;;
esac
exit 0
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment