Add nodectl-instance

1 files changed, 191 insertions, 0 deletions

diff --git a/node-admin/include/nodectl-instance.sh b/node-admin/include/nodectl-instance.sh
new file mode 100644
index 00000000000..b3d3dbea440
--- /dev/null
+++ b/node-admin/include/nodectl-instance.sh
@@ -0,0 +1,191 @@
+#!/bin/sh
+# Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+
+# WARNING: This script should be kept in sync with the file used by Chef:
+#     vespa-cookbooks/hosted/files/default/prepost-instance.sh
+# TODO: Remove the above cookbook file (with the down-side that a new script
+# requires a new vespa release, instead of just a hosted release).
+#
+# Usage: nodectl-instance.sh [resume|start|stop|suspend]
+#
+# resume: Set the node "in service" by e.g. undraining container traffic
+#
+# start: Start services on the node. Can be seen as a boot of a non-Docker node.
+#        start can be assumed to have completed successfully.
+#
+# stop: Stop services on the node. Can be seen as a shutdown of a non-Docker node.
+#
+# suspend: Prepare for a short suspension, e.g. there's a pending upgrade. Set the
+# node "out of service" by draining container traffic, and flush index for a
+# quick start after the suspension. There's no need to stop.
+
+# BEGIN environment bootstrap section
+# Do not edit between here and END as this section should stay identical in all scripts
+
+function debug {
+    echo "$(date -Is) $*"
+}
+
+findpath () {
+    myname=${0}
+    mypath=${myname%/*}
+    myname=${myname##*/}
+    if [ "$mypath" ] && [ -d "$mypath" ]; then
+        return
+    fi
+    mypath=$(pwd)
+    if [ -f "${mypath}/${myname}" ]; then
+        return
+    fi
+    echo "FATAL: Could not figure out the path where $myname lives from $0"
+    exit 1
+}
+
+COMMON_ENV=libexec/vespa/common-env.sh
+
+source_common_env () {
+    if [ "$VESPA_HOME" ] && [ -d "$VESPA_HOME" ]; then
+        # ensure it ends with "/" :
+        VESPA_HOME=${VESPA_HOME%/}/
+        export VESPA_HOME
+        common_env=$VESPA_HOME/$COMMON_ENV
+        if [ -f "$common_env" ]; then
+            . $common_env
+            return
+        fi
+    fi
+    return 1
+}
+
+findroot () {
+    source_common_env && return
+    if [ "$VESPA_HOME" ]; then
+        echo "FATAL: bad VESPA_HOME value '$VESPA_HOME'"
+        exit 1
+    fi
+    if [ "$ROOT" ] && [ -d "$ROOT" ]; then
+        VESPA_HOME="$ROOT"
+        source_common_env && return
+    fi
+    findpath
+    while [ "$mypath" ]; do
+        VESPA_HOME=${mypath}
+        source_common_env && return
+        mypath=${mypath%/*}
+    done
+    echo "FATAL: missing VESPA_HOME environment variable"
+    echo "Could not locate $COMMON_ENV anywhere"
+    exit 1
+}
+
+findroot
+
+# END environment bootstrap section
+
+has_servicename() {
+    local name="$1"
+    $VESPA_HOME/bin/vespa-model-inspect host $(hostname) | grep -q "$name @ "
+    return $?
+}
+
+has_container() {
+    has_servicename container || has_servicename qrserver
+}
+
+has_searchnode() {
+    has_servicename searchnode
+}
+
+container_drain() {
+    # TODO: Implement proper draining
+    sleep 60
+}
+
+Resume() {
+    # Always start vip for now
+    $echo $VESPA_HOME/bin/vespa-routing vip -u node-admin in
+
+    # Start filebeat if configured
+    if [ -f /etc/filebeat/filebeat.yml ] ; then
+        $echo service filebeat start
+    fi
+}
+
+# Start all services, can be seen as a reboot of a non-Docker node
+Start() {
+    debug "Configuring rsyslog service to work"
+    # Disable kernel log module
+    sed -i.bak 's/^\$ModLoad imklog/#$ModLoad imklog/' /etc/rsyslog.conf
+    debug "Starting rsyslog service"
+    service rsyslog start
+
+    debug "Starting crond service"
+    service crond start
+
+    debug "Yinst settings"
+    yinst set zpe_policy_updater.domains=$ATHENS_DOMAIN \
+        zpe_policy_updater.autostart=on \
+        zpe_policy_updater.cron_start_hour=* \
+        zpe_policy_updater.cron_start_min=*/30 \
+        zpe_policy_updater.cron_start_delay=30
+    debug "Starting all yinst packages"
+    # Start yinst the way it is done when a non-Docker node is booted.
+    # As this is implemented in yinst now (2017-02-08), this will take care of
+    # cleaning up /home/y/tmp and /home/y/var/run
+    /etc/rc.d/init.d/yinst start
+    debug "yinst started, exited with $?"
+}
+
+# Stop all services, can be seen as a shutdown of a non-Docker node
+Stop() {
+    debug "Stopping services and other yinst packages running"
+    # Stop yinst the way it is done when a non-Docker node is shutdown.
+    /etc/rc.d/init.d/yinst stop
+    debug "Stopping crond service"
+    service crond stop
+    debug "Stopping rsyslog service"
+    service rsyslog stop
+}
+
+Suspend() {
+    # Always stop vip for now
+    $echo $VESPA_HOME/bin/vespa-routing vip -u node-admin out
+
+    if has_searchnode; then
+        $echo $VESPA_HOME/bin/vespa-proton-cmd --local prepareRestart
+    fi
+
+    if has_container; then
+        $echo container_drain
+    fi
+}
+
+main() {
+    if [ $# -lt 1 ]; then
+        echo "Usage: $0 [-e] start|stop|suspend" >&2
+        exit 1
+    fi
+
+    echo=""
+    if [ "$1" = "-e" ]; then
+        echo=echo
+        shift
+    fi
+
+    action="$1"
+
+    if [ "$action" = "start" ]; then
+        Start
+    elif [ "$action" = "stop" ]; then
+        Stop
+    elif [ "$action" = "suspend" ]; then
+        Suspend
+    elif [ "$action" = "resume" ]; then
+        Resume
+    else
+        echo "Unknown action: $action" >&2
+        exit 1
+    fi
+}
+
+main "$@"
\ No newline at end of file