在.NET Core中处理一个接口多个不同实现的依赖注入问题

前言

近段时间在准备公司的技术分享，所以这段时间将大部分时间放在准备分享内容上去了。博客也就停了一下下。

在.NET Core中处理依赖注入问题时，往往是定义好了一个操作规范的接口，会有N多个基于不同技术的实现，根据实际情况在项目中去使用某一个实现。

但是偶尔会出现这样的情况，在某一个地方，需要同时使用到两种或两种以上的实现，这个时候我们要怎么处理呢？

借助Autofac等第三方组件时，是可以很容易的实现，但是在写一些基础类库时会避免直接引用太多依赖组件。

所以这里是只用微软自带的DI(Microsoft.Extensions.DependencyInjection)去处理。

例子引入

现在有一个接口和两个实现类。

public interface IDemoService
{
    string Get();
}

public class DemoServiceA : IDemoService
{
    public string Get()
    {
        return "Service A";
    }
}

public class DemoServiceB : IDemoService
{
    public string Get()
    {
        return "Service B";
    }
}

常规的方法，我们先在Startup中的ConfigureServices方法中添加我们的service。

public void ConfigureServices(IServiceCollection services)
{
    services.AddSingleton<IDemoService, DemoServiceA>();
    services.AddSingleton<IDemoService, DemoServiceB>();

    services.AddMvc();
}

然后在控制器中使用

private IDemoService _serviceA;

private IDemoService _serviceB;

public ValuesController(IDemoService serviceA, IDemoService serviceB)
{
    _serviceA = serviceA;
    _serviceB = serviceB;
}

// GET api/values
[HttpGet]
public string Get()
{
    return $"{_serviceA.Get()}-{_serviceB.Get()}";
}

我们的预期结果是：Service A-Service B，可是上面代码的实际结果却并不像我们想的那么简单！！

可以看到这里输出的都是Service B，连Service A的影子都没有看到。

其实，从代码都可以看出来，它只能拿到其中一个Service的实现类！

那么我们要息怎样处理才能达到我们想要的效果呢？

其实思路比较简单，上面导致不能拿到对应实现类，本质上来讲应该说是它区分不了那个才是想要的！我们想个办法让它能区分就好了。

处理方法

给我们的Service起个别名！

先是Startup中的ConfigureServices方法。

public void ConfigureServices(IServiceCollection services)
{
    services.AddSingleton<DemoServiceA>();
    services.AddSingleton<DemoServiceB>();

    services.AddSingleton(factory =>
    {
        Func<string, IDemoService> accesor = key =>
        {
            if (key.Equals("ServiceA"))
            {
                return factory.GetService<DemoServiceA>();
            }
            else if (key.Equals("ServiceB"))
            {
                return factory.GetService<DemoServiceB>();
            }
            else
            {
                throw new ArgumentException($"Not Support key : {key}");
            }
        };
        return accesor;
    });

    services.AddMvc();
}

这里并没有直接向上面那样一次性指定接口和对应的实现类，而是用了AddSingleton的另一个重载方法。

先将实现类注册一下
然后再注册一下Func<string, IDemoService>

先来说说这个Func<string, IDemoService>里面的string和IDemoService都分别代表什么。

string 毫无疑问就是我们上面说到的别名
IDemoService 这个就是我们要用的Service

核心在于，factory参数是IServiceProvider类型的！所以我们可以根据这个factory去找到我们前面注册的实现类。这样解释一下，是不是就清晰了呢？

然后再来看看在控制器上面怎么用。

private IDemoService _serviceA;

private IDemoService _serviceB;

private readonly Func<string, IDemoService> _serviceAccessor;

public ValuesController(Func<string, IDemoService> serviceAccessor)
{
    this._serviceAccessor = serviceAccessor;

    _serviceA = _serviceAccessor("ServiceA");
    _serviceB = _serviceAccessor("ServiceB");
}

// GET api/values
[HttpGet]
public string Get()
{
    return $"{_serviceA.Get()}-{_serviceB.Get()}";
}