排查环境更新和升级问题

Cloud Composer 1 | Cloud Composer 2

本页面针对更新或升级 Cloud Composer 环境时可能遇到的问题提供了问题排查信息。

如需了解与创建环境相关的问题排查信息，请参阅排查环境创建问题。

在更新 Cloud Composer 环境时，导致大多数问题的原因如下：

  • 服务账号权限问题
  • PyPI 依赖项问题
  • Airflow 数据库的大小

权限不足，无法更新或升级环境

如果 Cloud Composer 由于权限不足而无法更新或升级环境，则会输出以下错误消息：

ERROR: (gcloud.composer.environments.update) PERMISSION_DENIED: The caller does not have permission

解决方案：按照访问权限控制中所述，为您的帐号和环境的服务帐号分配角色。

环境的服务账号权限不足

创建 Cloud Composer 环境时，您需要指定一个服务帐号，用于运行该环境的 GKE 集群节点。如果此服务帐号没有足够的权限来执行请求的操作，则 Cloud Composer 会输出错误：

    UPDATE operation on this environment failed 3 minutes ago with the
    following error message:
    Composer Backend timed out. Currently running tasks are [stage:
    CP_COMPOSER_AGENT_RUNNING
    description: "No agent response published."
    response_timestamp {
      seconds: 1618203503
      nanos: 291000000
    }
    ].

解决方案：按照访问权限控制中所述，为您的帐号和环境的服务帐号分配角色。

Airflow 数据库太大，无法执行此操作

Cloud Composer 升级操作可能会失败，因为 Airflow 数据库过大，导致升级操作无法成功。

如果 Airflow 数据库的大小超过 16 GB，Cloud Composer 会输出以下错误：

Airflow database uses more than 16 GB. Please clean the database before upgrading.

解决方案：执行 Airflow 数据库清理，如 Airflow 数据库维护中所述。

由于 PyPI 软件包冲突，升级到新的 Cloud Composer 版本失败

当您升级具有已安装的自定义 PyPI 软件包的环境时，您可能会遇到与 PyPI 软件包冲突相关的错误。可能会发生这种情况，因为新的 Cloud Composer 映像包含较新版本的预安装软件包，导致依赖项与您环境中安装的 PyPI 软件包发生冲突。

解决方案：

  • 如需获取有关软件包冲突的详细信息，请运行升级检查。
  • 放宽已安装的自定义 PyPI 软件包的版本限制条件。例如，不要将版本指定为 ==1.0.1，而应将其指定为 >=1.0.1。
  • 如需详细了解如何更改版本要求以解决存在冲突的依赖项，请参阅 pip 文档。

缺少 DNS 连接可能会导致在执行升级或更新时出现问题

此类连接问题可能会导致出现如下日志条目：

WARNING - Compute Engine Metadata server unavailable attempt 1 of 5. Reason: [Errno -3] Temporary failure in name resolution Error

这通常意味着没有通向 DNS 的路由，因此请确保 metadata.google.internal DNS 名称可以从集群、Pod 和服务网络中解析为 IP 地址。检查您是否在创建环境的 VPC（在宿主项目或服务项目中）中启用了专用 Google 访问通道。

更多信息：

触发器 CPU 超出了 1 个 vCPU 的限制

版本 2.4.4 及更高版本中的 Cloud Composer 2 引入了不同的触发器资源分配策略来改进性能伸缩。 如果您在执行环境更新时遇到与触发器 CPU 相关的错误，则表示您当前的触发器已配置为每个触发器使用 1 个以上的 vCPU。

解决方案：

检查失败的迁移警告

将 Airflow 升级到更高版本时，有时会对 Airflow 数据库应用新的限制条件。如果无法应用这些约束条件，则 Airflow 会创建新表来存储无法应用这些约束条件的行。在重命名或删除移动的数据表之前，Airflow 界面会显示警告消息。

解决方案：

您可以使用以下两个 DAG 检查移动的数据并重命名表。

list_moved_tables_after_upgrade_dag DAG 列出从无法应用限制条件的每个表中移动的行。检查数据并决定是否要保留它。如需保留它，您需要手动修复 Airflow 数据库中的数据。例如，重新添加包含正确数据的行。

如果您不需要数据或者已经修复了数据，则可以运行 rename_moved_tables_after_upgrade_dag DAG。此 DAG 会重命名移动的表。 这些表及其数据不会被删除，因此您可以稍后查看数据。

"""
When upgrading Airflow to a newer version,
it might happen that some data cannot be migrated,
often because of constraint changes in the metadata base.
This file contains 2 DAGs:

1. 'list_moved_tables_after_upgrade_dag'
  Prints the rows which failed to be migrated.
2. 'rename_moved_tables_after_upgrade_dag'
  Renames the table which contains the failed migrations. This will remove the
  warning message from airflow.
"""

import datetime
import logging

from airflow import DAG
from airflow.operators.python import PythonOperator
from airflow.providers.postgres.hooks.postgres import PostgresHook
from airflow.settings import AIRFLOW_MOVED_TABLE_PREFIX

def get_moved_tables():
    hook = PostgresHook(postgres_conn_id="airflow_db")
    return hook.get_records(
        "SELECT schemaname, tablename FROM pg_catalog.pg_tables WHERE tablename"
        f" LIKE '{AIRFLOW_MOVED_TABLE_PREFIX}_%'"
    )

def list_moved_records():
    tables = get_moved_tables()
    if not tables:
        logging.info("No moved tables found")
        return

    hook = PostgresHook(postgres_conn_id="airflow_db")
    for schema, table in tables:
        df = hook.get_pandas_df(f"SELECT * FROM {schema}.{table}")
        logging.info(df.to_markdown())

def rename_moved_tables():
    tables = get_moved_tables()
    if not tables:
        return

    hook = PostgresHook(postgres_conn_id="airflow_db")
    for schema, table in tables:
        hook.run(f"ALTER TABLE {schema}.{table} RENAME TO _abandoned_{table}")

with DAG(
    dag_id="list_moved_tables_after_upgrade_dag",
    start_date=datetime.datetime(2023, 1, 1),
    schedule_interval=None,
    catchup=False,
):
    t1 = PythonOperator(
        task_id="list_moved_records", python_callable=list_moved_records
    )

with DAG(
    dag_id="rename_moved_tables_after_upgrade_dag",
    start_date=datetime.datetime(2023, 1, 1),
    schedule_interval=None,
    catchup=False,
) as dag:
    t1 = PythonOperator(
        task_id="rename_moved_tables", python_callable=rename_moved_tables
    )

后续步骤