CLD-479: Adding ML Host changes commit

* CLD-479: Adding ML Host changes commit

* CLD-479: Updated the Shut Down retry logic

* CLD-479: Improve logging message format

* CLD-479: Improved logging logic

* CLD-479: Updated the logging logic to capture the correct timestamp

Co-authored-by: Sumanth Ravipati <[email protected]>

Author: sumanthravipati

charts/templates/statefulset.yaml

@@ -20,6 +20,7 @@ spec:
       {{- with .Values.affinity }}
       affinity: {{- toYaml . | nindent 8}}
       {{- end }}
+      terminationGracePeriodSeconds: {{ .Values.terminationGracePeriod }}
       containers:
         - name: marklogic-server
           image: "{{ .Values.image.repository }}:{{ .Values.image.tag }}"
@@ -60,6 +61,53 @@ spec:
             {{- if .Values.extraContainerPorts }}
               {{- toYaml .Values.extraContainerPorts | nindent 12 }}
             {{- end }}
+          lifecycle:
+            preStop:
+              exec:
+                command:
+                  - bash
+                  - '-c'
+                  - |
+                    log () {
+                      local TIMESTAMP=$(date +"%Y-%m-%d %T.%3N")
+                      echo "${TIMESTAMP} $@" > /proc/$pid/fd/1
+                    }
+
+                    pid=$(ps aux | grep -i '/bin/bas[h] /usr' | awk {'print $2'})
+                    log "Info: [prestop] Prestop Hook Execution"
+
+                    my_host=$(hostname -f)
+
+                    log "Info: [prestop] MarkLogic Pod Hostname: "$my_host
+
+                    for ((i = 0; i < 5; i = i + 1)); do
+                        res_code=$(curl --anyauth --user $MARKLOGIC_ADMIN_USERNAME:$MARKLOGIC_ADMIN_PASSWORD \
+                            -o /dev/null -m 10 -s -w %{http_code} \
+                            -i -X POST --data "state=shutdown&failover=true" \
+                            -H "Content-type: application/x-www-form-urlencoded" \
+                            http://localhost:8002/manage/v2/hosts/$my_host?format=json)
+
+                        if [[ ${res_code} -eq 202 ]]; then
+                            log "Info: [prestop] Host shut down response code: "$res_code
+
+                            while (true)
+                            do
+                               ml_status=$(service MarkLogic status)
+                               log "Info: [prestop] MarkLogic Status: "$ml_status
+                               if [[ "$ml_status" =~ "running" ]]; then
+                                  sleep 5s
+                                  continue
+                               else
+                                  break
+                               fi
+                            done
+                            break
+                        else
+                            log "ERROR: [prestop] Retry Attempt: "$i
+                            log "ERROR: [prestop] Host shut down expected response code 202, got "$res_code
+                            sleep 10s
+                        fi
+                    done
           {{- if .Values.livenessProbe.enabled }}
           livenessProbe:
             httpGet:

charts/values.yaml

@@ -3,10 +3,13 @@
 # Number of Marklogic nodes
 replicaCount: 1
 
+# Termination Grace Period
+terminationGracePeriod: 120
+
 # Marklogic image parameters
 image:
   repository: marklogicdb/marklogic-db
-  tag: 10.0-9.1-centos-1.0.0-ea4
+  tag: 10.0-9.2-centos-1.0.0-ea4
   pullPolicy: IfNotPresent
 
 # Configure the imagePullSecret to pull the image from private repository that requires credential