DysonNetwork/Swarm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

⚡ Add rate limiter

Browse Source
This commit is contained in:
LittleSheep 2025-04-26 12:24:32 +08:00
parent 38b7e8c1a1
commit 3080e273cb
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
DysonNetwork.Sphere/Program.cs
View File

@ -2,6 +2,7 @@ using System.Net;
using System.Security.Cryptography; using System.Security.Cryptography;
using System.Text; using System.Text;
using System.Text.Json; using System.Text.Json;
using System.Threading.RateLimiting;
using Casbin; using Casbin;
using Casbin.Persist.Adapter.EFCore; using Casbin.Persist.Adapter.EFCore;
using DysonNetwork.Sphere; using DysonNetwork.Sphere;
@ -12,6 +13,7 @@ using DysonNetwork.Sphere.Storage;
using Microsoft.AspNetCore.Authorization; using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.HttpOverrides; using Microsoft.AspNetCore.HttpOverrides;
using Microsoft.AspNetCore.Mvc; using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.RateLimiting;
using Microsoft.EntityFrameworkCore; using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Options; using Microsoft.Extensions.Options;
using Microsoft.IdentityModel.Tokens; using Microsoft.IdentityModel.Tokens;
@ -61,6 +63,13 @@ builder.Services.AddSingleton<IAuthorizationHandler, CasbinAuthorizationHandler>
// Other pipelines // Other pipelines
builder.Services.AddRateLimiter(o => o.AddFixedWindowLimiter(policyName: "fixed", opts =>
{
opts.Window = TimeSpan.FromMinutes(1);
opts.PermitLimit = 120;
opts.QueueLimit = 2;
opts.QueueProcessingOrder = QueueProcessingOrder.OldestFirst;
}));
builder.Services.AddCors(); builder.Services.AddCors();
builder.Services.AddAuthorization(); builder.Services.AddAuthorization();
builder.Services.AddAuthentication("Bearer").AddJwtBearer(options => builder.Services.AddAuthentication("Bearer").AddJwtBearer(options =>
@ -175,13 +184,14 @@ app.UseCors(opts =>
.AllowAnyMethod() .AllowAnyMethod()
); );
app.UseRateLimiter();
app.UseHttpsRedirection(); app.UseHttpsRedirection();
app.UseAuthorization(); app.UseAuthorization();
app.UseMiddleware<UserInfoMiddleware>(); app.UseMiddleware<UserInfoMiddleware>();
app.MapControllers(); app.MapControllers().RequireRateLimiting("fixed");
app.MapStaticAssets(); app.MapStaticAssets().RequireRateLimiting("fixed");
app.MapRazorPages(); app.MapRazorPages().RequireRateLimiting("fixed");
var tusDiskStore = new tusdotnet.Stores.TusDiskStore( var tusDiskStore = new tusdotnet.Stores.TusDiskStore(
builder.Configuration.GetSection("Tus").GetValue<string>("StorePath")! builder.Configuration.GetSection("Tus").GetValue<string>("StorePath")!