Apache Beam无法正确从Google Cloud Storage接收pub/sub消息。
创始人
2024-09-03 15:02:18
0

要从Google Cloud Storage接收pub/sub消息,您需要使用Google Cloud Pub/Sub I/O模块的ReadFromPubSub方法。以下是一个使用Apache Beam的Python代码示例,演示如何正确从Google Cloud Storage接收pub/sub消息:

import apache_beam as beam
from apache_beam.options.pipeline_options import PipelineOptions

# 定义自定义选项
class MyOptions(PipelineOptions):
    @classmethod
    def _add_argparse_args(cls, parser):
        parser.add_argument('--input_topic', help='Input Pub/Sub topic')
        parser.add_argument('--output_bucket', help='Output GCS bucket')
        parser.add_argument('--output_prefix', help='Output GCS prefix')

# 定义处理函数
def process_message(message):
    # 处理接收到的消息
    # 这只是一个示例,您可以根据实际需求进行处理
    # 在这个示例中,我们将消息写入Google Cloud Storage
    output_bucket = options.output_bucket
    output_prefix = options.output_prefix
    output_file = f'{output_bucket}/{output_prefix}/output.txt'

    with beam.io.gcp.gcsio.GcsIO().open(output_file, 'w') as file:
        file.write(message.data)

# 解析命令行参数
options = MyOptions()
options.view_as(beam.options.pipeline_options.StandardOptions).runner = 'DirectRunner'
pipeline = beam.Pipeline(options=options)

# 从Pub/Sub读取消息
input_topic = options.input_topic
messages = (
    pipeline
    | 'Read from Pub/Sub' >> beam.io.ReadFromPubSub(topic=input_topic)
    | 'Process messages' >> beam.Map(process_message)
)

# 运行管道
pipeline.run()

在上述代码中,您需要将--input_topic参数设置为您的Pub/Sub主题,--output_bucket参数设置为您的GCS存储桶,--output_prefix参数设置为输出文件的前缀。process_message函数是用于处理接收到的pub/sub消息的自定义函数。在这个示例中,它将消息写入了Google Cloud Storage。

请确保已安装apache_beamgoogle-cloud-pubsub Python包,并设置正确的Google Cloud凭据。

相关内容

热门资讯

【NI Multisim 14...   目录 序言 一、工具栏 🍊1.“标准”工具栏 🍊 2.视图工具...
银河麒麟V10SP1高级服务器... 银河麒麟高级服务器操作系统简介: 银河麒麟高级服务器操作系统V10是针对企业级关键业务...
不能访问光猫的的管理页面 光猫是现代家庭宽带网络的重要组成部分,它可以提供高速稳定的网络连接。但是,有时候我们会遇到不能访问光...
AWSECS:访问外部网络时出... 如果您在AWS ECS中部署了应用程序,并且该应用程序需要访问外部网络,但是无法正常访问,可能是因为...
Android|无法访问或保存... 这个问题可能是由于权限设置不正确导致的。您需要在应用程序清单文件中添加以下代码来请求适当的权限:此外...
北信源内网安全管理卸载 北信源内网安全管理是一款网络安全管理软件,主要用于保护内网安全。在日常使用过程中,卸载该软件是一种常...
AWSElasticBeans... 在Dockerfile中手动配置nginx反向代理。例如,在Dockerfile中添加以下代码:FR...
AsusVivobook无法开... 首先,我们可以尝试重置BIOS(Basic Input/Output System)来解决这个问题。...
ASM贪吃蛇游戏-解决错误的问... 要解决ASM贪吃蛇游戏中的错误问题,你可以按照以下步骤进行:首先,确定错误的具体表现和问题所在。在贪...
月入8000+的steam搬砖... 大家好,我是阿阳 今天要给大家介绍的是 steam 游戏搬砖项目,目前...