google-bigquery - 使用 BQ API 使用 Write_Truncate 将数据加载到 “Partitioned Tables”

Question

可以使用 write_truncate 将数据加载到特定分区吗？我的用例是覆盖特定分区的批加载过程，同时保持表的其余部分不变。此引用仅提及命令行工具: https://cloud.google.com/bigquery/docs/creating-partitioned-tables#restating_data_in_a_partition

Answer 1

如果您想使用 BQ API 写入特定分区，请像写入表一样继续操作，但在表 ID 中包含分区装饰器。

如果你有一个 date partitioned表 transactions 并且您想将数据加载到对应于 2021-10-11 的分区，将 transactions$20211011 传递给 API。

如果你有一个 ingestion-time partitioned具有小时分区的表 transactions 并且您想将数据加载到对应于 2021-05-07 17:00:00 的分区，传递 transactions$2021050717 到 API。

示例:如果您正在使用 python 的 API 将 DataFrame 加载到 BQ 中并且您想要覆盖分区，您可以这样做:

from google.cloud import bigquery
client = bigquery.Client(project='your_project')

job_config = bigquery.LoadJobConfig(
    write_disposition="WRITE_TRUNCATE", 
)

# Include target partition in the table id:
table_id = "your_project.your_dataset.your_table$20211021" 
job = client.load_table_from_dataframe(df, table_id, job_config=job_config) # Make an API request
job.result() # Wait for job to finish