借助pre-install、pre-upgrade Hook在Helm Chart中执行数据库迁移的问题

当Helm Chart自带数据库部署时，pre-install阶段数据库尚未创建或就绪，直接用pre-install Hook执行迁移或初始数据插入必然失败。以下是几种可行的解决方案：

1. 改用post-install/post-upgrade Hook + 数据库就绪等待 #

将Hook阶段从pre-install/pre-upgrade改为post-install（首次安装后）和post-upgrade（升级后），确保数据库资源已完成部署。同时在Job容器内加入等待逻辑，确认数据库完全就绪后再执行操作。

示例迁移Job配置：

apiVersion: batch/v1
kind: Job
metadata:
  name: migrate
  annotations:
    "helm.sh/hook": post-install, post-upgrade
    "helm.sh/hook-delete-policy": before-hook-creation, hook-succeeded
    "helm.sh/hook-weight": "5" # 权重设得比数据库部署高，保证数据库先启动
spec:
  template:
    spec:
      containers:
      - name: migrate
        image: your-migration-image:latest
        command:
        - sh
        - -c
        - |
          # 以PostgreSQL为例，等待数据库端口就绪
          until pg_isready -h {{ .Values.db.host }} -p {{ .Values.db.port }} -U {{ .Values.db.user }}; do
            echo "Waiting for database to be ready..."
            sleep 5
          done
          # 执行迁移命令
          ./migrate up
      restartPolicy: OnFailure

2. 首次安装的初始数据插入：post-install Hook + 幂等性判断 #

针对仅需执行一次的初始数据（如管理员账号），使用post-install Hook，并在脚本中加入幂等性检查，避免重复执行。

示例初始数据Job配置：

apiVersion: batch/v1
kind: Job
metadata:
  name: init-data
  annotations:
    "helm.sh/hook": post-install
    "helm.sh/hook-delete-policy": hook-succeeded
    "helm.sh/hook-weight": "6"
spec:
  template:
    spec:
      containers:
      - name: init-data
        image: your-app-image:latest
        command:
        - sh
        - -c
        - |
          # 等待数据库就绪
          until pg_isready -h {{ .Values.db.host }} -p {{ .Values.db.port }} -U {{ .Values.db.user }}; do
            echo "Waiting for database to be ready..."
            sleep 5
          done
          # 检查初始数据是否已存在，存在则跳过
          if psql -h {{ .Values.db.host }} -p {{ .Values.db.port }} -U {{ .Values.db.user }} -d {{ .Values.db.name }} -c "SELECT 1 FROM users WHERE username='admin'" | grep -q 1; then
            echo "Initial data already exists, skipping..."
            exit 0
          fi
          # 执行初始数据插入
          psql -h {{ .Values.db.host }} -p {{ .Values.db.port }} -U {{ .Values.db.user }} -d {{ .Values.db.name }} -f /init-data.sql
      restartPolicy: OnFailure

3. 整合迁移逻辑到应用启动流程（无需Hook） #

如果不想依赖Helm Hook，可以将迁移逻辑放到应用的init容器中，利用数据库锁确保只有一个实例执行迁移，避免多实例冲突。

示例Deployment配置：

apiVersion: apps/v1
kind: Deployment
metadata:
  name: your-app
spec:
  replicas: 3
  template:
    spec:
      initContainers:
      - name: migrate
        image: your-migration-image:latest
        command:
        - sh
        - -c
        - |
          # 等待数据库就绪
          until pg_isready -h {{ .Values.db.host }} -p {{ .Values.db.port }} -U {{ .Values.db.user }}; do
            sleep 5
          done
          # 使用PostgreSQL advisory lock确保单实例执行迁移
          psql -h {{ .Values.db.host }} -p {{ .Values.db.port }} -U {{ .Values.db.user }} -d {{ .Values.db.name }} -c "SELECT pg_advisory_lock(12345)"
          ./migrate up
          psql -h {{ .Values.db.host }} -p {{ .Values.db.port }} -U {{ .Values.db.user }} -d {{ .Values.db.name }} -c "SELECT pg_advisory_unlock(12345)"
      containers:
      - name: app
        image: your-app-image:latest
        # 应用启动命令
        command: ["./app"]

内容的提问来源于stack exchange，提问作者hari