如何在BigQuery存储过程中处理特定异常？含Python替代方案咨询

一、BigQuery存储过程实现方式 #

BigQuery不支持Oracle中类似ZERO_DIVIDE的错误名称捕获，而是通过错误代码识别特定异常。以除零错误为例，对应的错误代码是300001，你可以通过定义条件来捕获该错误：

示例代码 #

CREATE OR REPLACE PROCEDURE `你的项目ID.你的数据集ID.你的存储过程名`(OUT some_output INT64)
BEGIN
  -- 定义除零错误对应的条件
  DECLARE division_by_zero CONDITION FOR ERROR_CODE '300001';
  
  BEGIN
    -- 可能触发除零的业务逻辑
    SET some_output = 100 / 0;
  EXCEPTION
    -- 捕获除零错误并赋值
    WHEN division_by_zero THEN
      SET some_output = 5000;
    -- 若需兜底捕获其他错误，可保留WHEN OTHERS（按需选择）
    -- WHEN OTHERS THEN
    --   SET some_output = -1;
  END;
END;

说明 #

• 其他特定错误的代码可通过BigQuery官方文档查询；
• 自定义条件时使用DECLARE [条件名] CONDITION FOR ERROR_CODE '[错误码]'语法，后续在EXCEPTION块中直接引用即可。

二、Python替代方案 #

使用BigQuery Python客户端执行SQL逻辑，在代码层捕获并处理特定错误：

示例代码 #

from google.cloud import bigquery
from google.api_core.exceptions import BadRequest

# 初始化BigQuery客户端
client = bigquery.Client()

def handle_division_logic():
    some_output = None
    # 可能触发除零的SQL查询
    query = "SELECT 100 / 0 AS calculation_result"
    
    try:
        query_job = client.query(query)
        # 执行查询并获取结果
        results = query_job.result()
        for row in results:
            some_output = row.calculation_result
    except BadRequest as e:
        # 解析错误详情，判断是否为除零错误
        for error_detail in e.errors:
            if (error_detail.get("reason") == "invalidQuery" 
                and error_detail.get("location") == "divide"
                and "Division by zero" in error_detail.get("message", "")):
                some_output = 5000
    # 输出处理后的结果
    print(f"最终输出值: {some_output}")

# 调用函数执行逻辑
handle_division_logic()

说明 #

• 通过捕获BadRequest异常，解析错误详情中的reason、location和message字段，精准识别除零错误；
• 处理其他特定错误时，可根据对应的错误特征调整判断条件。

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