|
3 | 3 |
|
4 | 4 | namespace NHibernate.AdoNet
|
5 | 5 | {
|
6 |
| - /// <summary> Implementation of ColumnNameCache. </summary> |
| 6 | + /// <summary> Implementation of ColumnNameCache. Thread safe. </summary> |
7 | 7 | public class ColumnNameCache
|
8 | 8 | {
|
9 |
| - private readonly ReaderWriterLockSlim cacheLock = new ReaderWriterLockSlim(); |
10 |
| - private readonly Dictionary<string, int?> columnNameToIndexCache; |
| 9 | + private readonly ReaderWriterLockSlim _cacheLock = new ReaderWriterLockSlim(); |
| 10 | + private readonly Dictionary<string, int> _columnNameToIndexCache; |
11 | 11 |
|
12 | 12 | public ColumnNameCache(int columnCount)
|
13 | 13 | {
|
14 | 14 | // should *not* need to grow beyond the size of the total number of columns in the rs
|
15 |
| - columnNameToIndexCache = new Dictionary<string, int?>(columnCount); |
| 15 | + _columnNameToIndexCache = new Dictionary<string, int>(columnCount); |
16 | 16 | }
|
17 | 17 |
|
18 | 18 | public int GetIndexForColumnName(string columnName, ResultSetWrapper rs)
|
19 | 19 | {
|
20 |
| - int? cached = Read(columnName); |
21 |
| - if (cached.HasValue) |
| 20 | + int index; |
| 21 | + if (!TryRead(columnName, out index)) |
22 | 22 | {
|
23 |
| - return cached.Value; |
24 |
| - } |
25 |
| - else |
26 |
| - { |
27 |
| - int index = rs.Target.GetOrdinal(columnName); |
| 23 | + index = rs.Target.GetOrdinal(columnName); |
28 | 24 | Insert(columnName, index);
|
29 |
| - return index; |
30 | 25 | }
|
| 26 | + |
| 27 | + return index; |
31 | 28 | }
|
32 | 29 |
|
33 |
| - private int? Read(string key) |
| 30 | + private bool TryRead(string key, out int value) |
34 | 31 | {
|
35 |
| - cacheLock.EnterReadLock(); |
| 32 | + _cacheLock.EnterReadLock(); |
36 | 33 | try
|
37 | 34 | {
|
38 |
| - int? value; |
39 |
| - columnNameToIndexCache.TryGetValue(key, out value); |
40 |
| - return value; |
| 35 | + return _columnNameToIndexCache.TryGetValue(key, out value); |
41 | 36 | }
|
42 | 37 | finally
|
43 | 38 | {
|
44 |
| - cacheLock.ExitReadLock(); |
| 39 | + _cacheLock.ExitReadLock(); |
45 | 40 | }
|
46 | 41 | }
|
47 | 42 |
|
48 | 43 | private void Insert(string key, int value)
|
49 | 44 | {
|
50 |
| - cacheLock.EnterWriteLock(); |
| 45 | + _cacheLock.EnterWriteLock(); |
51 | 46 | try
|
52 | 47 | {
|
53 |
| - columnNameToIndexCache[key] = value; |
| 48 | + _columnNameToIndexCache[key] = value; |
54 | 49 | }
|
55 | 50 | finally
|
56 | 51 | {
|
57 |
| - cacheLock.ExitWriteLock(); |
| 52 | + _cacheLock.ExitWriteLock(); |
58 | 53 | }
|
59 | 54 | }
|
60 | 55 | }
|
|
0 commit comments