Create backup functionality.

Author: binbin21avery

addons/halo/dataprotection/backup-info-collector.sh

@@ -0,0 +1,29 @@
+function get_current_time() {
+  curr_time=$(psql -U ${DP_DB_USER} -d halo0root -t -c "SELECT now() AT TIME ZONE 'UTC'")
+  echo $curr_time
+}
+
+function stat_and_save_backup_info() {
+  export PATH="$PATH:$DP_DATASAFED_BIN_PATH"
+  export DATASAFED_BACKEND_BASE_PATH="$DP_BACKUP_BASE_PATH"
+  START_TIME=$1
+  STOP_TIME=$2
+  if [ -z $STOP_TIME ]; then
+    STOP_TIME=$(get_current_time)
+  fi
+  START_TIME=$(date -d "${START_TIME}" -u '+%Y-%m-%dT%H:%M:%SZ')
+  STOP_TIME=$(date -d "${STOP_TIME}" -u '+%Y-%m-%dT%H:%M:%SZ')
+  TOTAL_SIZE=$(datasafed stat / | grep TotalSize | awk '{print $2}')
+  echo "{\"totalSize\":\"$TOTAL_SIZE\",\"timeRange\":{\"start\":\"${START_TIME}\",\"end\":\"${STOP_TIME}\"}}" >"${DP_BACKUP_INFO_FILE}"
+}
+
+# if the script exits with a non-zero exit code, touch a file to indicate that the backup failed,
+# the sync progress container will check this file and exit if it exists
+function handle_exit() {
+  exit_code=$?
+  if [ $exit_code -ne 0 ]; then
+    echo "failed with exit code $exit_code"
+    touch "${DP_BACKUP_INFO_FILE}.exit"
+    exit 1
+  fi
+}

addons/halo/dataprotection/halo-basebackup-backup.sh

@@ -0,0 +1,13 @@
+set -e
+set -o pipefail
+export PATH="$PATH:$DP_DATASAFED_BIN_PATH"
+export DATASAFED_BACKEND_BASE_PATH="$DP_BACKUP_BASE_PATH"
+export PGPASSWORD=${DP_DB_PASSWORD}
+
+trap handle_exit EXIT
+
+START_TIME=`get_current_time`
+echo ${DP_DB_PASSWORD} | pg_basebackup -Ft -Pv -c fast -Xf -D - -h ${DP_DB_HOST} -U ${DP_DB_USER} -W | datasafed push -z zstd-fastest - "/${DP_BACKUP_NAME}.tar.zst"
+
+# stat and save the backup information
+stat_and_save_backup_info $START_TIME

addons/halo/templates/actionset-halobasebackup.yaml

@@ -0,0 +1,33 @@
+apiVersion: dataprotection.kubeblocks.io/v1alpha1
+kind: ActionSet
+metadata:
+  name: pg-basebackup
+  labels:
+    clusterdefinition.kubeblocks.io/name: halo
+    {{- include "halo.labels" . | nindent 4 }}
+spec:
+  backupType: Full
+  env:
+    - name: DATA_DIR
+      value: {{ .Values.dataMountPath }}
+    - name: DP_DB_USER
+      value: halo
+    - name: DP_BACKUP_BASE_PATH
+      value: /u01/app/halo/product/dbms/14/bin
+    - name: DP_DB_HOST
+      value: localhost
+  backup:
+    preBackup: []
+    postBackup: []
+    backupData: 
+      image: {{ .Values.image.registry }}/{{ .Values.image.repository }}:{{ .Values.image.tag }}
+      command:
+      - bash
+      - -c
+      - |
+        {{- .Files.Get "dataprotection/backup-info-collector.sh" | nindent 8 -}}
+        {{- .Files.Get "dataprotection/halo-basebackup-backup.sh" | nindent 8 -}}
+      syncProgress:
+        enabled: true
+        intervalSeconds: 5
+

addons/halo/templates/backuppolicytemplate.yaml

@@ -0,0 +1,25 @@
+apiVersion: apps.kubeblocks.io/v1alpha1
+kind: BackupPolicyTemplate
+metadata:
+  name: halo-backup-policy-template
+  labels:
+    clusterdefinition.kubeblocks.io/name: halo
+    {{- include "halo.labels" . | nindent 4 }}
+spec:
+  clusterDefinitionRef: halo
+  backupPolicies:
+  - componentDefRef: halo
+    backupMethods:
+    - name: pg-basebackup
+      snapshotVolumes: false
+      actionSetName: pg-basebackup
+      targetVolumes:
+        volumeMounts:
+        - name: data
+          mountPath: {{ .Values.dataMountPath }}
+    schedules: 
+    - backupMethod: pg-basebackup
+      enabled: false
+      cronExpression: "*/5 * * * *"
+      retentionPeriod: 7d
+