如何填充排序列表的并发字典?

时间:2012-02-23 09:49:07

标签: c# concurrency task-parallel-library

让我们想象一下我有这个课程:

public class FileData
{
   public Weigth { get; set; } // Not the file size, but a business weight
   public Name { get; set; } // Name of the file
   public FullPath { get; set; } // full path of the file
}

我想从各种网络位置探索一堆独立的文件夹。我们的想法是,在给定特定文件名的情况下,能够检索按文件权重排序的所有候选文件。

我想在单独的线程中拆分每个文件夹的探索。在这种情况下,我的“商店”应该是并发访问感知。

我尝试使用ConcurrentDictionary<string, SortedSet<CdlFileInfo>>类来存储探索结果。

但是,我对填充内部排序集的正确方法感到有点挣扎。

我试过这个:

class Program 
{
    private readonly static ConcurrentDictionary<string, SortedSet<CdlFileInfo>> g_Files = new ConcurrentDictionary<string,SortedSet<CdlFileInfo>>();
    public static void Main()
    {
        PopulateFileList();

        // Do something with the list of files
    }
    private static void PopulateFileList()
    {
        var sources = AnyMethodToGetFoldersList(); // IEnumarable<string>
        sources.AsParallel().ForAll(x =>
        {
            Console.WriteLine("Enumerating files in {0}", x.Folder);
            var allFiles = Directory.GetFiles(x.Folder, "*.*", SearchOption.AllDirectories);

            foreach (var file in allFiles)
            {
                var fd = new FileData {
                    Weigth = GetWeigth(file), // returns a int... the method is not important
                    Name = Path.GetFileName(file),
                    FullPath = file
                };

                            // Here is the key piece of my code

                g_Files.AddOrUpdate(
                    fileName,
                    new SortedSet<FileData>() { fd },
                    (filePath, source) =>
                    {
                        g_Files[fileName].Add(fd);
                        return g_Files[fileName];
                    }
                );
            }
        });
    }

    public class FileData
    {
       public Weigth { get; set; } // Not the file size, but a business weight
       public Name { get; set; } // Name of the file
       public FullPath { get; set; } // full path of the file
    }
    public class FileDataWeightComparer : IComparer<CdlFileInfo>
    {
        public int Compare(FileData x, FileData y)
        {
            return Comparer<int>.Default.Compare(x.Weigth,y.Weigth);
        }
    }
}

此代码“似乎”可行。这是正确的方法吗?访问现有的SortedSet时,此代码是否会阻止线程问题?

此代码似乎不起作用。我可以看到一些找到的值丢失了。我怀疑AddOrUpdate方法的最后一个参数不能正确锁定内部的SortedSet。

1 个答案:

答案 0 :(得分:2)

更改代码的这一部分:

(filePath, source) =>
{
    g_Files[fileName].Add(fd);
    return g_Files[fileName];
}

对此:

(filePath,source) =>
{
    source.Add(fd);
    return source;
}