U
.Wj$ @ s d Z ddlmZ ddlmZ ddlmZmZ ddlmZ G dd dZ G dd d eZ
ee
G d
d d Z G d d
d
e Z G dd de Z
G dd dejZG dd deZG dd dZdS )a
Hierarchical Token Bucket traffic shaping.
Patterned after U{Martin Devera's Hierarchical Token Bucket traffic
shaper for the Linux kernel}.
@seealso: U{HTB Linux queuing discipline manual - user guide
}
@seealso: U{Token Bucket Filter in Linux Advanced Routing & Traffic Control
HOWTO}
)time)Optional) Interface implementer)pcpc @ sP e Zd ZU dZdZee ed< dZee ed< dZ d ddZ
dd Z d
d Z dS )
Bucketa
Implementation of a Token bucket.
A bucket can hold a certain number of tokens and it drains over time.
@cvar maxburst: The maximum number of tokens that the bucket can
hold at any given time. If this is L{None}, the bucket has
an infinite size.
@type maxburst: C{int}
@cvar rate: The rate at which the bucket drains, in number
of tokens per second. If the rate is L{None}, the bucket
drains instantaneously.
@type rate: C{int}
Nmaxburstrater c C s d| _ || _t | _dS )a(
Create a L{Bucket} that may have a parent L{Bucket}.
@param parentBucket: If a parent Bucket is specified,
all L{add} and L{drip} operations on this L{Bucket}
will be applied on the parent L{Bucket} as well.
@type parentBucket: L{Bucket}
r N)content parentBucketr lastDrip)selfr r `c:\users\psant\downloads\hotstarbackend\school\venv38\Lib\site-packages\twisted/protocols/htb.py__init__3 s zBucket.__init__c C sR | | jdkr|}nt|| j| j }| jdk r@| j|}| j|7 _|S )at
Adds tokens to the L{Bucket} and its C{parentBucket}.
This will add as many of the C{amount} tokens as will fit into both
this L{Bucket} and its C{parentBucket}.
@param amount: The number of tokens to try to add.
@type amount: C{int}
@returns: The number of tokens that actually fit.
@returntype: C{int}
N)dripr minr
r add)r
amountZ allowabler r r r @ s
z
Bucket.addc C sb | j dk r| j | jdkr&d| _n2t }|| j }|| j }td| j| | _|| _| jdkS )z
Let some of the bucket drain.
The L{Bucket} drains at the rate specified by the class
variable C{rate}.
@returns: C{True} if the bucket is empty after this drip.
@returntype: C{bool}
Nr )r r r r
r r max)r
nowZ deltaTimeZ deltaTokensr r r r X s
z Bucket.drip)N)
__name__
__module__ __qualname____doc__r r int__annotations__r _refcountr r r r r r r r s
r c @ s e Zd Zdd ZdS )
IBucketFilterc O s dS )zn
Return a L{Bucket} corresponding to the provided parameters.
@returntype: L{Bucket}
Nr )Z
somethingsZsome_kwr r r getBucketForq s zIBucketFilter.getBucketForN)r r r r r r r r r p s r c @ sH e Zd ZU dZeZdZee e d< d ddZ
dd Z dd Z d
d Z
dS )
HierarchicalBucketFilterz
Filter things into buckets that can be nested.
@cvar bucketFactory: Class of buckets to make.
@type bucketFactory: L{Bucket}
@cvar sweepInterval: Seconds between sweeping out the bucket cache.
@type sweepInterval: C{int}
N
sweepIntervalc C s i | _ || _t | _d S N)buckets parentFilterr lastSweep)r
r$ r r r r s z!HierarchicalBucketFilter.__init__c O s| | j dk r$t | j | j kr$| | jr@| jj| f||}nd}| j||}| j|}|dkrx| |}|| j|< |S )z
Find or create a L{Bucket} corresponding to the provided parameters.
Any parameters are passed on to L{getBucketKey}, from them it
decides which bucket you get.
@returntype: L{Bucket}
N)
r! r r% sweepr$ r getBucketKeyr# get
bucketFactory)r
akwr keybucketr r r r s
z%HierarchicalBucketFilter.getBucketForc O s dS )a%
Construct a key based on the input parameters to choose a L{Bucket}.
The default implementation returns the same key for all
arguments. Override this method to provide L{Bucket} selection.
@returns: Something to be used as a key in the bucket cache.
Nr )r
r* r+ r r r r' s z%HierarchicalBucketFilter.getBucketKeyc C s> | j D ]&\}}| }|jdkr
|r
| j |= q
t | _dS )z'
Remove empty buckets.
r N)r# itemsr r r r% )r
r, r- Zbucket_is_emptyr r r r& s
z HierarchicalBucketFilter.sweep)N)r r r r r r) r! r r r r r r' r& r r r r r y s
r c @ s e Zd ZdZdZdd ZdS ) FilterByHostzF
A Hierarchical Bucket filter with a L{Bucket} for each host.
i c C s | d S N )ZgetPeerr
transportr r r r' s zFilterByHost.getBucketKeyNr r r r r! r' r r r r r/ s r/ c @ s e Zd ZdZdZdd ZdS )FilterByServerzI
A Hierarchical Bucket filter with a L{Bucket} for each service.
Nc C s | d S )N )ZgetHostr2 r r r r' s zFilterByServer.getBucketKeyr4 r r r r r5 s r5 c @ s, e Zd ZdZdZdd Zdd Zdd Zd S )
ShapedConsumerzL
Wraps a C{Consumer} and shapes the rate at which it receives data.
Fc C s( t j| | || _| j jd7 _d S r0 )r ProducerConsumerProxyr r- r )r
consumerr- r r r r s zShapedConsumer.__init__c C s& | j t|}tj| |d | S r" )r- r lenr r8 _writeSomeData)r
datar r r r r; s z ShapedConsumer._writeSomeDatac C s t j| | j jd8 _d S r0 )r r8
stopProducingr- r )r
r r r r= s z ShapedConsumer.stopProducingN)r r r r iAmStreamingr r; r= r r r r r7 s
r7 c @ s e Zd ZdZdZdd ZdS )ShapedTransportaR
Wraps a C{Transport} and shapes the rate at which it receives data.
This is a L{ShapedConsumer} with a little bit of magic to provide for
the case where the consumer it wraps is also a C{Transport} and people
will be attempting to access attributes this does not proxy as a
C{Consumer} (e.g. C{loseConnection}).
Fc C s t | j|S r" )getattrr9 )r
namer r r __getattr__ s zShapedTransport.__getattr__N)r r r r r> rB r r r r r? s r? c @ s e Zd ZdZdd Zdd ZdS )ShapedProtocolFactorya
Dispense C{Protocols} with traffic shaping on their transports.
Usage::
myserver = SomeFactory()
myserver.protocol = ShapedProtocolFactory(myserver.protocol,
bucketFilter)
Where C{SomeServerFactory} is a L{twisted.internet.protocol.Factory}, and
C{bucketFilter} is an instance of L{HierarchicalBucketFilter}.
c C s || _ || _dS )a
Tell me what to wrap and where to get buckets.
@param protoClass: The class of C{Protocol} this will generate
wrapped instances of.
@type protoClass: L{Protocol}
class
@param bucketFilter: The filter which will determine how
traffic is shaped.
@type bucketFilter: L{HierarchicalBucketFilter}.
N)protocol bucketFilter)r
Z
protoClassrE r r r r s z ShapedProtocolFactory.__init__c s* j ||}|j fdd}||_|S )z
Make a C{Protocol} instance with a shaped transport.
Any parameters will be passed on to the protocol's initializer.
@returns: A C{Protocol} instance with a L{ShapedTransport}.
c s j | }t| |} |S r" )rE r r? )r3 r- ZshapedTransportZorigMakeConnectionr
r r makeConnection, s
z6ShapedProtocolFactory.__call__..makeConnection)rD rG )r
r* r+ protorG r rF r __call__! s
z ShapedProtocolFactory.__call__N)r r r r r rI r r r r rC s
rC N)r r typingr Zzope.interfacer r Ztwisted.protocolsr r r r r/ r5 r8 r7 r? rC r r r r s R C