删除Apache Beam2.2.0中的Bigtable行

import apache_beam as beam
from apache_beam.io.gcp.bigtableio import WriteToBigTable

# 创建Bigtable连接
project_id = 'your-project-id'
instance_id = 'your-instance-id'
table_id = 'your-table-id'
connection_config = {
    'project_id': project_id,
    'instance_id': instance_id,
    'table_id': table_id
}

# 获取要删除的行的RowKey或RowKeys
row_keys = ['row_key1', 'row_key2']

# 删除行的方法
def delete_rows(row):
    return beam.RowMutation(delete=row)

# 构建删除操作的PCollection
delete_pcoll = (
    beam.Create(row_keys)
    | beam.Map(delete_rows)
)

# 执行删除操作
delete_pcoll | WriteToBigTable(connection_config)

# 运行Pipeline
result = beam.run().wait_until_finish()

# 验证删除操作是否成功
if result == beam.PipelineState.DONE:
    print('Rows deleted successfully.')
else:
    print('Failed to delete rows.')