Pathway 自定义 Python Connector 实战:用 `ConnectorSubject` 把 Twitter 等任意流式数据接入实时数据管道
2026/9/8 20:23:22 网站建设 项目流程

Pathway 自定义 Python Connector 实战:用ConnectorSubject把 Twitter 等任意流式数据接入实时数据管道

【免费下载链接】pathwayPython ETL framework for stream processing, real-time analytics, LLM pipelines, and RAG.项目地址: https://gitcode.com/GitHub_Trending/pa/pathway

本文基于 Pathway 官方示例仓库中的 Twitter 自定义连接器(examples/projects/custom-python-connector-twitter)展开,讲解如何通过继承pw.io.python.ConnectorSubject编写自己的 Python 输入连接器,把外部数据源(文中以 Twitter 流为例)变成一张实时变化的 Pathway 表并持续写出到 CSV。读完本文,你将掌握自定义 Python 连接器的生命周期回调、数据投喂方法、Schema 声明与批量提交语义,并能把同一套模式复用到任意可被 Python 拉取的流式数据源上。

示例背景与适用场景

示例 README(examples/projects/custom-python-connector-twitter/README.md)开篇给出了一个重要前提:

⚠️ Twitter has turned off its free tier streaming API.(Twitter 已关闭其免费版流式 API。)

也就是说,Twitter 在这里只是一个演示用的数据源。README 明确指出,示例的核心目的是展示如何实现自定义 Python 连接器,Twitter API 仅作为例子——任何数据流都可以通过同样的方式送入 Pathway 的实时数据框架(Live Data Framework)。

因此,本示例的学习价值并不依赖 Twitter 是否可用,而在于它完整演示了"连接器(Subject)→pw.io.python.read→ Pathway 表 → 输出"这条自定义数据接入链路的全部写法。

整条管线的运行逻辑

示例脚本 twitter_connector_example.py 把一条"推文流"变成了实时表并落盘到output.csv,整体结构如下:

Twitter 流 (tweepy.StreamingClient.sample) │ on_response 回调 ▼ TwitterSubject 继承 pw.io.python.ConnectorSubject │ run() 中被引擎放入独立线程执行;next(key=..., text=...) 投喂数据 ▼ pw.io.python.read(subject, schema=..., autocommit_duration_ms=1000) → 实时 Table ▼ pw.io.csv.write(table, "output.csv") 每次提交将表的变更流写入 CSV ▼ pw.run() 阻塞运行,CTRL+C 触发 KeyboardInterrupt 退出

管线的四个组成部分都集中在同一个文件中,下面逐段拆解。

源码逐段解析

1. 引擎与许可证初始化

import pathway as pw # To use advanced features with Pathway Live Data Framework Scale, get your free license key... # To use Pathway Live Data Framework Community, comment out the line below. pw.set_license_key("demo-license-key-with-telemetry") BEARER_TOKEN = os.environ["TWITTER_API_TOKEN"]

脚本通过pw.set_license_key(...)声明运行模式:使用 Scale 高级功能时填入从官方获取的许可证;仅使用 Community(社区版)时按注释说明注释掉这一行即可。演示脚本中的demo-license-key-with-telemetry是一个带遥测的演示占位 key。Bearer Token 从环境变量TWITTER_API_TOKEN读取,而不是硬编码在文件里——这是示例刻意示范的安全做法。

2.TwitterClient:对接第三方 SDK,把回调翻译成next

class TwitterClient(tweepy.StreamingClient): _subject: TwitterSubject def __init__(self, subject: TwitterSubject) -> None: super().__init__(BEARER_TOKEN) self._subject = subject def on_response(self, response) -> None: self._subject.next( key=response.data.id, text=response.data.text, )

TwitterClient继承 tweepy 的StreamingClient,是纯粹的第三方 SDK 封装层,Pathway 对它的存在一无所知。关键在on_response回调:每当流式 API 推来一条推文,它就把它转换成符合后续 Schema 的关键字参数(keytext),再调用self._subject.next(...)投喂给 Subject。建议将"第三方回调 → 标准化的 Schema 列值"的翻译逻辑放在这一层,让 Subject 只关心 Pathway 侧的数据格式。

注意:该示例通过next传的是结构化字段key/text两个命名列),而非把整条消息打包进单一的data列。这一点与 ConnectorSubject.next 的文档化示例 一致:next按关键字参数发送一行,参数必须与传给read的 Schema 兼容。

3.TwitterSubject:连接器的核心实现

class TwitterSubject(pw.io.python.ConnectorSubject): _twitter_client: TwitterClient def __init__(self) -> None: super().__init__() self._twitter_client = TwitterClient(self) def run(self) -> None: self._twitter_client.sample() def on_stop(self) -> None: self._twitter_client.disconnect()

这是整篇文章的技术核心。ConnectorSubject是 python/pathway/io/python/init.py 中定义的一个抽象基类,只要求实现run,其余全部是内嵌能力。示例用三个方法写清了连接器的完整契约:

方法在示例中的职责引擎何时调用
__init__实例化TwitterClient并把自己传给它,形成回调回路构造阶段
run调用self._twitter_client.sample()启动(近乎无限期的)推文流由引擎在独立线程中启动(见源码 start())
on_stop调用disconnect()优雅断开流式连接,做清理run结束或异常后(源码:finally分支中先on_stop()close()

从 start() 的实现 可以看到精确的生命周期语义:start会启动一个守护线程执行run,一旦run返回(有限连接器)或抛出异常,finally里会先调用on_stop()做清理,再调用close()向缓冲区放入结束哨兵(FINISH_LITERAL)。所以对无限流而言,正确的中止路径是"外部中断runon_stop清理",示例通过CTRL+C触发 KeyboardInterrupt 走到这条路径。

4. Schema:把投喂数据声明为一张表

class InputSchema(pw.Schema): key: int = pw.column_definition(primary_key=True) text: str

Schema 把 Subject 通过next发出的关键字参数一一映射成表的列:key被声明为主键列(primary_key=True),text为普通文本列。主键的存在意味着引擎对同 key 行的处理遵循"按主键更新/去重"语义。关于 Schema 的完整定义方式(列类型、默认值、主键、数据类型),可参考仓库中配套教程 docs/2.developers/4.user-guide/20.connect/99.connectors/30.custom-python-connectors.md。

5. 读入、落盘与运行

input = pw.io.python.read( TwitterSubject(), schema=InputSchema, autocommit_duration_ms=1000, ) pw.io.csv.write(input, "output.csv") try: pw.run() except KeyboardInterrupt: print("Done.")

三个调用分别对应:接入(read)、输出(write)、运行(run)

  • pw.io.python.read(subject, schema=..., autocommit_duration_ms=1000):把 Subject 注册为数据源。其中autocommit_duration_ms=1000表示每 1000 毫秒自动提交一次累积的更新并推入数据流图;该参数在 read 的签名 中默认值为1500毫秒。提交语义决定了实时性的粒度——commit 之间的数据会被缓冲,只有提交后引擎才开始处理。
  • pw.io.csv.write(input, "output.csv"):把表的变更流(而非最终快照)持续写入output.csv。如 python/pathway/io/csv/init.py 的文档示例所示,CSV 会在数据列之外追加time(批次序号)与diff1表示新增、-1表示删除)两列来描述增量变更。
  • try/except KeyboardInterruptpw.run()阻塞运行整个数据流;用户按下CTRL+C后抛出KeyboardInterrupt并打印Done.,这正是 README 第 5 步"按 CTRL+C 停止脚本"的代码体现。

ConnectorSubject的运行机制:引擎与缓冲区

为了让上面的写法不只是"能跑",还需要理解它背后的机制。从源码 python/pathway/io/python/init.py 可以看到,每个 Subject 内部持有一个线程安全的Queue作为缓冲区,所有投喂方法最终都落到self._buffer.put(...)

  • next(**kwargs):发送一行结构化数据,字段与 Schema 列一一对应(实现);
  • next_str(...)/next_bytes(...)/next_json(...):分别以raw/binary/json三种历史格式发送,通常落到单一的data列,在新代码中推荐直接用next
  • commit():发送提交信号;配合autocommit_duration_ms,决定"何时把缓冲的更新真正交还给引擎处理";
  • close():发送结束哨兵,声明不再有消息;run正常结束时会被自动调用。

引擎侧通过 start/end/read/seek 等回调(封装进api.PythonSubject)消费这个缓冲区。由此可以理解几个对本示例有用的推论:

  1. run在独立线程中执行:阻塞式的sample()不会卡死引擎主循环,on_response里的next只是写入缓冲区。
  2. 无限流不会自动close:只有run返回(有限数据源)或出错时,finally才触发on_stop()close();无限流的中止完全依赖外部的KeyboardInterrupt/disconnect
  3. 一个 Subject 对象只能被一个连接器使用一次:源码在 read 中做了守卫,重复使用会抛出ValueError,需要创建两个独立对象。

运行步骤与环境准备

按照示例 README 的 5 步启动:

第 1 步:安装 Pathway

pip install pathway

安装方式细节(含可选扩展、Python 版本要求)见仓库内的安装教程 docs/2.developers/4.user-guide/10.introduction/20.installation.md。示例 README 中还提示:若要使用 Scale 高级特性,需要申请免费许可证 key 并填入脚本中的pw.set_license_key(...);仅用 Community 则注释掉该行。

第 2 步:安装附加依赖

pip install -r requirements.txt

该示例所需的第三方库极简——requirements.txt 中只有一个固定版本依赖:

tweepy==4.13.0

第 3 步:提供 Twitter Bearer Token

export TWITTER_API_TOKEN=<BEARER_TOKEN>

Token 需要在 Twitter(X)开发者平台创建应用后获取,脚本通过os.environ["TWITTER_API_TOKEN"]读取。由于 Twitter 已关闭免费版流式 API,此步骤在实际执行时可能需要付费档的访问权限——这也再次说明本示例的价值在于模式本身,而非 Twitter 这个具体数据源

第 4 步:运行示例

python twitter_connector_example.py

脚本启动后,推文流被逐步写入output.csv(含time/diff增量列),可以随时用另一个终端查看:

tail -f output.csv

第 5 步:停止脚本

# 在前台终端按下 CTRL+C

KeyboardInterrupt被捕获后打印Done.并优雅退出。

代码仓库层面的行为验证

Python 连接器的行为在仓库测试中有充分覆盖,可作为理解本示例的佐证:

  • python/pathway/tests/test_io.py 中的 test_python_connector:演示了next_json/next_str/next_bytes三种投喂方式产生的行最终在表中等价于手写数据,验证了"回调里发多少条、表里就有多少条"的语义;
  • 同一文件中的 test_python_connector_on_stop:在run为空实现(立即返回)的有限连接器上断言on_stop一定被引擎调用——印证上文介绍的"run结束后必然清理"生命周期。

配套的官方教程 自定义 Python 连接器(custom Python connectors) 与本文档一一对应:先讲"读取静态文件"的最简有限场景,再讲本例"Twitter 无限流"的进阶场景,并系统列出ConnectorSubject需要实现的方法(runon_stop)与内嵌方法(nextnext_strnext_bytescommitclose)参考。

把 Twitter 换成你自己的数据源

README 强调:任何数据流都可以用这种方法喂给 Pathway。把示例抽象出来,一个自定义 Python 连接器只需满足一个最小骨架:

import pathway as pw class MySubject(pw.io.python.ConnectorSubject): def run(self) -> None: # 从任意源(WebSocket、消息队列、轮询 HTTP、读文件、生成器等)取数, # 每拿到一条就调用一次 next,字段与下方 Schema 的列一一对应 self.next(key=..., text=...) class InputSchema(pw.Schema): key: int = pw.column_definition(primary_key=True) text: str table = pw.io.python.read( MySubject(), schema=InputSchema, autocommit_duration_ms=1000, # 每 1000ms 提交一批,可调小以获得更低延迟 ) pw.io.csv.write(table, "output.csv") pw.run()

替换数据源时只需要修改run内的取数逻辑(以及on_stop里的清理逻辑),其余——Schema 声明、提交节奏、输出方式——完全复用。仓库中另一篇教程 从 Web 抓取新闻的自定义连接器 就是以这个思路对接网页抓取场景的又一实例。

使用建议与注意事项

  • 无限流的中止:不要在无限流里期待run自行返回,务必像示例一样用on_stopdisconnect()之类的清理,并依靠CTRL+C/外部信号结束进程。
  • 提交节奏autocommit_duration_ms是延迟与批量的折中——追求低延迟可调小,追求吞吐可调大;不传则采用源码中的默认值1500毫秒(见 read 签名)。
  • 大批量突发数据:若数据源会一次性爆发海量消息,read还支持max_backlog_size参数限制积压、防止内存尖峰;设置后 Subject 内部的缓冲队列有界,投喂方法会在队列满时阻塞(详见 ConnectorSubject 类注释与 read 参数说明)。
  • Twitter API 现状:由于免费档流式 API 已关闭,运行本示例需要具备相应访问权限的 Token;把关注点放在"Subject + Schema + read + write"这套自定义接入模式上,才是本示例最值得沉淀的能力。

【免费下载链接】pathwayPython ETL framework for stream processing, real-time analytics, LLM pipelines, and RAG.项目地址: https://gitcode.com/GitHub_Trending/pa/pathway

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询