Apply Mike's TokenBucket fix

avinationmerge
Melanie Thielker 2010-05-28 02:44:12 +02:00
parent ba49319c07
commit 66b4309f34
1 changed files with 30 additions and 42 deletions

View File

@ -133,7 +133,7 @@ namespace OpenSim.Region.ClientStack.LindenUDP
this.parent = parent; this.parent = parent;
MaxBurst = maxBurst; MaxBurst = maxBurst;
DripRate = dripRate; DripRate = dripRate;
lastDrip = Environment.TickCount & Int32.MaxValue; lastDrip = Environment.TickCount;
} }
/// <summary> /// <summary>
@ -143,41 +143,31 @@ namespace OpenSim.Region.ClientStack.LindenUDP
/// <returns>True if the requested number of tokens were removed from /// <returns>True if the requested number of tokens were removed from
/// the bucket, otherwise false</returns> /// the bucket, otherwise false</returns>
public bool RemoveTokens(int amount) public bool RemoveTokens(int amount)
{
bool dummy;
return RemoveTokens(amount, out dummy);
}
/// <summary>
/// Remove a given number of tokens from the bucket
/// </summary>
/// <param name="amount">Number of tokens to remove from the bucket</param>
/// <param name="dripSucceeded">True if tokens were added to the bucket
/// during this call, otherwise false</param>
/// <returns>True if the requested number of tokens were removed from
/// the bucket, otherwise false</returns>
public bool RemoveTokens(int amount, out bool dripSucceeded)
{ {
if (maxBurst == 0) if (maxBurst == 0)
{ {
dripSucceeded = true;
return true; return true;
} }
dripSucceeded = Drip(); if (amount > maxBurst)
if (content - amount >= 0)
{ {
if (parent != null && !parent.RemoveTokens(amount)) throw new Exception("amount " + amount + " exceeds maxBurst " + maxBurst);
return false;
content -= amount;
return true;
} }
else
Drip();
if (content < amount)
{ {
return false; return false;
} }
if (parent != null && !parent.RemoveTokens(amount))
{
return false;
}
content -= amount;
return true;
} }
/// <summary> /// <summary>
@ -193,25 +183,23 @@ namespace OpenSim.Region.ClientStack.LindenUDP
content = maxBurst; content = maxBurst;
return true; return true;
} }
else
int now = Environment.TickCount;
int deltaMS = now - lastDrip;
lastDrip = now;
if (deltaMS <= 0)
{ {
int now = Environment.TickCount & Int32.MaxValue; return false;
int deltaMS = now - lastDrip;
if (deltaMS <= 0)
{
if (deltaMS < 0)
lastDrip = now;
return false;
}
int dripAmount = deltaMS * tokensPerMS;
content = Math.Min(content + dripAmount, maxBurst);
lastDrip = now;
return true;
} }
long dripAmount = (long)deltaMS * (long)tokensPerMS + (long)content;
if (dripAmount > maxBurst)
{
dripAmount = maxBurst;
}
content = (int)dripAmount;
return true;
} }
} }
} }