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

C++如何实现带限流的API调用客户端?(令牌桶算法)

胖宇姑娘_8387

胖宇姑娘_8387

发布时间:2026-03-01 15:36:02

|

284人浏览过

|

来源于php中文网

原创

令牌桶比简单计时器更适合API限流,因其能平滑吸收突发流量并保证长期速率不超限:桶容量控制最大突发量,填充速率控制长期均值,且不依赖固定时间窗口切分。

c++如何实现带限流的api调用客户端?(令牌桶算法)

为什么令牌桶比简单计时器更适合API限流

因为真实API调用有突发性,而单纯用 std::chrono::steady_clock 做固定窗口计数(比如“每秒最多10次”)会导致临界时刻被打穿——前一秒末尾发10次,后一秒开头又发10次,实际20次/秒。令牌桶能平滑吸收突发,同时保证长期速率不超限。

核心在于:它维护一个随时间匀速填充的“桶”,每次调用前尝试取走一个令牌;桶空则阻塞或拒绝,不依赖窗口切分。

  • 桶容量 = 最大允许突发请求数(如 burst_size = 5)
  • 填充速率 = 单位时间补充令牌数(如 10.0 / std::chrono::seconds(1))
  • 必须用 std::atomic 或互斥锁保护桶状态,多线程下调用客户端时不然会丢令牌

如何用 std::atomic 实现无锁令牌桶(C++20起推荐)

C++20 的 std::atomic<float></float> 支持浮点原子操作,可直接存当前令牌数;但更稳妥的做法是用 std::atomic<int64_t></int64_t> 存纳秒级“下次可取令牌时间戳”,避免浮点精度漂移和 ABA 问题。

示例关键逻辑:

C++14
C++14

C++14 对 C++11 的修正与增强版本,适合旧系统维护和较老工具链兼容。

下载

立即学习“C++免费学习笔记(深入)”;

class TokenBucket {
    std::atomic<int64_t> next_refill_time_{0}; // 下次能取令牌的时间点(纳秒)
    const double rate_per_sec_;
    const int burst_size_;
    const std::chrono::nanoseconds refill_interval_;

public:
    TokenBucket(double rate_per_sec, int burst_size)
        : rate_per_sec_(rate_per_sec), burst_size_(burst_size),
          refill_interval_(std::chrono::nanoseconds(
              static_cast<int64_t>(1e9 / rate_per_sec))) {}

    bool try_acquire() {
        auto now = std::chrono::steady_clock::now().time_since_epoch().count();
        auto expected = next_refill_time_.load();
        while (true) {
            if (now >= expected) {
                // 可以取走一个,更新下次时间(+ refill_interval_)
                auto desired = expected + refill_interval_.count();
                if (next_refill_time_.compare_exchange_weak(expected, desired))
                    return true;
                // CAS失败:说明其他线程已更新,重试
                continue;
            }
            // 还没到时间,检查是否在burst范围内(即桶里还有“预存”额度)
            // 这里简化处理:只允许一次立即获取,否则返回false
            return false;
        }
    }
};
  • 注意 refill_interval_ 是倒算出来的,不是每周期 sleep;靠时间戳比较驱动,无忙等
  • compare_exchange_weak 必须循环使用,单次失败不等于不可用
  • 该实现不自动填充“历史欠额”,适合严格速率控制;若需支持突发后缓慢恢复,得额外记录当前令牌数 std::atomic<int></int>

集成到HTTP客户端时怎么避免阻塞整个请求线程

直接在 curl_easy_perform 或 boost::beast::http::async_write 前调用 try_acquire() 并 sleep 等待,会卡死线程——尤其在异步客户端里完全不可接受。

  • 正确做法:把限流判断做成前置钩子,失败时返回 std::nullopt 或抛出 std::system_error(带 std::errc::resource_unavailable_try_again),由上层决定重试策略
  • 不要在限流逻辑里调用 std::this_thread::sleep_for;等待应交给调度器(如 asio::steady_timer)或业务层退避
  • 如果用 libcurl 多线程,确保每个线程独享一个 TokenBucket 实例,或加 std::mutex ——原子版虽快,但高竞争下 CAS 自旋开销也不小

测试时最容易漏掉的边界情况

单元测试常只验证“匀速调用不被限”,但线上出问题多在边界上:

  • rate_per_sec 设为 0.1(即每10秒1次)时,refill_interval_.count() 可能截断为0,导致永远无法填充 → 必须对 refill_interval_ 做最小值约束(如不低于1ms)
  • 程序启动瞬间大量请求涌入,next_refill_time_ 初始为0,首次调用总成功,但可能瞬间耗尽 burst_size → 要么启动时预设一个合理初始时间,要么在 try_acquire 中加入“首次填充”逻辑
  • 系统时间被NTP校准回拨,steady_clock 不受影响,但若误用 system_clock,now() 可能跳变,导致桶“倒流”补令牌 → 务必只用 std::chrono::steady_clock

真正难调的不是算法本身,而是时间语义、原子操作顺序、以及 burst 和 rate 在不同负载下的耦合表现。上线前至少用压测工具模拟 3 种节奏:匀速、脉冲、抖动,观察响应延迟和 429 比例是否符合预期。

热门AI工具

更多
咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

DeepSeek

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

WorkBuddy

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

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

Lovart
Lovart Hot

一款面向视觉设计创作的AI设计平台,可通过智能体和画布工作流辅助制作海报、Logo、网页、PPT及其他视觉内容。

豆包大模型

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

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

PixPix
PixPix Hot

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

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

相关专题

更多
counta和count的区别
counta和count的区别

Count函数用于计算指定范围内数字的个数,而CountA函数用于计算指定范围内非空单元格的个数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

3208

2023.11.20

线程和进程的区别
线程和进程的区别

线程和进程的区别:线程是进程的一部分,用于实现并发和并行操作,而线程共享进程的资源,通信更方便快捷,切换开销较小。本专题为大家提供线程和进程区别相关的各种文章、以及下载和课程。

4238

2023.08.10

Python 多线程与异步编程实战
Python 多线程与异步编程实战

本专题系统讲解 Python 多线程与异步编程的核心概念与实战技巧,包括 threading 模块基础、线程同步机制、GIL 原理、asyncio 异步任务管理、协程与事件循环、任务调度与异常处理。通过实战示例,帮助学习者掌握 如何构建高性能、多任务并发的 Python 应用。

1189

2025.12.24

java多线程相关教程合集
java多线程相关教程合集

本专题整合了java多线程相关教程,阅读专题下面的文章了解更多详细内容。

403

2026.01.21

C++多线程相关合集
C++多线程相关合集

本专题整合了C++多线程相关教程,阅读专题下面的的文章了解更多详细内容。

559

2026.01.21

C# 多线程与异步编程
C# 多线程与异步编程

本专题深入讲解 C# 中多线程与异步编程的核心概念与实战技巧,包括线程池管理、Task 类的使用、async/await 异步编程模式、并发控制与线程同步、死锁与竞态条件的解决方案。通过实际项目,帮助开发者掌握 如何在 C# 中构建高并发、低延迟的异步系统,提升应用性能和响应速度。

389

2026.02.06

C++多线程并发控制与线程安全设计实践
C++多线程并发控制与线程安全设计实践

本专题围绕 C++ 在高性能系统开发中的并发控制技术展开,系统讲解多线程编程模型与线程安全设计方法。内容包括互斥锁、读写锁、条件变量、原子操作以及线程池实现机制,同时结合实际案例分析并发竞争、死锁避免与性能优化策略。通过实践讲解,帮助开发者掌握构建稳定高效并发系统的关键技术。

233

2026.03.16

Java 多线程与并发编程实战
Java 多线程与并发编程实战

深入讲解 Java 并发编程体系,涵盖 Thread / Runnable / Callable 线程创建方式、线程生命周期与状态转换、synchronized 同步锁与 Lock/ReentrantLock 显式锁、volatile 可见性保证、ThreadPool 线程池配置与调优(核心参数/拒绝策略)、JUC 并发工具类(CountDownLatch/CyclicBarrier/Semaphore/ConcurrentHashMa

349

2026.04.17

Playwright 测试失败怎么办
Playwright 测试失败怎么办

Playwright测试失败专题汇集超时、定位器匹配、浏览器提前关闭、依赖与插件配置、文件上传、SSL及Docker启动等故障文章,并介绍通过截图、报告和Trace定位错误。

0

2026.10.21

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
C++ Core Guidelines
C++ Core Guidelines

共0课时 | 0人学习

C++ Reference
C++ Reference

共0课时 | 0人学习

C++ 官方标准说明
C++ 官方标准说明

共0课时 | 0人学习

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

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