/// <summary>
/// Limits the bandwith used by the subsequent stages in the owin pipeline.
/// </summary>
/// <param name="getMaxBytesPerSecond">
/// A delegate to retrieve the maximum number of bytes per second to be transferred.
/// Allows you to supply different values at runtime. Use 0 or a negative number to specify infinite bandwidth.
/// </param>
/// <returns>An OWIN middleware delegate.</returns>
/// <exception cref="System.ArgumentNullException">getMaxBytesPerSecond</exception>
public static MidFunc MaxBandwidthPerRequest(Func<RequestContext, int> getMaxBytesPerSecond)
{
getMaxBytesPerSecond.MustNotNull("getMaxBytesPerSecond");
return
next =>
async env =>
{
var context = new OwinContext(env);
Stream requestBodyStream = context.Request.Body ?? Stream.Null;
Stream responseBodyStream = context.Response.Body;
var limitsRequestContext = new RequestContext(context.Request);
var requestTokenBucket = new FixedTokenBucket(
() => getMaxBytesPerSecond(limitsRequestContext));
var responseTokenBucket = new FixedTokenBucket(
() => getMaxBytesPerSecond(limitsRequestContext));
using (requestTokenBucket.RegisterRequest())
using (responseTokenBucket.RegisterRequest())
{
context.Request.Body = new ThrottledStream(requestBodyStream, requestTokenBucket);
context.Response.Body = new ThrottledStream(responseBodyStream, responseTokenBucket);
//TODO consider SendFile interception
await next(env).ConfigureAwait(false);
}
};
}
/// <summary>
/// Limits the bandwith used by the subsequent stages in the owin pipeline.
/// </summary>
/// <param name="getBytesPerSecond">
/// A delegate to retrieve the maximum number of bytes per second to be transferred.
/// Allows you to supply different values at runtime. Use 0 or a negative number to specify infinite bandwidth.
/// </param>
/// <param name="loggerName">(Optional) The name of the logger log messages are written to.</param>
/// <returns>An OWIN middleware delegate.</returns>
/// <exception cref="System.ArgumentNullException">getMaxBytesToWrite</exception>
public static MidFunc MaxBandwidthGlobal(Func<int> getBytesPerSecond, string loggerName = null)
{
getBytesPerSecond.MustNotNull("getMaxBytesToWrite");
loggerName = string.IsNullOrWhiteSpace(loggerName)
? "LimitsMiddleware.MaxBandwidthGlobal"
: loggerName;
var logger = LogProvider.GetLogger(loggerName);
var requestTokenBucket = new FixedTokenBucket(getBytesPerSecond);
var responseTokenBucket = new FixedTokenBucket(getBytesPerSecond);
logger.Debug("Configure streams to be globally limited to.");
return
next =>
async env =>
{
using (requestTokenBucket.RegisterRequest())
using (responseTokenBucket.RegisterRequest())
{
var context = new OwinContext(env);
Stream requestBodyStream = context.Request.Body ?? Stream.Null;
Stream responseBodyStream = context.Response.Body;
context.Request.Body = new ThrottledStream(requestBodyStream, requestTokenBucket);
context.Response.Body = new ThrottledStream(responseBodyStream, responseTokenBucket);
//TODO consider SendFile interception
await next(env).ConfigureAwait(false);
}
};
}
public ThrottledStream(
Stream innerStream,
FixedTokenBucket fixedTokenBucket)
{
innerStream.MustNotNull("innerStream");
_innerStream = innerStream;
_tokenBucket = fixedTokenBucket;
}
public FixedTokenBucketTests()
{
_bucket = new FixedTokenBucket(() => MaxTokens, () => _getUtcNow());
}
public FixedTokenBucketTests()
{
_bucket = new FixedTokenBucket(() => MaxTokens, () => _getUtcNow());
}
