蓝月亮

jt

2021-06-10 5d0d028456874576560552f5a5c4e8b801786f11

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
namespace HW.Utility.Data
{
    using System;
    using System.Collections.Generic;
    using System.Data.Common;
    using System.Transactions;
 
    internal static class TransactionScopeConnections
    {
        private static Dictionary<Transaction, Dictionary<string, DbConnection>> transactionConnections = new Dictionary<Transaction, Dictionary<string, DbConnection>>();
 
        public static DbConnection GetConnection(Database db)
        {
            Dictionary<string, DbConnection> dictionary;
            DbConnection newOpenConnection;
            Transaction current = Transaction.Current;
            if (current == null)
            {
                return null;
            }
            transactionConnections.TryGetValue(current, out dictionary);
            if (dictionary != null)
            {
                dictionary.TryGetValue(db.ConnectionString.ToString(), out newOpenConnection);
                if (newOpenConnection != null)
                {
                    return newOpenConnection;
                }
            }
            else
            {
                dictionary = new Dictionary<string, DbConnection>();
                lock (transactionConnections)
                {
                    transactionConnections.Add(current, dictionary);
                }
            }
            if (dictionary.ContainsKey(db.ConnectionString))
            {
                newOpenConnection = dictionary[db.ConnectionString];
            }
            else
            {
                newOpenConnection = db.GetNewOpenConnection();
                current.TransactionCompleted += new TransactionCompletedEventHandler(TransactionScopeConnections.OnTransactionCompleted);
                dictionary.Add(db.ConnectionString, newOpenConnection);
            }
            return newOpenConnection;
        }
 
        private static void OnTransactionCompleted(object sender, TransactionEventArgs e)
        {
            Dictionary<string, DbConnection> dictionary;
            transactionConnections.TryGetValue(e.Transaction, out dictionary);
            if (dictionary != null)
            {
                lock (transactionConnections)
                {
                    transactionConnections.Remove(e.Transaction);
                }
                foreach (DbConnection connection in dictionary.Values)
                {
                    connection.Dispose();
                }
            }
        }
    }
}