Autofac - 如何使用参数创建生成的工厂

Question

我正在尝试使用 Autofac 创建一个“生成”工厂，它将根据枚举参数实时解决依赖关系。

给定以下接口/类：

public delegate IConnection ConnectionFactory(ConnectionType connectionType);

public enum ConnectionType
{
    Telnet,
    Ssh
}

public interface IConnection
{
    bool Open();
}

public class SshConnection : ConnectionBase, IConnection
{
    public bool Open()
    {
        return false;
    }
}

public class TelnetConnection : ConnectionBase, IConnection
{
    public bool Open()
    {
        return true;
    }
}

public interface IEngine
{
    string Process(ConnectionType connectionType);
}

public class Engine : IEngine
{
    private ConnectionFactory _connectionFactory;

    public Engine(ConnectionFactory connectionFactory)
    {
        _connectionFactory = connectionFactory;
    }

    public string Process(ConnectionType connectionType)
    {
        var connection = _connectionFactory(connectionType);

        return connection.Open().ToString();
    }
}

我想使用 Autofac 生成某种工厂，该工厂有一个接收一个参数的方法：ConnectionType 并返回正确的连接对象。

我从以下注册开始：

builder.RegisterType<AutoFacConcepts.Engine.Engine>()
    .As<IEngine>()
    .InstancePerDependency();

builder.RegisterType<SshConnection>()
    .As<IConnection>();
builder.RegisterType<TelnetConnection>()
    .As<IConnection>();

然后我继续使用不同选项的 TelnetConnection/SshConnection 注册：

1. 命名
2. 键控
3. 元数据

我找不到允许我定义生成的工厂委托的正确注册组合，该委托将返回正确的连接对象（用于 ConnectionType.Ssh 的 SshConnection 和用于 ConnectionType.Telnet 的 TelnetConnection）。

Answer 1

更新Engine 类以采用Func<ConnectionType, IConnection> 而不是委托。 Autofacsupports creating delegate factories on the fly via Func<T>.

public class Engine : IEngine
{
    private Func<ConnectionType, IConnection> _connectionFactory;

    public Engine(Func<ConnectionType, IConnection> connectionFactory)
    {
        _connectionFactory = connectionFactory;
    }

    public string Process(ConnectionType connectionType)
    {
        var connection = _connectionFactory(connectionType);

        return connection.Open().ToString();
    }
}

在您的注册中，使用 lambda 获取参数并返回正确的 IConnection 实例。

builder.Register<IConnection>((c, p) =>
{
    var type = p.TypedAs<ConnectionType>();
    switch (type)
    {
        case ConnectionType.Ssh:
            return new SshConnection();
        case ConnectionType.Telnet:
            return new TelnetConnection();
        default:
            throw new ArgumentException("Invalid connection type");
    }
})
.As<IConnection>();

如果连接本身需要依赖项，您可以在 c 参数上调用 Resolve 以从当前调用上下文中解析它。例如，new SshConnection(c.Resolve<IDependency>())。

Answer 2

如果需要根据参数选择实现类型需要使用IIndex<T,B> implicit relation type:

public class Engine : IEngine
{
    private IIndex<ConnectionType, IConnection> _connectionFactory;

    public Engine(IIndex<ConnectionType, IConnection> connectionFactory)
    {
        _connectionFactory = connectionFactory;
    }

    public string Process(ConnectionType connectionType)
    {
        var connection = _connectionFactory[connectionType];

        return connection.Open().ToString();
    }
}

并使用枚举键注册您的 IConnection 实现：

builder.RegisterType<Engine>()
.           As<IEngine>()
    .InstancePerDependency();

builder.RegisterType<SshConnection>()
    .Keyed<IConnection>(ConnectionType.Ssh);
builder.RegisterType<TelnetConnection>()
    .Keyed<IConnection>(ConnectionType.Telnet);

如果您想保留您的ConnectionFactory，您可以手动注册它以在内部使用IIndex<T,B>：

builder.Register<ConnectionFactory>(c =>
{
    var context = c.Resolve<IComponentContext>();
    return t => context.Resolve<IIndex<ConnectionType, IConnection>>()[t];
});

在这种情况下，您仍然需要将 IConnection 类型注册为键控类型，但您的 Engine 实现可以保持不变。