98 lines
3.2 KiB
C#
98 lines
3.2 KiB
C#
using System.Text.Json;
|
|
using System.Text.Json.Serialization;
|
|
using System.Threading.RateLimiting;
|
|
using CorePush.Apple;
|
|
using CorePush.Firebase;
|
|
using DysonNetwork.Ring.Connection;
|
|
using DysonNetwork.Ring.Email;
|
|
using DysonNetwork.Ring.Notification;
|
|
using DysonNetwork.Ring.Services;
|
|
using DysonNetwork.Shared.Cache;
|
|
using Microsoft.AspNetCore.RateLimiting;
|
|
using NodaTime;
|
|
using NodaTime.Serialization.SystemTextJson;
|
|
using StackExchange.Redis;
|
|
|
|
namespace DysonNetwork.Ring.Startup;
|
|
|
|
public static class ServiceCollectionExtensions
|
|
{
|
|
public static IServiceCollection AddAppServices(this IServiceCollection services, IConfiguration configuration)
|
|
{
|
|
services.AddDbContext<AppDatabase>();
|
|
services.AddSingleton<IClock>(SystemClock.Instance);
|
|
services.AddHttpContextAccessor();
|
|
services.AddSingleton<ICacheService, CacheServiceRedis>();
|
|
|
|
services.AddHttpClient();
|
|
|
|
// Register gRPC services
|
|
services.AddGrpc(options =>
|
|
{
|
|
options.EnableDetailedErrors = true; // Will be adjusted in Program.cs
|
|
options.MaxReceiveMessageSize = 16 * 1024 * 1024; // 16MB
|
|
options.MaxSendMessageSize = 16 * 1024 * 1024; // 16MB
|
|
});
|
|
|
|
// Register gRPC reflection for service discovery
|
|
services.AddGrpc();
|
|
|
|
// Register gRPC services
|
|
services.AddScoped<RingServiceGrpc>();
|
|
|
|
// Register OIDC services
|
|
services.AddControllers().AddJsonOptions(options =>
|
|
{
|
|
options.JsonSerializerOptions.NumberHandling = JsonNumberHandling.AllowNamedFloatingPointLiterals;
|
|
options.JsonSerializerOptions.PropertyNamingPolicy = JsonNamingPolicy.SnakeCaseLower;
|
|
options.JsonSerializerOptions.DictionaryKeyPolicy = JsonNamingPolicy.SnakeCaseLower;
|
|
|
|
options.JsonSerializerOptions.ConfigureForNodaTime(DateTimeZoneProviders.Tzdb);
|
|
});
|
|
|
|
return services;
|
|
}
|
|
|
|
public static IServiceCollection AddAppRateLimiting(this IServiceCollection services)
|
|
{
|
|
services.AddRateLimiter(o => o.AddFixedWindowLimiter(policyName: "fixed", opts =>
|
|
{
|
|
opts.Window = TimeSpan.FromMinutes(1);
|
|
opts.PermitLimit = 120;
|
|
opts.QueueLimit = 2;
|
|
opts.QueueProcessingOrder = QueueProcessingOrder.OldestFirst;
|
|
}));
|
|
|
|
return services;
|
|
}
|
|
|
|
public static IServiceCollection AddAppAuthentication(this IServiceCollection services)
|
|
{
|
|
services.AddAuthorization();
|
|
return services;
|
|
}
|
|
|
|
public static IServiceCollection AddAppFlushHandlers(this IServiceCollection services)
|
|
{
|
|
services.AddSingleton<FlushBufferService>();
|
|
services.AddScoped<NotificationFlushHandler>();
|
|
|
|
return services;
|
|
}
|
|
|
|
public static IServiceCollection AddAppBusinessServices(this IServiceCollection services)
|
|
{
|
|
services.AddSingleton<WebSocketService>();
|
|
services.AddScoped<EmailService>();
|
|
services.AddScoped<PushService>();
|
|
|
|
// Register QueueService as a singleton since it's thread-safe
|
|
services.AddSingleton<QueueService>();
|
|
|
|
// Register the background service
|
|
services.AddHostedService<QueueBackgroundService>();
|
|
|
|
return services;
|
|
}
|
|
}
|