52 lines
1.6 KiB
C#
52 lines
1.6 KiB
C#
using System;
|
||
using System.Collections.Generic;
|
||
using System.Threading;
|
||
|
||
namespace Common
|
||
{
|
||
public class RateLimiter
|
||
{
|
||
private static readonly object lockObject = new object();
|
||
private static readonly int requestsPerSecond = 5;//每秒允许请求5次
|
||
private static int remainingRequests;//剩余请求次数
|
||
private static long lastRequestTime;//上次请求时间
|
||
/// <summary>
|
||
/// 实现速率限制。它根据指定的每秒请求数量来控制请求是否被允许
|
||
/// </summary>
|
||
static RateLimiter()
|
||
{
|
||
remainingRequests = requestsPerSecond;
|
||
lastRequestTime = DateTime.UtcNow.Ticks;
|
||
}
|
||
/// <summary>
|
||
/// 实现速率限制：默认每秒允许请求5次
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
public static bool TryRequest()
|
||
{
|
||
lock (lockObject)
|
||
{
|
||
long currentTime = DateTime.UtcNow.Ticks;//当前请求时间
|
||
long elapsedTicks = currentTime - lastRequestTime;
|
||
double elapsedSeconds = elapsedTicks / TimeSpan.TicksPerSecond;
|
||
|
||
remainingRequests += (int)(elapsedSeconds * requestsPerSecond);
|
||
|
||
if (remainingRequests > requestsPerSecond)
|
||
{
|
||
remainingRequests = requestsPerSecond;
|
||
}
|
||
|
||
if (remainingRequests > 0)
|
||
{
|
||
remainingRequests--;
|
||
lastRequestTime = currentTime;
|
||
return true;
|
||
}
|
||
|
||
return false;
|
||
}
|
||
}
|
||
}
|
||
}
|