c# - 是否可以根据 API 端点所在的服务器来禁用它？

Question

这个问题已经有答案了:

ASP.NET core it's possible to configure an action in controller only in development mode? (3 个回答)

已关闭 4 年前。

我有一个“ClearAllData” Controller 端点，它在调用时会清除 SQL 数据库中的所有数据。客户出于测试目的请求此操作，但我想在生产中禁用此操作。我该怎么做呢？我正在使用 .NET Core 2.2 和 C#

Answer 1

您可以添加如下资源过滤器:

public class DevelopmentOnlyAttribute : Attribute, IResourceFilter
{
    public void OnResourceExecuted(ResourceExecutedContext context)
    {
    }

    public void OnResourceExecuting(ResourceExecutingContext context)
    {
        var env = context.HttpContext.RequestServices.GetRequiredService<IHostingEnvironment>();
        if (!env.IsDevelopment())
        {
            context.Result = new NotFoundResult();
        }
    }
}

然后，只需使用此属性装饰操作即可:

[DevelopmentOnly]
public IActionResult ClearAllData()