Csharp 注册所有IMediatR类

发布于02月22日

有什么形式可以避免注册所有实现IMediator服务的程序集吗？

我try 使用这些选项，但它们都不能帮助我:

builder.Services.AddMediatR(Assembly.GetExecutingAssembly());

builder.Services.AddMediatR(cfg => cfg.RegisterServicesFromAssembly(Assembly.GetExecutingAssembly()))

目前，唯一适合我的 Select 是，如果我像这样注册所有服务:

builder.Services.AddMediatR(cfg => cfg.RegisterServicesFromAssembly(typeof(MYQUERY).Assembly));

但这是非常乏味的，因为我的程序.cs将有很多行，有没有其他 Select 来做到这一点？

推荐答案

但这非常乏味，因为我的程序.cs将有很多行

就我个人而言，我会 Select "手动"方法.如果您不想"丢弃"Program.cs，那么您始终可以将安装程序的这一部分移到某个扩展方法:

public static class MediatrRegistrationExts
{
    public static IServiceCollection AddMediatorWithAll(this IServiceCollection services)
    {
        var types = new[]
        {
            typeof(MediatrAssemlyOneLandmark),
            typeof(MediatrAssemlyTwoLandmark)
        };
        services.AddMediatR(types.Select(t => t.Assembly).ToArray());
        return services;
    }
}

然后:

builder.Services.AddMediatorWithAll();

但如果你真的-真的想要进入全动态模式，那么你可以try 下面这样的方法(未测试):

builder.Services.AddMediatR(GetAllAssemblies().ToArray());

static IEnumerable<Assembly> GetAllAssemblies()
{
    var assemblies = new HashSet<string>();
    var stack = new Stack<Assembly>();

    stack.Push(Assembly.GetEntryAssembly());

    while (stack.Count > 0)
    {
        var asm = stack.Pop();

        yield return asm;

        foreach (var reference in asm.GetReferencedAssemblies())
        {
            if (!assemblies.Contains(reference.FullName))
            {
                stack.Push(Assembly.Load(reference));
                assemblies.Add(reference.FullName);
            }
        }

    }
}

但我强烈建议不要这么做.

附注:

请确保将这些类型实际放置在不同的程序集中，因为根据注释，它们很可能实际上位于单个程序集中.