使用多线程/并行查找工作端口

Question

我有一个类似下面的函数来连接一个端口不断变化的 FTP

private int FindWorkingPort(int from, int to)
{
    for (int port = from; port <= to; port++)
    {
        try
        {
            new FtpClient(_host, port).Connect();  // Instant on correct port, very slow on wrong port
            return port;
        }
        catch { }
    }

    throw new Exception("None of the port is working");
}

当第一次尝试正确时，它是即时的。否则，每次尝试可能需要 10 秒。你可以想象如果正确的是最后一个。

有没有办法让它同时尝试所有可能的端口？ （我是多线程的菜鸟）

编辑：

我正在使用FluentFTP，它确实有.ConnectAsync()

Answer 1

这不是最好但可行的解决方案：

var tasks = Enumerable.Range(from, to - from).ToList().Select(port => {
    try
    {
        return new FtpClient(_host, port).ConnectAsync().ContinueWith(_ => (int?)port);
    }
    catch
    {
        return Task.FromResult((int?)null);
    }
});
var foundPort = (await Task.WhenAll(tasks)).FirstOrDefault(p => p != null);

基本上，您创建一个您想要检查的所有端口的列表，并为每个端口启动一项任务。您等到所有这些都异步完成并获得第一个不是null 的。所有任务将并行运行，但await Task.WhenAll 将等待所有任务完成，这意味着它将始终等待约 10 秒（根据您的问题，检查不正确的端口通常需要约 10 秒）。

这可能会通过Task.WhenAny 得到改善，但是当其他任务抛出异常时返回正确的端口是很棘手的。

Answer 2

这是另一种方法。这应该在第一个有效任务完成后立即返回：

public async Task<int> FindWorkingPortAsync(int from, int to)
{
    var wrapper = new TaskCompletionSource<int>();
    var tasks = Enumerable.Range(from, to - from).ToList().Select(port =>
    {
        return new FtpClient(_host, port).ConnectAsync().ContinueWith(t => {
            try
            {
                if(!wrapper.Task.IsCompleted && !t.IsFaulted && !t.IsCanceled)
                    wrapper.SetResult(port);
                return t;
            }
            catch
            {
                return Task.FromResult(-1);
            }
        });
    }).ToList();


    return await wrapper.Task;
}

让我知道它是否有效。我用HttpClient而不是FtpClient对其进行了测试。

Answer 3

您可以尝试使用多个线程同时连接到所有端口，但是随着更多线程的启动，性能会降低。我建议你试试 Task。

这是一个任务和事件的例子。这不完全是解决方案，但你会得到这个要点。您可能需要更改代码结构。

    using System;
    using System.Threading.Tasks;

    public class Program
    {
        public class MyEventArgs : EventArgs
        {
             public int port {get; set; }

             public MyEventArgs(int port)
             {
                 this.port = port;
             }
        }

        delegate void OpenPort(object sender, MyEventArgs  e);
        event OpenPort OnOpen;
        int from = 10, to = 20;
        int Connected = 16;

        public static void Main()
        {
            var obj = new Program();
            // here you register your method ShowPort for OnOpen event. 
            obj.OnOpen += new OpenPort(ShowPort);
            for(int i = obj.from; i < obj.to; i++) {
                obj.CheckPort(obj, i);
            }
        }

       private async void CheckPort(Program obj, int i) 
       {
           await CheckPortTask(obj, i);
       }

       private Task CheckPortTask(Program obj, int i)
       {
             return Task.Run(() =>  {
                if(i == obj.Connected)
                {
                  obj.OnOpen(this, new MyEventArgs(i));
                }
         });
       }            

        // This method will be called after connecting.
        public static void ShowPort(Object p, MyEventArgs args)
        {
            Console.WriteLine(args.port);
        }
    }

如果您想尝试使用线程，请将上述代码中的 Task.Run 块替换为：

new Thread(()=>{
                        if(i == obj.w) 
                        {
                            // this raises the event and calls the ShowPort method.
                            obj.OnOpen(obj, new MyEventArgs(i));
                        }
                    })).Start();