/
DataReader.cs
99 lines (78 loc) · 2.31 KB
/
DataReader.cs
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
using System;
using System.Collections.Generic;
using System.Data;
using System.Data.Common;
using System.Diagnostics;
using System.Diagnostics.CodeAnalysis;
using System.Linq;
namespace LinqToDB.Data
{
public class DataReader : IDisposable
{
internal DataReaderWrapper? ReaderWrapper { get; private set; }
public DbDataReader? Reader => ReaderWrapper?.DataReader;
public CommandInfo? CommandInfo { get; }
internal int ReadNumber { get; set; }
private DateTime StartedOn { get; } = DateTime.UtcNow;
private Stopwatch Stopwatch { get; } = Stopwatch.StartNew();
public DataReader(CommandInfo commandInfo, DataReaderWrapper dataReader)
{
CommandInfo = commandInfo;
ReaderWrapper = dataReader;
}
public void Dispose()
{
if (ReaderWrapper != null)
{
if (CommandInfo?.DataConnection.TraceSwitchConnection.TraceInfo == true)
{
CommandInfo.DataConnection.OnTraceConnection(new TraceInfo(CommandInfo.DataConnection, TraceInfoStep.Completed, TraceOperation.ExecuteReader, false)
{
TraceLevel = TraceLevel.Info,
Command = ReaderWrapper.Command,
StartTime = StartedOn,
ExecutionTime = Stopwatch.Elapsed,
RecordsAffected = ReadNumber,
});
}
ReaderWrapper.Dispose();
ReaderWrapper = null;
}
}
#region Query with object reader
public IEnumerable<T> Query<T>(Func<DbDataReader, T> objectReader)
{
while (Reader!.Read())
yield return objectReader(Reader);
}
#endregion
#region Query
public IEnumerable<T> Query<T>()
{
if (ReadNumber != 0)
if (!Reader!.NextResult())
return Enumerable.Empty<T>();
ReadNumber++;
return CommandInfo!.ExecuteQuery<T>(Reader!, FormattableString.Invariant($"{CommandInfo.CommandText}$$${ReadNumber}"));
}
#endregion
#region Query with template
public IEnumerable<T> Query<T>(T template)
{
return Query<T>();
}
#endregion
#region Execute scalar
[return: MaybeNull]
public T Execute<T>()
{
if (ReadNumber != 0)
if (!Reader!.NextResult())
return default(T);
ReadNumber++;
var sql = FormattableString.Invariant($"{CommandInfo!.CommandText}$$${ReadNumber}");
return CommandInfo.ExecuteScalar<T>(Reader!, sql);
}
#endregion
}
}