讲师中心 微信公众号
AI工具推荐 视频效率加速

如何应用自定义的限流迭代器(RateLimitingIterator)实战在数据遍历输出时进行秒级频次控制

落枫大大_5208

落枫大大_5208

发布时间:2026-05-26 13:44:20

|

303人浏览过

|

来源于php中文网

原创

RateLimitingIterator 是一个轻量级秒级限流迭代器,每秒最多处理 N 条数据,通过 time.time() 滑动控制间隔、自动 sleep 补时,支持任意可迭代对象,适用于 Tushare 等配额受限 API 的匀速调用。

直接用自定义的 ratelimitingiterator 控制数据遍历节奏,核心是把“取数据”和“等时间”封装进迭代逻辑里,让每次 next() 都自动守时。它不依赖外部中间件或全局状态,适合本地脚本、批量拉取、爬虫、对接有配额限制的 api(如 tushare、instagram、大模型接口)等场景。

设计一个轻量级秒级限流迭代器

目标:每秒最多处理 N 条数据,不堆积、不超发、不阻塞主线程逻辑。下面是一个基于时间戳滑动判断的简洁实现:

  • 使用 time.time() 记录上一次处理时间
  • 每次迭代前计算距上次的间隔,若不足 1/N 秒,则 sleep 补足
  • 支持传入任意可迭代对象(列表、生成器、数据库游标等)
  • 线程安全非必需,但加锁可适配多线程消费场景

示例代码:

import time
from typing import Iterator, Any
<p>class RateLimitingIterator:
def <strong>init</strong>(self, iterable: Iterator[Any], per_second: int = 1):
self.iterable = iter(iterable)
self.interval = 1.0 / max(1, per_second)  # 每条最小间隔(秒)
self.last_called = 0.0</p><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">def __iter__(self):
    return self

def __next__(self):
    now = time.time()
    elapsed = now - self.last_called
    if elapsed < self.interval:
        time.sleep(self.interval - elapsed)
    self.last_called = time.time()
    return next(self.iterable)</code>

实战:控制 Tushare 财务数据分批拉取频率

Tushare 免费用户常受限于「每分钟 60 次」或「单次最多 60 条」,用该迭代器可轻松拆解大批量日期范围,并匀速调用:

  • 先构造所有待查日期(如过去 5 年每季度首日)
  • 将日期列表传入 RateLimitingIterator,设 <code>per_second=1 → 实际约 60 次/分钟
  • 循环中每次只 fetch 一个日期的数据,天然规避并发超限

片段示意:

Evez Github Manager
Evez Github Manager

在 OpenClaw 中管理 GitHub 仓库、Pull Request、Issue 与工作流。适用于审查 PR、管理 Issue、自动化 GitHub Actions、同步仓库等场景。

下载
from datetime import datetime, timedelta
import tushare as ts
<h1>构造季度日期列表(2020Q1 ~ 2025Q1)</h1><p>dates = []
for year in range(2020, 2026):
for quarter in [1, 4, 7, 10]:
dt = datetime(year, quarter, 1)
dates.append(dt.strftime('%Y%m%d'))</p><h1>匀速遍历调用</h1><p>rl_iter = RateLimitingIterator(dates, per_second=1)
pro = ts.pro_api()</p><p>for trade_date in rl_iter:
try:
df = pro.fina_indicator(ts_code='000001.SZ', period=trade_date[:6])
print(f"Fetched {trade_date}: {len(df)} rows")
except Exception as e:
print(f"Failed on {trade_date}: {e}")

搭配异步协程提升吞吐(进阶用法)

若需更高效率又不突破频控,可将限流逻辑下沉到异步任务内部 —— 即每个协程自己守时,再用 asyncio.gather 并发多个低频通道:

  • 例如启动 3 个协程,每个按 0.5 QPS(即每 2 秒 1 次),整体稳定在 1.5 QPS
  • 比单线程串行快 3 倍,仍严格满足服务端限流要求
  • 避免因某次请求慢拖累全局节奏

关键点:限流器必须在协程内实例化,不可跨协程共享计时状态。

注意边界与容错细节

真实场景中容易忽略但影响稳定性的问题:

  • 首次启动偏差:初始化时 last_called=0 会导致第一条立即执行,建议设为 time.time() - interval
  • 系统时间跳变:NTP 校准可能使 time.time() 回跳,可改用 time.monotonic() 更鲁棒
  • 异常中断后恢复:若某次请求失败并跳过 next(),下次仍会补时;如需跳过限流,应在捕获异常后手动调用 next() 并丢弃
  • 精度取舍:time.sleep() 在 Linux 下通常误差 ±10ms,对秒级控制足够;毫秒级需用 asyncio.sleep() 或专用调度库

热门AI工具

更多
立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

创客贴AI

创客贴AI是一款AI图像与设计工具,AI辅助的智能在线设计工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

超级麦吉

一款通过数字员工协作完成调研、数据分析与文档制作的AI办公平台。

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

相关专题

更多
java
java

Java是一个通用术语,用于表示Java软件及其组件,包括“Java运行时环境 (JRE)”、“Java虚拟机 (JVM)”以及“插件”。php中文网还为大家带了Java相关下载资源、相关课程以及相关文章等内容,供大家免费下载使用。

11377

2023.06.15

java正则表达式语法
java正则表达式语法

java正则表达式语法是一种模式匹配工具,它非常有用,可以在处理文本和字符串时快速地查找、替换、验证和提取特定的模式和数据。本专题提供java正则表达式语法的相关文章、下载和专题,供大家免费下载体验。

8482

2023.07.05

java自学难吗
java自学难吗

Java自学并不难。Java语言相对于其他一些编程语言而言,有着较为简洁和易读的语法,本专题为大家提供java自学难吗相关的文章,大家可以免费体验。

7312

2023.07.31

java配置jdk环境变量
java配置jdk环境变量

Java是一种广泛使用的高级编程语言,用于开发各种类型的应用程序。为了能够在计算机上正确运行和编译Java代码,需要正确配置Java Development Kit(JDK)环境变量。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

1224

2023.08.01

java保留两位小数
java保留两位小数

Java是一种广泛应用于编程领域的高级编程语言。在Java中,保留两位小数是指在进行数值计算或输出时,限制小数部分只有两位有效数字,并将多余的位数进行四舍五入或截取。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

988

2023.08.02

java基本数据类型
java基本数据类型

java基本数据类型有:1、byte;2、short;3、int;4、long;5、float;6、double;7、char;8、boolean。本专题为大家提供java基本数据类型的相关的文章、下载、课程内容,供大家免费下载体验。

1456

2023.08.02

java有什么用
java有什么用

java可以开发应用程序、移动应用、Web应用、企业级应用、嵌入式系统等方面。本专题为大家提供java有什么用的相关的文章、下载、课程内容,供大家免费下载体验。

2989

2023.08.02

java在线网站
java在线网站

Java在线网站是指提供Java编程学习、实践和交流平台的网络服务。近年来,随着Java语言在软件开发领域的广泛应用,越来越多的人对Java编程感兴趣,并希望能够通过在线网站来学习和提高自己的Java编程技能。php中文网给大家带来了相关的视频、教程以及文章,欢迎大家前来学习阅读和下载。

20251

2023.08.03

Vite 项目如何打包和部署
Vite 项目如何打包和部署

Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

20

2026.10.28

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
vscode手册
vscode手册

共0课时 | 0人学习

Git 教程
Git 教程

共21课时 | 8.1万人学习

Git版本控制工具
Git版本控制工具

共8课时 | 1.9万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn