ApacheBeamPython的Dataflow在使用GCPPub/Sub计数时进行了过多的计数。
创始人
2024-09-05 11:30:37
0次

可以通过将窗口持续时间设置为 Pub/Sub 保留的最长时间来解决该问题,比如1小时。具体实现代码示例如下:

# 定义窗口持续时间为1小时
window_duration_in_seconds = 60 * 60

# 定义 Pub/Sub 的话题
input_topic = "projects/{project}/topics/{topic}".format(
   project=project, topic=topic)

# 定义数据流管道
pipeline = beam.Pipeline(options=options)

# 从 Pub/Sub 话题中读取数据流
messages = (
    pipeline
    | "Read Pub/Sub Messages" >> beam.io.gcp.pubsub.ReadFromPubSub(topic=input_topic)
    | "Decode Messages from JSON" >> beam.Map(lambda x: json.loads(x)))

# 固定窗口持续时间为1小时
fixed_windows = (
    messages
    | "add timestamp to messages" >> beam.ParDo(AddTimestampFn())
    | "Create Fixed Windows with Hourly duration" >> beam.WindowInto(
        beam.window.FixedWindows(window_duration_in_seconds))
)

# 进行计数操作
message_count = (
    fixed_windows
    | "Extract Pub/Sub message ids" >> beam.Map(lambda message: message.get("message_id"))
    | "Count messages" >> beam.combiners.Count.PerElement())

# 将计数结果写入云存储
message_count | "Write results to GCS" >> WriteToText(output_path)

在以上示例代码中,“AddTimestampFn”是一个自定义的 DoFn 函数,它的作用是为每条消息添加时间戳信息,以便进行窗口操作中的时间约束。最后计数结果将被写入云存储中。

相关内容

热门资讯

每日必看教程开挂攻略!指尖四川... 每日必看教程开挂攻略!指尖四川麻将多功能开挂攻略,指尖四川麻将切实真的是有开挂攻略1、指尖四川麻将辅...
实测必看作弊器技巧!梦黄麻将多... 实测必看作弊器技巧!梦黄麻将多功能作弊器技巧,梦黄麻将果然存在有作弊器技巧1、梦黄麻将免费脚本咨询教...
研究成果辅助器教程!小白大作战... 研究成果辅助器教程!小白大作战多功能辅助器教程,小白大作战确实真的有辅助器教程所有人都在同一条线上,...
分享认知辅助软件!小闲巴渝棋牌... 分享认知辅助软件!小闲巴渝棋牌多功能辅助软件,小闲巴渝棋牌竟然是有辅助软件1、不需要AI权限,帮助你...
分享认知透视器app!温岭逸趣... 分享认知透视器app!温岭逸趣麻将多功能透视器app,温岭逸趣麻将真是是有透视器app1)温岭逸趣麻...
专业讨论作弊器插件!欢乐划水麻... 专业讨论作弊器插件!欢乐划水麻将多功能作弊器插件,欢乐划水麻将果然真的是有作弊器插件1、用户打开应用...
最新技巧外挂攻略!逍遥皖北棋牌... 最新技巧外挂攻略!逍遥皖北棋牌多功能外挂攻略,逍遥皖北棋牌都是存在有外挂攻略亲,关键说明,逍遥皖北棋...
今日公布辅助app!乐通多功能... 今日公布辅助app!乐通多功能辅助app,乐通一直有辅助app1、乐通破解器简单,乐通机器人操作,乐...
揭秘一下辅助攻略!青橙竞技多功... 揭秘一下辅助攻略!青橙竞技多功能辅助攻略,青橙竞技一贯真的是有辅助攻略1、玩家可以在青橙竞技线上大神...
推荐一款透视挂技巧!无锡麻将多... 推荐一款透视挂技巧!无锡麻将多功能透视挂技巧,无锡麻将原来有透视挂技巧1、起透看视 无锡麻将辅助软件...