防止 ASP.NET Core 在单独的程序集中发现控制器

Question

我有一个 ASP.NET Core API，它引用了一个包含 Controller 的 nuget 包。

默认情况下，此控制器已注册并且可以响应请求。但是，我只想在某些情况下添加它 - 例如如果它在 DEV 环境中。

我的Startup 看起来像这样：

services.AddControllers()
.AddMvcOptions(cfg => {
   cfg.Filters.Add(new CustomExceptionFilterAttribute())
});

我希望在调用AddCointrollers 后需要调用AddApplicationPart(typeof(ClassInPackage).Assembly) 来注册此控制器？

有人可以建议我启用/禁用此控制器注册的方法吗？

Answer 1

好的，我找到了解决方案 - 删除包含 Controller 的 ApplicationPart。仍然可以使用程序集中的任何其他依赖项。

在 Startup.cs / 任何你做 IoC 的地方：

if(hideControllerFromOtherAssembly)
{
   var appPartManager = (ApplicationPartManager)services.FirstOrDefault(a => a.ServiceType == typeof(ApplicationPartManager)).ImplementationInstance;
   var mockingPart = appPartManager.ApplicationParts.FirstOrDefault(a => a.Name == "MyMockLibrary.Namespace");

   if(mockingPart != null)
   {
      appPartManager.ApplicationParts.Remove(mockingPart);
   }
}

您还可以通过扩展方法操作ApplicationParts：

AddMvc().ConfigureApplicationPartManager()

这不适合我，因为我在我的 nuget 包中编写了扩展方法

https://docs.microsoft.com/en-us/aspnet/core/mvc/advanced/app-parts?view=aspnetcore-5.0