refactor: 移除租户头校验并后置租户解析

This commit is contained in:
2026-01-29 05:22:09 +00:00
parent d7821fa1af
commit be0282af9f
3 changed files with 35 additions and 33 deletions

View File

@@ -164,10 +164,12 @@ builder.Services.AddCors(options =>
// 8. 构建应用并配置中间件管道 // 8. 构建应用并配置中间件管道
var app = builder.Build(); var app = builder.Build();
app.UseCors("AdminApiCors"); app.UseCors("AdminApiCors");
app.UseTenantResolution();
app.UseRateLimiter(); app.UseRateLimiter();
app.UseSharedWebCore(); app.UseSharedWebCore();
app.UseAuthentication(); app.UseAuthentication();
// 8.1 (空行后) 解析租户:在认证后才能读取 Token Claimtenant_id
app.UseTenantResolution();
app.UseAuthorization(); app.UseAuthorization();
if (app.Environment.IsDevelopment()) if (app.Environment.IsDevelopment())
{ {

View File

@@ -9,7 +9,6 @@ using TakeoutSaaS.Domain.Tenants.Repositories;
using TakeoutSaaS.Shared.Abstractions.Constants; using TakeoutSaaS.Shared.Abstractions.Constants;
using TakeoutSaaS.Shared.Abstractions.Exceptions; using TakeoutSaaS.Shared.Abstractions.Exceptions;
using TakeoutSaaS.Shared.Abstractions.Ids; using TakeoutSaaS.Shared.Abstractions.Ids;
using TakeoutSaaS.Shared.Abstractions.Tenancy;
namespace TakeoutSaaS.Application.App.Tenants.Handlers; namespace TakeoutSaaS.Application.App.Tenants.Handlers;
@@ -18,8 +17,7 @@ namespace TakeoutSaaS.Application.App.Tenants.Handlers;
/// </summary> /// </summary>
public sealed class BindInitialTenantSubscriptionCommandHandler( public sealed class BindInitialTenantSubscriptionCommandHandler(
ITenantRepository tenantRepository, ITenantRepository tenantRepository,
IIdGenerator idGenerator, IIdGenerator idGenerator)
ITenantProvider tenantProvider)
: IRequestHandler<BindInitialTenantSubscriptionCommand, TenantSubscriptionDto> : IRequestHandler<BindInitialTenantSubscriptionCommand, TenantSubscriptionDto>
{ {
private static readonly ConcurrentDictionary<long, SemaphoreSlim> TenantLocks = new(); private static readonly ConcurrentDictionary<long, SemaphoreSlim> TenantLocks = new();
@@ -27,23 +25,28 @@ public sealed class BindInitialTenantSubscriptionCommandHandler(
/// <inheritdoc /> /// <inheritdoc />
public async Task<TenantSubscriptionDto> Handle(BindInitialTenantSubscriptionCommand request, CancellationToken cancellationToken) public async Task<TenantSubscriptionDto> Handle(BindInitialTenantSubscriptionCommand request, CancellationToken cancellationToken)
{ {
// 1. 校验租户上下文 // 1. 校验请求参数
var currentTenantId = tenantProvider.GetCurrentTenantId(); if (request.TenantId <= 0)
if (currentTenantId == 0 || currentTenantId != request.TenantId)
{ {
throw new BusinessException(ErrorCodes.Forbidden, "租户上下文不匹配,请在请求头 X-Tenant-Id 指定目标租户"); throw new BusinessException(ErrorCodes.BadRequest, "租户 ID 无效");
} }
// 1.2 获取租户级幂等锁,避免并发重复创建 // 1.2 (空行后) 校验套餐 ID
if (request.TenantPackageId <= 0)
{
throw new BusinessException(ErrorCodes.BadRequest, "套餐 ID 无效");
}
// 2. 获取租户级幂等锁,避免并发重复创建
var tenantLock = GetTenantLock(request.TenantId); var tenantLock = GetTenantLock(request.TenantId);
await tenantLock.WaitAsync(cancellationToken); await tenantLock.WaitAsync(cancellationToken);
try try
{ {
// 2. 获取租户 // 3. 获取租户
var tenant = await tenantRepository.FindByIdAsync(request.TenantId, cancellationToken) var tenant = await tenantRepository.FindByIdAsync(request.TenantId, cancellationToken)
?? throw new BusinessException(ErrorCodes.NotFound, "租户不存在"); ?? throw new BusinessException(ErrorCodes.NotFound, "租户不存在");
// 3. 幂等校验:若已存在订阅则直接返回 // 4. 幂等校验:若已存在订阅则直接返回
var existing = await tenantRepository.GetActiveSubscriptionAsync(request.TenantId, cancellationToken); var existing = await tenantRepository.GetActiveSubscriptionAsync(request.TenantId, cancellationToken);
if (existing is not null) if (existing is not null)
{ {
@@ -51,7 +54,7 @@ public sealed class BindInitialTenantSubscriptionCommandHandler(
?? throw new BusinessException(ErrorCodes.InternalServerError, "订阅读取失败"); ?? throw new BusinessException(ErrorCodes.InternalServerError, "订阅读取失败");
} }
// 4. 创建 0 个月订阅(待支付/待生效) // 5. 创建 0 个月订阅(待支付/待生效)
var now = DateTime.UtcNow; var now = DateTime.UtcNow;
var subscription = new TenantSubscription var subscription = new TenantSubscription
{ {
@@ -66,7 +69,7 @@ public sealed class BindInitialTenantSubscriptionCommandHandler(
Notes = "初次绑定订阅" Notes = "初次绑定订阅"
}; };
// 5. 记录订阅与历史 // 6. 记录订阅与历史
await tenantRepository.AddSubscriptionAsync(subscription, cancellationToken); await tenantRepository.AddSubscriptionAsync(subscription, cancellationToken);
await tenantRepository.AddSubscriptionHistoryAsync(new TenantSubscriptionHistory await tenantRepository.AddSubscriptionHistoryAsync(new TenantSubscriptionHistory
{ {
@@ -83,7 +86,7 @@ public sealed class BindInitialTenantSubscriptionCommandHandler(
Notes = "初次绑定订阅0 个月)" Notes = "初次绑定订阅0 个月)"
}, cancellationToken); }, cancellationToken);
// 6. 记录审计日志 // 7. 记录审计日志
await tenantRepository.AddAuditLogAsync(new TenantAuditLog await tenantRepository.AddAuditLogAsync(new TenantAuditLog
{ {
TenantId = tenant.Id, TenantId = tenant.Id,
@@ -92,16 +95,16 @@ public sealed class BindInitialTenantSubscriptionCommandHandler(
Description = $"套餐 {request.TenantPackageId},时长 0 月" Description = $"套餐 {request.TenantPackageId},时长 0 月"
}, cancellationToken); }, cancellationToken);
// 7. 保存变更 // 8. 保存变更
await tenantRepository.SaveChangesAsync(cancellationToken); await tenantRepository.SaveChangesAsync(cancellationToken);
// 8. 返回 DTO // 9. 返回 DTO
return subscription.ToSubscriptionDto() return subscription.ToSubscriptionDto()
?? throw new BusinessException(ErrorCodes.InternalServerError, "订阅绑定失败"); ?? throw new BusinessException(ErrorCodes.InternalServerError, "订阅绑定失败");
} }
finally finally
{ {
// 9. 释放幂等锁 // 10. 释放幂等锁
tenantLock.Release(); tenantLock.Release();
} }
} }

View File

@@ -6,7 +6,6 @@ using TakeoutSaaS.Domain.Tenants.Enums;
using TakeoutSaaS.Domain.Tenants.Repositories; using TakeoutSaaS.Domain.Tenants.Repositories;
using TakeoutSaaS.Shared.Abstractions.Constants; using TakeoutSaaS.Shared.Abstractions.Constants;
using TakeoutSaaS.Shared.Abstractions.Exceptions; using TakeoutSaaS.Shared.Abstractions.Exceptions;
using TakeoutSaaS.Shared.Abstractions.Tenancy;
namespace TakeoutSaaS.Application.App.Tenants.Handlers; namespace TakeoutSaaS.Application.App.Tenants.Handlers;
@@ -17,27 +16,25 @@ public sealed class CheckTenantQuotaCommandHandler(
ITenantRepository tenantRepository, ITenantRepository tenantRepository,
ITenantPackageRepository packageRepository, ITenantPackageRepository packageRepository,
ITenantQuotaUsageRepository quotaUsageRepository, ITenantQuotaUsageRepository quotaUsageRepository,
ITenantQuotaUsageHistoryRepository quotaUsageHistoryRepository, ITenantQuotaUsageHistoryRepository quotaUsageHistoryRepository)
ITenantProvider tenantProvider)
: IRequestHandler<CheckTenantQuotaCommand, QuotaCheckResultDto> : IRequestHandler<CheckTenantQuotaCommand, QuotaCheckResultDto>
{ {
/// <inheritdoc /> /// <inheritdoc />
public async Task<QuotaCheckResultDto> Handle(CheckTenantQuotaCommand request, CancellationToken cancellationToken) public async Task<QuotaCheckResultDto> Handle(CheckTenantQuotaCommand request, CancellationToken cancellationToken)
{ {
// 1. 校验请求参数 // 1. 校验请求参数
if (request.TenantId <= 0)
{
throw new BusinessException(ErrorCodes.BadRequest, "租户 ID 无效");
}
// 1.2 (空行后) 校验消耗量
if (request.Delta <= 0) if (request.Delta <= 0)
{ {
throw new BusinessException(ErrorCodes.BadRequest, "配额消耗量必须大于 0"); throw new BusinessException(ErrorCodes.BadRequest, "配额消耗量必须大于 0");
} }
// 2. 校验租户上下文 // 2. 获取租户与当前订阅
var currentTenantId = tenantProvider.GetCurrentTenantId();
if (currentTenantId == 0 || currentTenantId != request.TenantId)
{
throw new BusinessException(ErrorCodes.Forbidden, "租户上下文不匹配,请在请求头 X-Tenant-Id 指定目标租户");
}
// 3. 获取租户与当前订阅
_ = await tenantRepository.FindByIdAsync(request.TenantId, cancellationToken) _ = await tenantRepository.FindByIdAsync(request.TenantId, cancellationToken)
?? throw new BusinessException(ErrorCodes.NotFound, "租户不存在"); ?? throw new BusinessException(ErrorCodes.NotFound, "租户不存在");
@@ -52,7 +49,7 @@ public sealed class CheckTenantQuotaCommandHandler(
var limit = ResolveLimit(package, request.QuotaType); var limit = ResolveLimit(package, request.QuotaType);
// 4. 加载配额使用记录并计算 // 3. 加载配额使用记录并计算
var usage = await quotaUsageRepository.FindAsync(request.TenantId, request.QuotaType, cancellationToken) var usage = await quotaUsageRepository.FindAsync(request.TenantId, request.QuotaType, cancellationToken)
?? new TenantQuotaUsage ?? new TenantQuotaUsage
{ {
@@ -63,7 +60,7 @@ public sealed class CheckTenantQuotaCommandHandler(
ResetCycle = ResolveResetCycle(request.QuotaType) ResetCycle = ResolveResetCycle(request.QuotaType)
}; };
// 4.1 记录是否为首次初始化(用于落库历史) // 3.1 记录是否为首次初始化(用于落库历史)
var isNewUsage = usage.Id == 0; var isNewUsage = usage.Id == 0;
var usedAfter = usage.UsedValue + request.Delta; var usedAfter = usage.UsedValue + request.Delta;
@@ -74,12 +71,12 @@ public sealed class CheckTenantQuotaCommandHandler(
throw new BusinessException(ErrorCodes.Conflict, $"{request.QuotaType} 配额不足"); throw new BusinessException(ErrorCodes.Conflict, $"{request.QuotaType} 配额不足");
} }
// 5. 更新使用并保存 // 4. 更新使用并保存
usage.LimitValue = limit ?? usage.LimitValue; usage.LimitValue = limit ?? usage.LimitValue;
usage.UsedValue = usedAfter; usage.UsedValue = usedAfter;
usage.ResetCycle ??= ResolveResetCycle(request.QuotaType); usage.ResetCycle ??= ResolveResetCycle(request.QuotaType);
// 5.1 落库历史(初始化 + 本次消耗) // 4.1 落库历史(初始化 + 本次消耗)
var now = DateTime.UtcNow; var now = DateTime.UtcNow;
if (isNewUsage) if (isNewUsage)
{ {
@@ -109,7 +106,7 @@ public sealed class CheckTenantQuotaCommandHandler(
await PersistUsageAsync(usage, quotaUsageRepository, cancellationToken); await PersistUsageAsync(usage, quotaUsageRepository, cancellationToken);
// 6. 返回结果 // 5. 返回结果
return new QuotaCheckResultDto return new QuotaCheckResultDto
{ {
QuotaType = request.QuotaType, QuotaType = request.QuotaType,