0

0

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

冰火之心

冰火之心

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

|

264人浏览过

|

来源于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 问题。

示例关键逻辑:

遨虾
遨虾

1688推出的跨境电商AI智能体

下载

立即学习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_performboost::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_clocknow() 可能跳变,导致桶“倒流”补令牌 → 务必只用 std::chrono::steady_clock

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

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

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

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

201

2023.11.20

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

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

723

2023.08.10

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

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

372

2025.12.24

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

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

27

2026.01.21

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

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

25

2026.01.21

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

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

101

2026.02.06

页面置换算法
页面置换算法

页面置换算法是操作系统中用来决定在内存中哪些页面应该被换出以便为新的页面提供空间的算法。本专题为大家提供页面置换算法的相关文章,大家可以免费体验。

483

2023.08.14

http500解决方法
http500解决方法

http500解决方法有检查服务器日志、检查代码错误、检查服务器配置、检查文件和目录权限、检查资源不足、更新软件版本、重启服务器或寻求专业帮助等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

482

2023.11.09

Golang 测试体系与代码质量保障:工程级可靠性建设
Golang 测试体系与代码质量保障:工程级可靠性建设

Go语言测试体系与代码质量保障聚焦于构建工程级可靠性系统。本专题深入解析Go的测试工具链(如go test)、单元测试、集成测试及端到端测试实践,结合代码覆盖率分析、静态代码扫描(如go vet)和动态分析工具,建立全链路质量监控机制。通过自动化测试框架、持续集成(CI)流水线配置及代码审查规范,实现测试用例管理、缺陷追踪与质量门禁控制,确保代码健壮性与可维护性,为高可靠性工程系统提供质量保障。

24

2026.02.28

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
C# 教程
C# 教程

共94课时 | 10.5万人学习

C 教程
C 教程

共75课时 | 5.1万人学习

C++教程
C++教程

共115课时 | 20万人学习

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号