summaryrefslogtreecommitdiffstats
path: root/tools/helm-healer.sh
blob: 47479a9bc1f80f15385ee27fd9b90c37f5b3034e (plain)
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
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
#!/bin/bash

PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

#
# globals and defaults
#

NAMESPACE=
OVERRIDES=
HELM_CHART_RELEASE_NAME=
HELM_DELETE_ALL=
HELM_SKIP_DEPLOY=
VOLUME_STORAGE=
HELM_TIMEOUT=3600
RELEASE_PREFIX=onap

#
# control variables
#

CMD=$(basename "$0")
COLOR_ON_RED='\033[0;31;1m'
COLOR_ON_GREEN='\033[0;32;1m'
COLOR_OFF='\033[0m'


#
# functions
#

help()
{
cat <<EOF
${CMD} - simple tool for fixing onap helm deployment

DESCRIPTION
    This script does nothing smart or special it just tries to
    redeploy onap component. It can fix only problems related to
    race conditions or timeouts. Nothing else. It will not fix
    broken ONAP - there is no such ambition - that effort should
    be directed in the upstream.

USAGE
    ${CMD} -h|--help
        This help

    ${CMD} -n|--namespace <namespace>
           (-f|--file <override>)...
           (-s|--storage <directory>)|--no-storage-deletion
           [-p|--release-prefix <release prefix>]
           [-t|--timeout <secs>]
           [(-c|--component <component release name>)...|
            (-D|--delete-all)]
           [-C|--clean-only]

EXAMPLES

    Usage 1: (simple heuristics - redeploy failed components):
        ${CMD} -n onap -f /some/override1.yml -s /dockerdata-nfs

    Usage 2: (redeploy ONLY explicitly listed components):
        ${CMD} -n onap -f /some/override1.yml -s /dockerdata-nfs \\
               -c onap-aaf -c onap-sdc -c onap-portal

    Usage 3: (delete EVERYTHING and redeploy):
        ${CMD} -n onap -f /some/override1.yml -s /dockerdata-nfs --delete-all

    Usage 4: (delete EVERYTHING and DO NOT redeploy - clean env.)
        ${CMD} -n onap -s /dockerdata-nfs --delete-all --clean-only

NOTES

    Namespace argument (always) and at least one override file (if you don't
    use '--delete-all') are mandatory for this script to execute. Also you must
    provide path to the storage ('--storage') OR explicitly request to not
    delete file storage of the component ('--no-storage-deletion').

    The storage should be a directory where persistent volume resides. It will
    work only if the component created the persistent volume with the same
    filename as its release name. Otherwise no files are deleted. The exception
    is when '--delete-all' is used - in that case all content of the storage is
    deleted (because ONAP is not consistent with the volume directory names
    - e.g.: sdnc).

    '--file' can be used multiple of times and it is used for override files
    which are passed on to helm. The order is significant because if two
    override files modify one value the latest one is used. This option is
    ignored if '--clean-only' is used.

    CAUTION 1: filename of an override file cannot contain whitespace! This is
    actually helm/onap deploy plugin issue which does not handle such files. So
    I dropped the more complicated version of this script when there is no
    reason to support something on what will helm deploy choke anyway.

    '--prefix' option is helm release argument - it is actually prefix when you
    list the helm releases - helm is little confusing here.

    CAUTION 2: By default release prefix is 'onap' - if you deployed release
    'onap' and now run this script with different prefix then it will skip all
    'onap-*' components and will deploy a new release with new prefix - BEWARE
    TO USE PROPER RELEASE PREFIX!

    Timeout sets the waiting time for helm deploy per component.

    '--component' references to the release name of the chart which you want to
    redeploy excplicitly - otherwise 'ALL FAILED' components will be
    redeployed. You can target more than one component at once - just use the
    argument multiple times.

    Component option is mutually exclusive with the '--delete-all' which will
    delete all components - healthy or not. Actually it will delete the whole
    NAMESPACE and everything in it. Also to be sure it will cleanup all
    orphaned images and volumes on all kubernetes nodes.

    '--clean-only' can be used with any usage: heuristics, explicit component
    list or with '--delete-all'. It basically just skips the last step - the
    actual redeploy.
EOF
}

use_help()
{
    printf "Try help: ${CMD} --help\n"
}

msg()
{
    printf "${COLOR_ON_GREEN}INFO: $@ ${COLOR_OFF}\n"
}

error()
{
    printf "${COLOR_ON_RED}ERROR: $@ ${COLOR_OFF}\n"
}

on_exit()
{
    printf "$COLOR_OFF"
}

# remove all successfully completed jobs
clean_jobs()
{
    kubectl get jobs -n ${NAMESPACE} \
        --ignore-not-found=true \
        --no-headers=true | \
        while read -r _job _completion _duration _age ; do
            _done=$(echo ${_completion} | awk 'BEGIN {FS="/";} {print $1;}')
            _desired=$(echo ${_completion} | awk 'BEGIN {FS="/";} {print $2;}')
            if [ "$_desired" -eq "$_done" ] ; then
                delete_job "$_job"
            fi
        done
}

get_failed_labels()
{
    get_labels 'status.phase==Failed'
}

# arg: [optional: selector]
get_labels()
{
    if [ -n "$1" ] ; then
        _selector="--field-selector=${1}"
    else
        _selector=
    fi

    kubectl get pods -n ${NAMESPACE} \
        --show-labels=true \
        ${_selector} \
        --ignore-not-found=true \
        --no-headers=true | \
        while read -r _pod _ready _status _restart _age _labels ; do
            [ -z "$_labels" ] && break
            for _label in $(echo "$_labels" | tr ',' ' ') ; do
                case "$_label" in
                    release=*)
                        _label=$(echo "$_label" | sed 's/release=//')
                        echo "$_label"
                        ;;
                esac
            done
        done | sort -u
}

# arg: <release name>
helm_undeploy()
{
    msg "Undeploy helm release name: ${1}"
    # Helm v3 does not support "--purge" flag since it's a default behavior for v3
    if [[ $(helm version --template "{{.Version}}") =~ ^v3 ]];then
        helm undeploy ${1}
    else
        helm undeploy ${1} --purge
    fi
    sleep 15s
}

# arg: <job name>
delete_job()
{
    kubectl delete job -n ${NAMESPACE} \
        --cascade=true \
        --now=true \
        --wait=true \
        ${1}

    # wait for job to be deleted
    _output=start
    while [ -n "$_output" ] && sleep 1 ; do
        _output=$(kubectl get pods -n ${NAMESPACE} \
            --ignore-not-found=true \
            --no-headers=true \
            --selector="job-name=${1}")
    done
}

#arg: <component>
get_resources_for_component()
{

helm status $1 | awk -f <(cat - <<-'EOD'
BEGIN {
  work="no"
  kind=""
  a["dummy"]=""
}

$1 ~ ":" {
  if ( $1 == "RESOURCES:" ) {
          work="yes"
} else {
  work="no"
}

}

$1 == "==>" {
  split($2, a, "[/(]")
  kind=a[2]
}

$1 != "NAME" && $1 != "==>" && work == "yes" && $1 !~ ":" && $1 != "" {
  printf "%s/%s\n", kind, $1
}

EOD
)
}

# arg: <resource>
delete_resource()
{
    local _resource="$1"
    local _kind="${_resource%/*}"
    local _name="${_resource#*/}"


    if kubectl get ${_resource} >/dev/null 2>&1; then
        msg "${_resource} has not been removed with helm undeploy, manual removal is required. Proceeding"
        kubectl delete ${_resource} -n ${NAMESPACE} \
            --cascade=true \
            --now=true \
            --wait=true \
            2>&1 | grep -iv 'not[[:space:]]*found'

        # wait for resource to be deleted
        _output=start
        while [ -n "$_output" ] && sleep 1 ; do
            _output=$(kubectl get ${_kind} ${_name} -n ${NAMESPACE} \
                --ignore-not-found=true \
                --no-headers=true )
        done
        msg "Done"
    fi
}

delete_namespace()
{
    msg "Delete the whole namespace: ${NAMESPACE}"
    kubectl delete namespace \
        --cascade=true \
        --now=true \
        --wait=true \
        "$NAMESPACE"

    # wait for namespace to be deleted
    _output=start
    while [ -n "$_output" ] && sleep 1 ; do
        _output=$(kubectl get all -n ${NAMESPACE} \
            --ignore-not-found=true \
            --no-headers=true)
    done
}

delete_persistent_volume()
{
    _persistent_volume=$1
     if kubectl get ${_persistent_volume} >/dev/null 2>&1; then
          msg "${_persistent_volume} has not been removed with helm undeploy, manual removal is required. Proceeding"
          #very often k8s hangs on Terminating state for pv due to  still active pvc. It is better to delete pvc directly
         _claim=$(kubectl get ${_persistent_volume} -o jsonpath='{ .spec.claimRef.name}')
         delete_resource PersistentVolumeClaim/${_claim}
     fi
}

# arg: [optional: directory]
delete_storage()
{
    _node=$(kubectl get nodes \
        --selector=node-role.kubernetes.io/worker \
        -o wide \
        --no-headers=true | \
        awk '{print $6}' | head -n 1)

    if [ -z "$_node" ] ; then
        error "Could not list kubernetes nodes - SKIPPING DELETION"
    else
        if [ -n "$1" ] ; then
            msg "Delete directory '${1}' on $_node"
            ssh $_node "rm -rf '${1}'"
        else
            msg "Delete directories '${VOLUME_STORAGE}/*' on $_node"
            ssh $_node "find '${VOLUME_STORAGE}' -maxdepth 1 -mindepth 1 -exec rm -rf '{}' \;"
        fi
    fi
}

docker_cleanup()
{
    _nodes=$(kubectl get nodes \
        --selector=node-role.kubernetes.io/worker \
        -o wide \
        --no-headers=true | \
        awk '{print $6}')

    if [ -z "$_nodes" ] ; then
        error "Could not list kubernetes nodes - SKIPPING docker cleanup"
        return
    fi

    for _node in $_nodes ; do
        msg "Docker cleanup on $_node"
        ssh $_node "docker system prune --force --all --volumes" >/dev/null &
    done

    msg "We are waiting now for docker cleanup to finish on all nodes..."
    wait
}

is_helm_serve_running()
{
    # healthy result: HTTP/1.1 200 OK
    _helm_serve_result=$(curl -w %{http_code} --silent --connect-timeout 3 http://127.0.0.1:8879/ -o /dev/null)

    if [ "$_helm_serve_result" == "200" ] ; then
        return 0
    else
        return 1
    fi
}

# arg: <release name>
undeploy_component()
{
    local _component=$1

    #Because Helm undeploy is not reliable: Gathering resources assigned to componen to track and remove orphans later
    _component_resources=($(get_resources_for_component ${_component}))

    declare -a _persistent_volumes
    declare -a _standard
    declare -a _unknown_kinds

    for resource in ${_component_resources[@]}; do
        case $resource in
            CronJob/* | Job/* | Secret/* | ConfigMap/* | Pod/* | Service/* | Deployment/* | StatefulSet/*)
                _standard+=(${resource});;
            #Ignoring PVC, they will be handled along with PV as 'helm' status does not return them for some components
            PersistentVolumeClaim/*)
                ;;
            PersistentVolume/*)
                _persistent_volumes+=(${resource});;
            *)
                _unknown_kinds+=(${resource})
        esac
    done


    #Gathering physical location of directories for persistent volumes to delete them after undeploy
    declare -a _physical_locations
    for volume in ${_persistent_volumes[@]}; do
        _physical_locations+=($(kubectl get ${volume} -o jsonpath='{ .spec.hostPath.path}' ))
    done

    helm_undeploy ${_component}

    #Manual items removal
    for resource in ${_standard[@]}; do
        delete_resource ${resource}
    done

    for volume in ${_persistent_volumes[@]}; do
        delete_persistent_volume ${volume}
    done

    for subdir in ${_physical_locations[@]}; do
        delete_storage ${subdir}
    done

    if [ "${#_unknown_kinds[@]}" -ne 0 ] ; then
        for resource in ${_unknown_kinds[@]}; do
            error "Untracked resource kind present: ${resource}, attempting to delete it..."
            delete_resource ${resource}
        done
        return
    fi
}

# arg: <release name>
deploy_component()
{
    # TODO: until I can verify that this does the same for this component as helm deploy
    #msg "Redeployment of the component ${1}..."
    #helm install "local/${_chart}" --name ${1} --namespace ${NAMESPACE} --wait --timeout ${HELM_TIMEOUT}
    error "NOT IMPLEMENTED"
}


#
# arguments
#

state=nil
arg_namespace=
arg_overrides=
arg_timeout=
arg_storage=
arg_nostorage=
arg_components=
arg_prefix=
arg_deleteall=
arg_cleanonly=
while [ -n "$1" ] ; do
    case $state in
        nil)
            case "$1" in
                -h|--help)
                    help
                    exit 0
                    ;;
                -n|--namespace)
                    state=namespace
                    ;;
                -f|--file)
                    state=override
                    ;;
                -t|--timeout)
                    state=timeout
                    ;;
                -s|--storage)
                    state=storage
                    ;;
                --no-storage-deletion)
                    if [ -n "$arg_storage" ] ; then
                        error "Usage of storage argument together with no storage deletion option!"
                        use_help
                        exit 1
                    elif [ -z "$arg_nostorage" ] ; then
                        arg_nostorage=nostorage
                    else
                        error "Duplicit argument for no storage option! (IGNORING)"
                    fi
                    ;;
                -c|--component)
                    if [ -n "$arg_deleteall" ] ; then
                        error "'Delete all components' used already - argument mismatch"
                        use_help
                        exit 1
                    fi
                    state=component
                    ;;
                -D|--delete-all)
                    if [ -n "$arg_components" ] ; then
                        error "Explicit component(s) provided already - argument mismatch"
                        use_help
                        exit 1
                    elif [ -z "$arg_deleteall" ] ; then
                        arg_deleteall=deleteall
                    else
                        error "Duplicit argument for 'delete all' option! (IGNORING)"
                    fi
                    ;;
                -p|--prefix)
                    state=prefix
                    ;;
                -C|--clean-only)
                    if [ -z "$arg_cleanonly" ] ; then
                        arg_cleanonly=cleanonly
                    else
                        error "Duplicit argument for 'clean only' option! (IGNORING)"
                    fi
                    ;;
                *)
                    error "Unknown parameter: $1"
                    use_help
                    exit 1
                    ;;
            esac
            ;;
        namespace)
            if [ -z "$arg_namespace" ] ; then
                arg_namespace="$1"
                state=nil
            else
                error "Duplicit argument for namespace!"
                use_help
                exit 1
            fi
            ;;
        override)
            if ! [ -f "$1" ] ; then
                error "Wrong filename for override file: $1"
                use_help
                exit 1
            fi
            arg_overrides="${arg_overrides} -f $1"
            state=nil
            ;;
        component)
            arg_components="${arg_components} $1"
            state=nil
            ;;
        prefix)
            if [ -z "$arg_prefix" ] ; then
                arg_prefix="$1"
                state=nil
            else
                error "Duplicit argument for release prefix!"
                use_help
                exit 1
            fi
            ;;
        timeout)
            if [ -z "$arg_timeout" ] ; then
                if ! echo "$1" | grep -q '^[0-9]\+$' ; then
                    error "Timeout must be an integer: $1"
                    use_help
                    exit 1
                fi
                arg_timeout="$1"
                state=nil
            else
                error "Duplicit argument for timeout!"
                use_help
                exit 1
            fi
            ;;
        storage)
            if [ -n "$arg_nostorage" ] ; then
                error "Usage of storage argument together with no storage deletion option!"
                use_help
                exit 1
            elif [ -z "$arg_storage" ] ; then
                arg_storage="$1"
                state=nil
            else
                error "Duplicit argument for storage!"
                use_help
                exit 1
            fi
            ;;
    esac
    shift
done

# sanity checks

if [ -z "$arg_namespace" ] ; then
    error "Missing namespace"
    use_help
    exit 1
else
    NAMESPACE="$arg_namespace"
fi

if [ -z "$arg_overrides" ] && [ -z "$arg_cleanonly" ] ; then
    error "Missing override file(s) or use '--clean-only'"
    use_help
    exit 1
else
    OVERRIDES="$arg_overrides"
fi

if [ -n "$arg_prefix" ] ; then
    RELEASE_PREFIX="$arg_prefix"
fi

if [ -n "$arg_timeout" ] ; then
    HELM_TIMEOUT="$arg_timeout"
fi

if [ -n "$arg_storage" ] ; then
    VOLUME_STORAGE="$arg_storage"
elif [ -z "$arg_nostorage" ] ; then
    error "Missing storage argument! If it is intended then use '--no-storage-deletion' option"
    use_help
    exit 1
fi

if [ -n "$arg_components" ] ; then
    HELM_CHART_RELEASE_NAME="$arg_components"
fi

if [ -n "$arg_deleteall" ] ; then
    HELM_DELETE_ALL=yes
fi

if [ -n "$arg_cleanonly" ] ; then
    HELM_SKIP_DEPLOY=yes
fi

# If running with helm v3 a time unit has to be appended to HELM_TIMEOUT
if [[ $(helm version --template "{{.Version}}") =~ ^v3 ]];then
    HELM_TIMEOUT="${HELM_TIMEOUT}s"
fi

#
# main
#

# set trap for this script cleanup
trap on_exit INT QUIT TERM EXIT

# another sanity checks
for tool in helm kubectl curl ; do
    if ! which "$tool" >/dev/null 2>&1 ; then
        error "Missing '${tool}' command"
        exit 1
    fi
done

if ! is_helm_serve_running ; then
    error "'helm serve' is not running (http://localhost:8879)"
    exit 1
fi

# if --delete-all is used then redeploy all components (the current namespace is deleted)
if [ -n "$HELM_DELETE_ALL" ] ; then
    # undeploy helm release (prefix)
    helm_undeploy "$RELEASE_PREFIX"

    # we will delete the whole namespace
    delete_namespace

    # we will cleanup docker on each node
    docker_cleanup

    # we will delete the content of storage (volumes)
    if [ -n "$VOLUME_STORAGE" ] ; then
        delete_storage
    fi
# delete and redeploy explicit or failed components...
else
    # if a helm chart release name was given then just redeploy said component and quit
    if [ -n "$HELM_CHART_RELEASE_NAME" ] ; then
        msg "Explicitly asked for component redeploy: ${HELM_CHART_RELEASE_NAME}"
        _COMPONENTS="$HELM_CHART_RELEASE_NAME"
    # simple heuristics: redeploy only failed components
    else
        msg "Delete successfully completed jobs..."
        clean_jobs

        msg "Find failed components..."
        _COMPONENTS=$(get_failed_labels)
    fi

    for _component in ${_COMPONENTS} ; do
        if echo "$_component" | grep -q "^${RELEASE_PREFIX}-" ; then
            msg "Redeploy component: ${_component}"
            undeploy_component ${_component}
        else
            error "Component release name '${_component}' does not match release prefix: ${RELEASE_PREFIX} (SKIP)"
        fi
    done
fi

if [ -z "$HELM_SKIP_DEPLOY" ] ; then
    # TODO: this is suboptimal - find a way how to deploy only the affected component...
    msg "Redeploy onap..."
    msg helm deploy ${RELEASE_PREFIX} local/onap --namespace ${NAMESPACE} ${OVERRIDES} --timeout ${HELM_TIMEOUT}
    helm deploy ${RELEASE_PREFIX} local/onap --namespace ${NAMESPACE} ${OVERRIDES} --timeout ${HELM_TIMEOUT}
else
    msg "Clean only option used: Skipping redeploy..."
fi

msg DONE

exit $?