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
69
70
71
72
73
74
75
76
77
78
79
|
//------------------------------------------------------------------------------
// <copyright file="SqlReferenceCollection.cs" company="Microsoft">
// Copyright (c) Microsoft Corporation. All rights reserved.
// </copyright>
// <owner current="true" primary="true">Microsoft</owner>
// <owner current="true" primary="false">Microsoft</owner>
//------------------------------------------------------------------------------
using System;
using System.Data;
using System.Data.Common;
using System.Diagnostics;
using System.Data.ProviderBase;
using System.Threading;
using System.Threading.Tasks;
namespace System.Data.SqlClient {
sealed internal class SqlReferenceCollection : DbReferenceCollection {
internal const int DataReaderTag = 1;
internal const int CommandTag = 2;
internal const int BulkCopyTag = 3;
override public void Add(object value, int tag) {
Debug.Assert(DataReaderTag == tag || CommandTag == tag || BulkCopyTag == tag, "unexpected tag?");
Debug.Assert(DataReaderTag != tag || value is SqlDataReader, "tag doesn't match object type: SqlDataReader");
Debug.Assert(CommandTag != tag || value is SqlCommand, "tag doesn't match object type: SqlCommand");
Debug.Assert(BulkCopyTag != tag || value is SqlBulkCopy, "tag doesn't match object type: SqlBulkCopy");
base.AddItem(value, tag);
}
internal void Deactivate() {
base.Notify(0);
}
internal SqlDataReader FindLiveReader(SqlCommand command) {
if (command == null) {
// if null == command, will find first live datareader
return FindItem<SqlDataReader>(DataReaderTag, (dataReader) => (!dataReader.IsClosed));
}
else {
// else will find live datareader assocated with the command
return FindItem<SqlDataReader>(DataReaderTag, (dataReader) => ((!dataReader.IsClosed) && (command == dataReader.Command)));
}
}
// Finds a SqlCommand associated with the given StateObject
internal SqlCommand FindLiveCommand(TdsParserStateObject stateObj) {
return FindItem<SqlCommand>(CommandTag, (command) => (command.StateObject == stateObj));
}
override protected void NotifyItem(int message, int tag, object value) {
Debug.Assert(0 == message, "unexpected message?");
Debug.Assert(DataReaderTag == tag || CommandTag == tag || BulkCopyTag == tag, "unexpected tag?");
if (tag == DataReaderTag) {
Debug.Assert(value is SqlDataReader, "Incorrect object type");
var rdr = (SqlDataReader)value;
if (!rdr.IsClosed) {
rdr.CloseReaderFromConnection();
}
}
else if (tag == CommandTag) {
Debug.Assert(value is SqlCommand, "Incorrect object type");
((SqlCommand)value).OnConnectionClosed();
}
else if (tag == BulkCopyTag) {
Debug.Assert(value is SqlBulkCopy, "Incorrect object type");
((SqlBulkCopy)value).OnConnectionClosed();
}
}
override public void Remove(object value) {
Debug.Assert(value is SqlDataReader || value is SqlCommand || value is SqlBulkCopy, "SqlReferenceCollection.Remove expected a SqlDataReader or SqlCommand or SqlBulkCopy");
base.RemoveItem(value);
}
}
}
|
