.NET 6中如何为带参自定义类创建ObjectPool？

我之前在Stack Overflow提过相关问题，现在想为FTP客户端实现对象池——因为FTP连接建立开销高，想复用已连接的客户端。计划用Microsoft.Extensions.ObjectPool NuGet包，同时要限制最多100个客户端（目标FTP服务器限制同时连接数100）。

现有MyFtpClient类如下：

public class MyFtpClient : IMyFtpClient
{
    private readonly RealFtpClient _realFtpClient;

    public MyFtpClient(string host, string username, string password)
    {
        // 初始化FTP客户端
        _realFtpClient = new RealFtpClient(host, username, password);
    }

    public async Task UploadFileAsync(Stream someFile, CancellationToken ct)
    {
        // 使用FTP客户端上传文件
        await _realFtpClient.UploadFile(someFile, ct);
    }
}

我原本尝试这样创建对象池：

var objectPool = new DefaultObjectPool<MyFtpClient>(new DefaultPooledObjectPolicy<MyFtpClient>(), 100);

但因为MyFtpClient没有无参构造函数，这段代码无法运行。

补充场景：我从队列拉取消息上传FTP，每次最多拉100条。处理流程是：为每条消息获取MyFtpClient，创建上传任务，最后await Task.WhenAll(tasks)。请问该怎么实现？

要解决无参构造函数的问题，你需要自定义PooledObjectPolicy<MyFtpClient>，负责实例的创建与回收，同时建议增加FTP连接有效性检查，避免复用已断开的连接。

1. 自定义对象池策略 #

public class MyFtpClientPoolPolicy : PooledObjectPolicy<MyFtpClient>
{
    private readonly string _host;
    private readonly string _username;
    private readonly string _password;

    public MyFtpClientPoolPolicy(string host, string username, string password)
    {
        _host = host;
        _username = username;
        _password = password;
    }

    // 创建新的FTP客户端实例
    public override MyFtpClient Create()
    {
        return new MyFtpClient(_host, _username, _password);
    }

    // 回收时检查客户端是否可用（必须根据RealFtpClient实际API调整）
    public override bool Return(MyFtpClient client)
    {
        // 示例：替换为RealFtpClient的连接状态检查逻辑，比如client._realFtpClient.IsConnected
        // 如果连接已失效，返回false，对象池会自动丢弃该实例
        return true;
    }
}

2. 初始化对象池 #

// 替换为你的FTP服务器配置
var ftpHost = "your-ftp-host";
var ftpUsername = "your-username";
var ftpPassword = "your-password";

// 用自定义策略创建对象池，设置最大容量100
var poolPolicy = new MyFtpClientPoolPolicy(ftpHost, ftpUsername, ftpPassword);
var objectPool = new DefaultObjectPool<MyFtpClient>(poolPolicy, 100);

3. 结合消息处理流程使用对象池 #

重点要保证每个任务完成后必须归还客户端到池，避免资源泄漏。用try/finally确保无论上传成功或失败都能归还：

// 假设messages是从队列拉取的消息列表，每个消息包含待上传的流
var messages = await PullMessagesFromQueueAsync(100);
var tasks = new List<Task>();
var ct = CancellationToken.None; // 根据实际业务传入取消令牌

foreach (var message in messages)
{
    tasks.Add(ProcessMessageAsync(message, objectPool, ct));
}

await Task.WhenAll(tasks);

// 单独封装消息处理逻辑
async Task ProcessMessageAsync(Message message, ObjectPool<MyFtpClient> pool, CancellationToken ct)
{
    MyFtpClient client = null;
    try
    {
        // 从池获取客户端
        client = pool.Get();
        // 执行上传操作
        await client.UploadFileAsync(message.FileStream, ct);
    }
    catch (Exception ex)
    {
        // 按需处理FTP上传异常，比如日志记录
        // 如果遇到致命连接错误，可以选择不归还客户端，让池创建新实例
        if (IsFatalConnectionError(ex))
        {
            return;
        }
        throw;
    }
    finally
    {
        // 确保归还客户端到池
        if (client != null)
        {
            pool.Return(client);
        }
    }
}

// 辅助方法：判断是否为致命连接错误（根据RealFtpClient的异常类型调整）
bool IsFatalConnectionError(Exception ex)
{
    // 示例：匹配连接超时、连接断开等异常类型
    return ex is IOException || ex is SocketException;
}

关键注意事项 #

• 连接有效性校验：Return方法里的连接状态检查必须实现，否则复用已断开的连接会导致大量上传失败。
• 资源安全：必须通过try/finally保证客户端归还，否则会出现池资源耗尽、无法获取新客户端的情况。
• 异常处理：针对FTP连接类异常，可选择直接丢弃失效客户端，避免污染对象池。

内容的提问来源于stack exchange，提问作者Pure.Krome