15行Python代码带你轻松理解令牌桶算法


Posted in Python onMarch 21, 2018

在网络中传输数据时,为了防止网络拥塞,需限制流出网络的流量,使流量以比较均匀的速度向外发送,令牌桶算法就实现了这个功能, 可控制发送到网络上数据的数目,并允许突发数据的发送。

什么是令牌

从名字上看令牌桶,大概就是一个装有令牌的桶吧,那么什么是令牌呢?

紫薇格格拿的令箭,可以发号施令,令行禁止。在计算机的世界中,令牌也有令行禁止的意思,有令牌,则相当于得到了进行操作的授权,没有令牌,就什么都不能做。

用令牌实现限速器

我们用1块令牌来代表发送1字节数据的资格,假设我们源源不断的发放令牌给程序,程序就有资格源源不断的发送数据,当我们不发放令牌给程序,程序就相当于被限流,无法发送数据了。接下来我们说说限速器,所谓限速器,就是让程序在单位时间内,最多只能发送一定大小的数据。假设在1秒发放10块令牌,那么程序发送数据的速度就会被限制在10bytes/s。如果1秒内有大于10bytes的数据需要发送,就会因为没有令牌而被丢弃。

改进限速器——加个桶

15行Python代码带你轻松理解令牌桶算法 

我们实现的限速器,速度是恒定的,但是在实际的应用中,往往会有突发的传输需求(需要更快速的发送,但是不会持续太久,也不会引起网络拥塞),这种数据碰上我们的限速器,就因为拿不到令牌而无法发送。

对限速器进行一下改动,依然1秒产生10块令牌,但是我们把产生出来的令牌先放到一个桶里,当程序需要发送的时候,从桶里取令牌,不需要的时候,令牌就会在桶里沉淀下来,假设桶里沉淀了10块令牌,程序最多就可以在1秒内发送20bytes的数据,满足了突发数据传输的要求,并且由于桶里的令牌被用完,下一秒最多依然只能发10bytes的数据,不会因为持续发送大量数据,对网络造成压力。

15行Python代码带你轻松理解令牌桶算法 

15行Python代码实践令牌桶

令牌桶需要以一定的速度生成令牌放入桶中,当程序要发送数据时,再从桶中取出令牌。这里似乎有点问题,如果我们使用一个死循环,来不停地发放令牌,程序就被阻塞住了,有没有更好的办法?

我们可以在取令牌的时候,用现在的时间减去上次取令牌的时间,乘以令牌的发放速度,计算出桶里可以取的令牌数量(当然不能超过桶的大小),从而避免循环发放的逻辑。

接下来看代码:

import time
class TokenBucket(object):
 # rate是令牌发放速度,capacity是桶的大小
 def __init__(self, rate, capacity):
  self._rate = rate
  self._capacity = capacity
  self._current_amount = 0
  self._last_consume_time = int(time.time())
 # token_amount是发送数据需要的令牌数
 def consume(self, token_amount):
  increment = (int(time.time()) - self._last_consume_time) * self._rate # 计算从上次发送到这次发送,新发放的令牌数量
  self._current_amount = min(
   increment + self._current_amount, self._capacity) # 令牌数量不能超过桶的容量
  if token_amount > self._current_amount: # 如果没有足够的令牌,则不能发送数据
   return False
  self._last_consume_time = int(time.time())
  self._current_amount -= token_amount
  return True

总结

以上所述是小编给大家介绍的15行Python代码带你轻松理解令牌桶算法,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对三水点靠木网站的支持!

Python 相关文章推荐
Python实现的检测web服务器健康状况的小程序
Sep 17 Python
在Django框架中伪造捕捉到的URLconf值的方法
Jul 18 Python
深入剖析Python的爬虫框架Scrapy的结构与运作流程
Jan 20 Python
Python获取文件所在目录和文件名的方法
Jan 12 Python
对python中的xlsxwriter库简单分析
May 04 Python
pycharm中使用anaconda部署python环境的方法步骤
Dec 19 Python
python实现海螺图片的方法示例
May 12 Python
Python调用.NET库的方法步骤
Dec 27 Python
python多线程使用方法实例详解
Dec 30 Python
Python关于反射的实例代码分享
Feb 20 Python
Python3使用xlrd、xlwt处理Excel方法数据
Feb 28 Python
python标准库ElementTree处理xml
May 20 Python
Python反转序列的方法实例分析
Mar 21 #Python
Python常见工厂函数用法示例
Mar 21 #Python
python如何统计序列中元素
Jul 31 #Python
python如何读写json数据
Mar 21 #Python
python如何读写csv数据
Mar 21 #Python
python如何让类支持比较运算
Mar 20 #Python
python如何为创建大量实例节省内存
Mar 20 #Python
You might like
基于empty函数的判断详解
2013/06/17 PHP
PHPstorm快捷键(分享)
2017/07/17 PHP
学习thinkphp5.0验证类使用方法
2017/11/16 PHP
laravel框架select2多选插件初始化默认选中项操作示例
2020/02/18 PHP
用javascript父窗口控制只弹出一个子窗口
2007/04/10 Javascript
javascript getElementsByClassName函数
2010/04/01 Javascript
面向对象的Javascript之一(初识Javascript)
2012/01/20 Javascript
UI Events 用户界面事件
2012/06/27 Javascript
为JS扩展Array.prototype.indexOf引发的问题探讨及解决
2013/04/24 Javascript
基于Jquery实现仿百度百科右侧导航代码附源码下载
2015/11/27 Javascript
解决JavaScript数字精度丢失问题的方法
2015/12/03 Javascript
浅析Bootstrap验证控件的使用
2016/06/23 Javascript
浅谈JS继承_寄生式继承 & 寄生组合式继承
2016/08/16 Javascript
JS如何设置iOS中微信浏览器的title
2016/11/22 Javascript
JS+CSS实现下拉刷新/上拉加载插件
2017/03/31 Javascript
element UI upload组件上传附件格式限制方法
2018/09/04 Javascript
小程序接口的promise化的实现方法
2019/12/11 Javascript
vue 微信扫码登录(自定义样式)
2020/01/06 Javascript
Python文件处理
2016/02/29 Python
python中执行shell的两种方法总结
2017/01/10 Python
详解python中的文件与目录操作
2017/07/11 Python
Scrapy的简单使用教程
2017/10/24 Python
用python实现百度翻译的示例代码
2018/03/09 Python
python中将一个全部为int的list 转化为str的list方法
2018/04/09 Python
python使用selenium实现批量文件下载
2019/03/11 Python
python中with语句结合上下文管理器操作详解
2019/12/19 Python
使用matplotlib的pyplot模块绘图的实现示例
2020/07/12 Python
详解Python模块化编程与装饰器
2021/01/16 Python
基于PyInstaller各参数的含义说明
2021/03/04 Python
我们在web应用开发过程中经常遇到输出某种编码的字符,如iso8859-1等,如何输出一个某种编码的字符串?
2014/03/30 面试题
毕业生自荐信的主要内容
2013/10/29 职场文书
店铺转让协议书
2015/01/29 职场文书
4S店收银员岗位职责
2015/04/07 职场文书
幼儿园2016圣诞节活动总结
2016/03/31 职场文书
MySQL分区表实现按月份归类
2021/11/01 MySQL
星际争霸:毕姥爷vs解冻01
2022/04/01 星际争霸