在互联网应用中,流量洪峰是常有的事情。在应对流量洪峰时,通用的处理模式通常有排队、限流,这样能够很是直接有效的保护系统,防止系统被打爆。另外,经过限流技术手段,可让整个系统的运行更加平稳。今天要与你们分享一下限流算法和C#版本的组件。算法
1、令牌桶算法:网络
令牌桶算法的基本过程以下:ui
工做过程包括3个阶段:产生令牌、消耗令牌和判断数据包是否经过。其中涉及到2个参数:令牌产生的速率和令牌桶的大小,这个过程的具体工做以下。this
下面是C#的一个实现方式spa
class TokenBucketLimitingService: ILimitingService
{
private LimitedQueue<object> limitedQueue = null;
private CancellationTokenSource cancelToken;
private Task task = null;
private int maxTPS;
private int limitSize;
private object lckObj = new object();
public TokenBucketLimitingService(int maxTPS, int limitSize)
{
this.limitSize = limitSize;
this.maxTPS = maxTPS;线程if (this.limitSize <= 0)
this.limitSize = 100;
if(this.maxTPS <=0)
this.maxTPS = 1;3dlimitedQueue = new LimitedQueue<object>(limitSize);
for (int i = 0; i < limitSize; i++)
{
limitedQueue.Enqueue(new object());
}
cancelToken = new CancellationTokenSource();
task = Task.Factory.StartNew(new Action(TokenProcess), cancelToken.Token);
}对象/// <summary>
/// 定时消息令牌
/// </summary>
private void TokenProcess()
{
int sleep = 1000 / maxTPS;
if (sleep == 0)
sleep = 1;blogDateTime start = DateTime.Now;
while (cancelToken.Token.IsCancellationRequested ==false)
{
try
{
lock (lckObj)
{
limitedQueue.Enqueue(new object());
}
}
catch
{
}
finally
{
if (DateTime.Now - start < TimeSpan.FromMilliseconds(sleep))
{
int newSleep = sleep - (int)(DateTime.Now - start).TotalMilliseconds;
if (newSleep > 1)
Thread.Sleep(newSleep - 1); //作一下时间上的补偿
}
start = DateTime.Now;
}
}
}资源public void Dispose()
{
cancelToken.Cancel();
}/// <summary>
/// 请求令牌
/// </summary>
/// <returns>true:获取成功,false:获取失败</returns>
public bool Request()
{
if (limitedQueue.Count <= 0)
return false;
lock (lckObj)
{
if (limitedQueue.Count <= 0)
return false;object data = limitedQueue.Dequeue();
if (data == null)
return false;
}return true;
}
}
public interface ILimitingService:IDisposable
{
/// <summary>
/// 申请流量处理
/// </summary>
/// <returns>true:获取成功,false:获取失败</returns>
bool Request();
}
public class LimitingFactory
{
/// <summary>
/// 建立限流服务对象
/// </summary>
/// <param name="limitingType">限流模型</param>
/// <param name="maxQPS">最大QPS</param>
/// <param name="limitSize">最大可用票据数</param>
public static ILimitingService Build(LimitingType limitingType = LimitingType.TokenBucket, int maxQPS = 100, int limitSize = 100)
{
switch (limitingType)
{
case LimitingType.TokenBucket:
default:
return new TokenBucketLimitingService(maxQPS, limitSize);
case LimitingType.LeakageBucket:
return new LeakageBucketLimitingService(maxQPS, limitSize);
}
}
}
/// <summary>
/// 限流模式
/// </summary>
public enum LimitingType
{
TokenBucket,//令牌桶模式
LeakageBucket//漏桶模式
}
public class LimitedQueue<T> : Queue<T>
{
private int limit = 0;
public const string QueueFulled = "TTP-StreamLimiting-1001";public int Limit
{
get { return limit; }
set { limit = value; }
}public LimitedQueue()
: this(0)
{ }public LimitedQueue(int limit)
: base(limit)
{
this.Limit = limit;
}public new bool Enqueue(T item)
{
if (limit > 0 && this.Count >= this.Limit)
{
return false;
}
base.Enqueue(item);
return true;
}
}
调用方法:
var service = LimitingFactory.Build(LimitingType.TokenBucket, 500, 200);
while (true)
{
var result = service.Request();
//若是返回true,说明能够进行业务处理,不然须要继续等待
if (result)
{
//业务处理......
}
else
Thread.Sleep(1);
}
2、漏桶算法
声明一个固定容量的桶,每接受到一个请求向桶中添加一个令牌,当令牌桶达到上线后请求丢弃或等待,具体算法以下:
工做过程也包括3个阶段:产生令牌、消耗令牌和判断数据包是否经过。其中涉及到2个参数:令牌自动消费的速率和令牌桶的大小,个过程的具体工做以下。
C#的一个实现方式:
class LeakageBucketLimitingService: ILimitingService
{
private LimitedQueue<object> limitedQueue = null;
private CancellationTokenSource cancelToken;
private Task task = null;
private int maxTPS;
private int limitSize;
private object lckObj = new object();
public LeakageBucketLimitingService(int maxTPS, int limitSize)
{
this.limitSize = limitSize;
this.maxTPS = maxTPS;if (this.limitSize <= 0)
this.limitSize = 100;
if (this.maxTPS <= 0)
this.maxTPS = 1;limitedQueue = new LimitedQueue<object>(limitSize);
cancelToken = new CancellationTokenSource();
task = Task.Factory.StartNew(new Action(TokenProcess), cancelToken.Token);
}private void TokenProcess()
{
int sleep = 1000 / maxTPS;
if (sleep == 0)
sleep = 1;DateTime start = DateTime.Now;
while (cancelToken.Token.IsCancellationRequested == false)
{
try
{if (limitedQueue.Count > 0)
{
lock (lckObj)
{
if (limitedQueue.Count > 0)
limitedQueue.Dequeue();
}
}
}
catch
{
}
finally
{
if (DateTime.Now - start < TimeSpan.FromMilliseconds(sleep))
{
int newSleep = sleep - (int)(DateTime.Now - start).TotalMilliseconds;
if (newSleep > 1)
Thread.Sleep(newSleep - 1); //作一下时间上的补偿
}
start = DateTime.Now;
}
}
}public void Dispose()
{
cancelToken.Cancel();
}public bool Request()
{
if (limitedQueue.Count >= limitSize)
return false;
lock (lckObj)
{
if (limitedQueue.Count >= limitSize)
return false;return limitedQueue.Enqueue(new object());
}
}
}
调用方法:
var service = LimitingFactory.Build(LimitingType.LeakageBucket, 500, 200);
while (true)
{
var result = service.Request();
//若是返回true,说明能够进行业务处理,不然须要继续等待
if (result)
{
//业务处理......
}
else
Thread.Sleep(1);
}
两类限流算法虽然很是类似,可是仍是有些区别的,供你们参考!