如何通过Python API在BigQuery中添加主键约束？

要给user_id字段添加主键约束，你需要利用BigQuery的TableConstraints类定义主键规则，再绑定到Table对象上。注意主键字段必须是REQUIRED模式，你的代码里user_id已经满足这个条件，可以直接设置。

修改后的完整代码如下：

from google.cloud import bigquery

def create_table():
    schema = [
        bigquery.SchemaField("user_id", "STRING", mode="REQUIRED"),
        bigquery.SchemaField("first_name", "STRING"),
        bigquery.SchemaField("last_name", "STRING"),
        bigquery.SchemaField("age", "INTEGER")
    ]
    # 定义主键约束
    table_constraints = bigquery.TableConstraints(
        primary_key=bigquery.PrimaryKey(columns=["user_id"])
    )
    # 创建Table对象并绑定约束
    table = bigquery.Table(table_id, schema=schema)
    table.constraints = table_constraints
    # 执行表创建
    table = client.create_table(table)

关键说明 #

• 主键字段必须设置为REQUIRED模式，否则创建表时会触发报错
• BigQuery的主键约束是强制执行的，会直接阻止插入主键重复的行
• 如果要给已存在的表添加主键，需使用client.update_table()方法，传入修改后的Table对象和["constraints"]作为更新字段

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