Вызов метода после завершения ThreadPool.QueueUserWorkItem - PullRequest
Новая
       29

Вызов метода после завершения ThreadPool.QueueUserWorkItem

1 голос
/ 08 мая 2019

Я работаю над консольным приложением, написанным на c #

Цель этого приложения - просмотреть все диски и файлы и что-то с ними сделать.Но просмотр всех файлов в одном потоке - это трудоемкий процесс, который не является моей целью.

Поэтому я решил использовать ThreadPool для его обработки следующим образом: 

class Program () {
    static void Main(string[] args) {
        foreach (var d in DriveInfo.GetDrives()) {
            ThreadPool.QueueUserWorkItem(x => Search(d.RootDirectory.GetDirectories()));
        }

        Console.WriteLine("Job is done.");
        Console.ReadKey();
    }

    private static void Search(DirectoryInfo[] dirs) {
        foreach (var dir in dirs) {
            try {
                foreach (var f in dir.GetFiles()) {
                    ThreadPool.QueueUserWorkItem(x => DoTheJob(f));
                }

                ThreadPool.QueueUserWorkItem(x => Search(dir.GetDirectories()));
            } catch (Exception ex) {
                continue;
            }
        }
    }       
}

проблема в Console.WriteLine("Job is done.") выполняется до того, как все потоки завершены.Я прочитал несколько вопросов и ответов, но ни один из них не решил мою проблему.

Как я могу вызвать метод после того, как все потоки в ThreadPool закончили свою работу?

Примечание: Как вы, возможно, знаете, я не знаю, сколько потоков будет создано, потому что я не знаю, сколько там файлов.И установка тайм-аута не вариант.

Ответы [ 2 ]

1 голос
/ 08 мая 2019

Использование QueueUserWorkItem () - это простой подход.Без контроля над вашей работой, это огонь и забыть.

Task s запускается поверх ThreadPool, и async/await может решить вашу проблему здесь.

Верхний уровень: 

var tasks = new List<Task>();
foreach (var d in DriveInfo.GetDrives())
{
    tasks.Add( Search(d.RootDirectory.GetDirectories()));
}
Task.WaitAll(tasks.ToArray());

, и тогда Search () становится 

private static async Task Search(DirectoryInfo[] dirs)
{
    ... 
    foreach(...)
    {
        await Task.Run(...);
    } 
    await Search(dir.GetDirectories());
}

То, что DoTheJob () в идеале должен использовать асинхронный ввод-вывод, но в противном случае вы можетеawait Task.Run( () => DoTheJob(f))

0 голосов
/ 08 мая 2019

Вот пример того, как вы можете использовать Parallel.ForEach для получения справедливой нагрузки: 

static IEnumerable<FileSystemInfo> GetFileSystemObjects(DirectoryInfo dirInfo)
{
    foreach (var file in dirInfo.GetFiles())
        yield return file;

    foreach (var dir in dirInfo.GetDirectories())
    {
        foreach (var fso in GetFileSystemObjects(dir))
            yield return fso;
        yield return dir;
    }
}

static void Main(string[] args)
{
    var files = GetFileSystemObjects(new DirectoryInfo(<some path>)).OfType<FileInfo>();

    Parallel.ForEach(files, f =>
    {
        DoTheJob(f);
    });
}

Если, однако, DoTheJob содержит операции, связанные с вводом / выводом, я бы посоветовал обработать его с await, поскольку Хенк Холтерман предложил, поскольку Parallel.ForEach не зависит от нагрузки ввода / вывода.

...