并发环境下的缓存容器性能优化（上）：不可变的哈希表
2009-11-12 00:03 by 老赵, 20994 visits我们在项目中经常会遇到这样的场景：一些信息读取开销较大，但只需要生成一次便可反复使用，因此我们会将其永久地缓存起来。例如在ASP.NET MVC中，系统会根据Controller的名称来缓存对应的元数据。这些缓存容器都有一些共同的特点，便是存储的对象数量有限（少则几十，多不过数千），但都需要在并发环境下被大量地读取，因此必须是线程安全的。那么，我们该如何设计这样的容器呢？
一个非常常见的做法是使用字典（System.Collections.Generic.Dictionary<TKey, TValue>）进行存储。字典使用了“哈希表”这个数据结构，其读取和写入的时间复杂度几乎都是O(1)，性能可谓非常之高。只不过，字典不是线程安全的，例如它在添加元素的时候可能会对内部的bucket数组进行动态调整，这显然不是一个原子操作。因此，它无法直接用户并行环境的读写，我们对它的操作必须加锁。
说起加锁，最简单的方式便是在读和写方法中使用同一个互斥体（mutex）进行lock：
object mutex = new object(); Dictionary<int, int> dict = new Dictionary<int, int>(); // read with lock lock (mutex) { var value = dict[1]; } // write with lock lock(mutex) { dict[1] = 1; }
这么做可以保证在任意时刻只有单个线程在访问字典，无论读写，这自然做到了线程安全。但是，这么做的话对于并发环境并不友好。因为一个字典是完全可以同时被多个线程“读”的，也就是说，如果我们简单使用lock便会大大降低“读”操作的可并发性，从而对性能产生负面影响。
那么我们能否对“写”操作进行lock，而让“读”操作完全自由呢？例如：
object mutex = new object(); Dictionary<int, int> dict = new Dictionary<int, int>(); // read directly var value = dict[1]; // write with lock lock(mutex) { dict[1] = 1; }
这也是不允许的。因为这样做无法避免在某个线程读取字典的同时，另一个线程正在修改字典。一旦出现读写操作同时进行的情况，字典就很可能被破坏了。因此，其实我们的要求有两点：
- 只能由单个线程写，但可以由多个线程同时读。
- 在进行读操作时，不可同时进行写操作。反之亦然。
这便是“读写锁”使用场景。只要使用“写”锁来保护修改操作，而使用“读”锁来保护读取操作，便可以让字典正确并高效地工作在并发环境下。“读写锁”在.NET框架中有着两个实现：ReaderWriterLock和ReaderWriterLockSlim。前者出现在.NET 1.x中，而后者随.NET 2.0发布。两者的区别在于前者性能低，后者性能高，因此在目前，基本上需要读写锁的场景都会建议使用ReaderWriterLockSlim（除了稍后会提到的情况）。
但是，ReaderWriterLockSlim并非不会带来开销，我们接下来的试验便要来验证这一点。首先，我们准备1000个数字，它们便是我们使用的测试数据源：
var source = Enumerable.Range(1, 1000).ToArray(); var dict = source.ToDictionary(i => i);
然后，我们使用CodeTimer测试它在三种情况下的性能：
int iteration = 10000; CodeTimer.Initialize(); CodeTimer.Time("Read Free", iteration, () => { foreach (var i in source) { var ignore = dict[i]; } }); var rwLockSlim = new ReaderWriterLockSlim(); CodeTimer.Time("ReaderWriterLockSlim", iteration, () => { foreach (var i in source) { rwLockSlim.EnterReadLock(); var ignore = dict[i]; rwLockSlim.ExitReadLock(); } }); var rwLock = new ReaderWriterLock(); CodeTimer.Time("ReaderWriterLock", iteration, () => { foreach (var i in source) { rwLock.AcquireReaderLock(0); var ignore = dict[i]; rwLock.ReleaseReaderLock(); } });
最后一种方式使用了性能较差的ReaderWriterLock，我的目的除了表现出它与ReaderWriterLockSlim的性能差距之外，还有一个原因在于.NET 3.5的ReaderWriterLockSlim类会在某些时候读取Environment.ProcesserCount，这在Medium Trust的环境下会引发SecurityException。因此您会发现，在目前的ASP.NET MVC框架中，所有需要读写锁的地方都使用了ReaderWriterLock而不是Slim组件。
试验结果如下：
Read Free
Time Elapsed: 228ms
CPU Cycles: 538,190,052
Gen 0: 0
Gen 1: 0
Gen 2: 0
ReaderWriterLockSlim
Time Elapsed: 1,161ms
CPU Cycles: 2,783,025,072
Gen 0: 0
Gen 1: 0
Gen 2: 0
ReaderWriterLock
Time Elapsed: 2,792ms
CPU Cycles: 6,682,368,396
Gen 0: 0
Gen 1: 0
Gen 2: 0
可见，尽管ReaderWriterLockSlim的性能比ReaderWriterLock要高出许多，但相对于无锁的读取还是有明显差距。这让我感到很不合算，因为这是由缓存容器的使用性质决定的。在一般情况下，这种容器都是“有限次地修改”，而几近“无限次地读取”，尤其到了程序运行后期，所以该使用的对象都已经缓存下来，字典内容更不会产生任何改变——但是，每一次读取还要继续承受ReaderWriterLockSlim的开销，这笔帐相信人人会算。那么，我们又如何能够做到“无锁”地读取呢？既然我们“锁”的目的是因为需要“修改”，那么我们使用“不可变（Immutable）”的容器是否就能有所帮助呢？为此，我向F#借来了Immutable Map。
F#中的Map类定义在Microsoft.FSharp.Core.dll中，这需要安装F# SDK for Visual Studio 2008，当然如果您直接安装了VS 2010自然也就有了相应的程序集——不过一些辅助类库（如Microsoft.FSharp.PowerPack），以及源文件（F#是个开源的语言）还是需要去之前的链接里下载一个压缩包。在项目中引用了程序集后，便可以使用Microsoft.FSharp.Collections命名空间下的FSharpMap类型了。我们也为它准备了同样的试验代码：
var map = source.Aggregate(FSharpMap<int, int>.Empty, (m, i) => m.Add(i, i)); CodeTimer.Time("Immutable Map", iteration, () => { foreach (var i in source) { var ignore = map[i]; } });
把它和ReaderWriterLockSlim相比，结果如下：
ReaderWriterLockSlim
Time Elapsed: 1,232ms
CPU Cycles: 2,963,830,380
Gen 0: 0
Gen 1: 0
Gen 2: 0
Immutable Map
Time Elapsed: 2,055ms
CPU Cycles: 4,956,894,996
Gen 0: 0
Gen 1: 0
Gen 2: 0
嗯？为什么使用Immutable Map之后，即便没有锁，性能还是不如ReaderWriterLockSlim的版本呢？阅读代码（fsharp\FSharp.Core\map.fs）之后发现，其原因在于数据结构的选取上。FSharpMap的数据结构是“AVL树”，即自平衡（self-balancing）的二叉搜索树，它的Get操作时间复杂度是O(logN)，N为节点个数——这自然比不过字典的O(1)时间复杂度了。那么，我们能否得到一个Get操作时间复杂度为O(1)的Immutable Map呢？
我在推特上提出这个问题之后不多久，许式伟大牛谈了他的看法：
我的理解中，FP中线性数据结构如数组，Hash等都会失效，所以没有O(1)的Immutable Map。
我同意这个看法。字典的高性能，在于求得key的Hash之后，可以通过数组下标“瞬间”定位到元素所在的bucket，但是在FP中无法做到这一点。不过幸好，我们用的C#并非是FP，我们有自己的“保底”方案：添加元素时将现有的元素完整复制到新的字典中，最后返回的是新的字典。例如：
public static class DictionaryExtensions { public static Dictionary<TKey, TValue> AddImmutably<TKey, TValue>( this Dictionary<TKey, TValue> source, TKey key, TValue value) { var result = source.ToDictionary(p => p.Key, p => p.Value); result.Add(key, value); return result; } }
这不就是一个“不可变的哈希表”吗？当然，对于真正可应用的缓存容器来说，还需要考虑更多一些东西。我在这里准备了两个缓存容器的基类，可用于此类容器的实现。其中ReadWriteCache类基于ReaderWriterLockSlim，而ReadFreeCache类则基于不可变的字典。那么，这两种做法的性能究竟如何，又分别适合什么样的场景，我们还有没有其他的做法呢？
关于这些问题，就留到下次再讨论吧。
哈哈. 沙发.
老赵. 你的产量怎么这么高啊. 一篇文章下来少则上千.
多则几千. 佩服.