summaryrefslogtreecommitdiffstats
path: root/node-admin/include/nodectl-instance.sh
blob: a1018d6ba121cc806a52fe264137ed773a411e29 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
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 "$@"