使用CountdownEvent和ManualResetEvent来控制ThreadPool中的线程

时间:2013-03-16 23:32:47

标签: c# multithreading threadpool manualresetevent countdownevent

我有以下多线程代码摘录,我一直在努力比较压缩副本和解压缩后的文件。 该应用程序正在压缩包含可变数量的各种大小的文件的文件夹,将文件复制到服务器,然后解压缩它们。然后比较文件,并将此比较线程化为ThreadPool

以下是当前完整方法

public void FolderMoverLogic(string folderPathToZip, string unzipOutputDir)
{
    string folderRootDir = Path.GetDirectoryName(folderPathToZip);
    string folderNameToZip = Path.GetFileName(folderPathToZip);

    try
    {
        //Zips files in <folderPathToZip> into folder <zippedLocal>
        TransferMethods.CreateZipExternal(folderPathToZip, zippedlocal);
        //Copies zipped folder to server location
        File.Copy(zippedlocal + "\\" + folderNameToZip + ".zip", zippedserver + "\\" + folderNameToZip + ".zip");
        //Unzips files to final server directory
        TransferMethods.UnZip(zippedserver + "\\" + folderNameToZip + ".zip", unzipOutputDir + "\\" + folderNameToZip, sizeof(Int32));

        TransferMethods m = new TransferMethods();

        //Enumerate Files for MD5 Hash Comparison
        var files = from file in Directory.EnumerateFiles(folderPathToZip, "*", SearchOption.AllDirectories)
                    select new
                    {
                        File = file,
                    };

        int fileCount = 0;
        CountdownEvent countdown = new CountdownEvent(10000); 
        using (ManualResetEvent resetEvent = new ManualResetEvent(false))
        {
            foreach (var f in files)
            {
                Interlocked.Increment(ref fileCount);
                countdown.Reset(fileCount);
                try
                {
                    ThreadPool.QueueUserWorkItem(
                        new WaitCallback(c => 
                            {
                                //Check if any of the hashes have been different and stop all threads for a reattempt
                                if (m.isFolderDifferent)
                                {
                                    resetEvent.Set();
                                    CancellationTokenSource cts = new CancellationTokenSource();
                                    cts.Cancel(); // cancels the CancellationTokenSource 
                                    try
                                    {
                                        countdown.Wait(cts.Token);
                                    }
                                    catch (OperationCanceledException)
                                    {
                                        Console.WriteLine("cde.Wait(preCanceledToken) threw OCE, as expected");
                                    }
                                    return;
                                }
                                else
                                {
                                    //Sets m.isFolderDifferent to true if any files fail MD5 comparison
                                    m.CompareFiles(f.File, folderRootDir, unzipOutputDir);
                                }
                                if (Interlocked.Decrement(ref fileCount) == 0)
                                {
                                    resetEvent.Set();
                                }
                                countdown.Signal();
                            }));

                }
                catch (Exception ex)
                {
                    Console.WriteLine(ex.ToString());
                }
            }
            countdown.Wait();
            resetEvent.WaitOne();
            resetEvent.Close();





        }
    }
    catch (Exception Ex)
    {
        Console.WriteLine(Ex.Message);
    }
}

到目前为止看到的有用资源:

Is it safe to signal and immediately close a ManualResetEvent?

Stopping all thread in .NET ThreadPool?

MSDN CountdownEvent

ThreadPool逻辑要求:

  • 在本地和服务器上比较所有枚举文件
  • 如果散列不匹配,则从所有线程返回

以前的ThreadPool代码

using (ManualResetEvent resetEvent = new ManualResetEvent(false))
{
    foreach (var f in files)
    {
        testCount++;
        try
        {
            //Thread t = new Thread(() => m.CompareFiles(f.File, unzipped, orglsource));
            //t.Start();
            //localThreads.Add(t);
            ThreadPool.QueueUserWorkItem(
                new WaitCallback(c => 
                    {
                        if (resetEvent.WaitOne(0))  //Here is the `ObjectDisposedException`
                        {
                            return;
                        }
                        if (!m.Folderdifferent)
                        {
                            m.CompareFiles(f.File, folderRootDir, unzipOutput);
                        }
                        else
                        {
                            resetEvent.Set();
                        }
                        if (Interlocked.Decrement(ref fileCountZipped) == 0)
                        {
                            resetEvent.Set();
                        }

                    }));

        }
        catch (Exception ex)
        {
            Console.WriteLine(ex.ToString());
        }

    }
    resetEvent.WaitOne();
}

我定期收到ObjectDisposedExceptions前面显示的代码。

我的问题是这样的:

  1. 当前方法是否是线程安全的?
  2. 逻辑是否合理?
  3. 有关性能或线程安全的任何改进建议
  4. 我在顶部的当前方法是否解决了以前的代码异常
  5. 我一直在测试这段代码,它一直没有例外,但我正在寻找一些更有经验的反馈。

1 个答案:

答案 0 :(得分:3)

一些注意事项:

  • 不应该是这样的?:
    CountdownEvent countdown = new CountdownEvent(files.Count()); 
  • 安全吗? - NO - 我根本不喜欢CountdownEvent的想法,如果ANY文件的任何操作失败你没有得到信号并且应用程序挂起countdown.Wait(),我更喜欢使用{{3而不是{而不是countdown.Wait()并使用Task.WaitAll(tasks)
  • 永远不要在线程中使用直接的“foreach变量”(TPL Tasks),而不是:

    foreach (var f in files)
    {
        Task.Run(() =>
        {
             var whateveryDoWithIt = f.File; 
        }
    }
    这样做:
    foreach (var f in files)
    {
        var ftemp = f;
        Task.Run(() =>
        {
             var whateveryDoWithIt = ftemp.File; 
        }
    }

  • 回答它是否是线程安全的我会回答:是的,如果你修正上面的要点并且其中使用的所有方法也是线程安全的