DataReaderExtensions.cs 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183
  1. // Licensed to the .NET Foundation under one or more agreements.
  2. // The .NET Foundation licenses this file to you under the MIT license.
  3. // See the LICENSE file in the project root for more information.
  4. using System.ComponentModel;
  5. using System.Data.Common;
  6. using System.IO;
  7. using System.Threading;
  8. using System.Threading.Tasks;
  9. namespace System.Data
  10. {
  11. // copied from CoreFX (NS2.1)
  12. public static class DataReaderExtensions
  13. {
  14. public static bool GetBoolean(this DbDataReader reader, string name)
  15. {
  16. AssertNotNull(reader);
  17. return reader.GetBoolean(reader.GetOrdinal(name));
  18. }
  19. public static byte GetByte(this DbDataReader reader, string name)
  20. {
  21. AssertNotNull(reader);
  22. return reader.GetByte(reader.GetOrdinal(name));
  23. }
  24. public static long GetBytes(this DbDataReader reader, string name, long dataOffset, byte[] buffer, int bufferOffset, int length)
  25. {
  26. AssertNotNull(reader);
  27. return reader.GetBytes(reader.GetOrdinal(name), dataOffset, buffer, bufferOffset, length);
  28. }
  29. public static char GetChar(this DbDataReader reader, string name)
  30. {
  31. AssertNotNull(reader);
  32. return reader.GetChar(reader.GetOrdinal(name));
  33. }
  34. public static long GetChars(this DbDataReader reader, string name, long dataOffset, char[] buffer, int bufferOffset, int length)
  35. {
  36. AssertNotNull(reader);
  37. return reader.GetChars(reader.GetOrdinal(name), dataOffset, buffer, bufferOffset, length);
  38. }
  39. [EditorBrowsable(EditorBrowsableState.Never)]
  40. public static DbDataReader GetData(this DbDataReader reader, string name)
  41. {
  42. AssertNotNull(reader);
  43. return reader.GetData(reader.GetOrdinal(name));
  44. }
  45. public static string GetDataTypeName(this DbDataReader reader, string name)
  46. {
  47. AssertNotNull(reader);
  48. return reader.GetDataTypeName(reader.GetOrdinal(name));
  49. }
  50. public static DateTime GetDateTime(this DbDataReader reader, string name)
  51. {
  52. AssertNotNull(reader);
  53. return reader.GetDateTime(reader.GetOrdinal(name));
  54. }
  55. public static decimal GetDecimal(this DbDataReader reader, string name)
  56. {
  57. AssertNotNull(reader);
  58. return reader.GetDecimal(reader.GetOrdinal(name));
  59. }
  60. public static double GetDouble(this DbDataReader reader, string name)
  61. {
  62. AssertNotNull(reader);
  63. return reader.GetDouble(reader.GetOrdinal(name));
  64. }
  65. public static Type GetFieldType(this DbDataReader reader, string name)
  66. {
  67. AssertNotNull(reader);
  68. return reader.GetFieldType(reader.GetOrdinal(name));
  69. }
  70. public static T GetFieldValue<T>(this DbDataReader reader, string name)
  71. {
  72. AssertNotNull(reader);
  73. return reader.GetFieldValue<T>(reader.GetOrdinal(name));
  74. }
  75. public static Task<T> GetFieldValueAsync<T>(this DbDataReader reader, string name, CancellationToken cancellationToken = default(CancellationToken))
  76. {
  77. AssertNotNull(reader);
  78. return reader.GetFieldValueAsync<T>(reader.GetOrdinal(name), cancellationToken);
  79. }
  80. public static float GetFloat(this DbDataReader reader, string name)
  81. {
  82. AssertNotNull(reader);
  83. return reader.GetFloat(reader.GetOrdinal(name));
  84. }
  85. public static Guid GetGuid(this DbDataReader reader, string name)
  86. {
  87. AssertNotNull(reader);
  88. return reader.GetGuid(reader.GetOrdinal(name));
  89. }
  90. public static short GetInt16(this DbDataReader reader, string name)
  91. {
  92. AssertNotNull(reader);
  93. return reader.GetInt16(reader.GetOrdinal(name));
  94. }
  95. public static int GetInt32(this DbDataReader reader, string name)
  96. {
  97. AssertNotNull(reader);
  98. return reader.GetInt32(reader.GetOrdinal(name));
  99. }
  100. public static long GetInt64(this DbDataReader reader, string name)
  101. {
  102. AssertNotNull(reader);
  103. return reader.GetInt64(reader.GetOrdinal(name));
  104. }
  105. [EditorBrowsable(EditorBrowsableState.Never)]
  106. public static Type GetProviderSpecificFieldType(this DbDataReader reader, string name)
  107. {
  108. AssertNotNull(reader);
  109. return reader.GetProviderSpecificFieldType(reader.GetOrdinal(name));
  110. }
  111. [EditorBrowsable(EditorBrowsableState.Never)]
  112. public static object GetProviderSpecificValue(this DbDataReader reader, string name)
  113. {
  114. AssertNotNull(reader);
  115. return reader.GetProviderSpecificValue(reader.GetOrdinal(name));
  116. }
  117. public static Stream GetStream(this DbDataReader reader, string name)
  118. {
  119. AssertNotNull(reader);
  120. return reader.GetStream(reader.GetOrdinal(name));
  121. }
  122. public static string GetString(this DbDataReader reader, string name)
  123. {
  124. AssertNotNull(reader);
  125. return reader.GetString(reader.GetOrdinal(name));
  126. }
  127. public static TextReader GetTextReader(this DbDataReader reader, string name)
  128. {
  129. AssertNotNull(reader);
  130. return reader.GetTextReader(reader.GetOrdinal(name));
  131. }
  132. public static object GetValue(this DbDataReader reader, string name)
  133. {
  134. AssertNotNull(reader);
  135. return reader.GetValue(reader.GetOrdinal(name));
  136. }
  137. public static bool IsDBNull(this DbDataReader reader, string name)
  138. {
  139. AssertNotNull(reader);
  140. return reader.IsDBNull(reader.GetOrdinal(name));
  141. }
  142. public static Task<bool> IsDBNullAsync(this DbDataReader reader, string name, CancellationToken cancellationToken = default(CancellationToken))
  143. {
  144. AssertNotNull(reader);
  145. return reader.IsDBNullAsync(reader.GetOrdinal(name), cancellationToken);
  146. }
  147. private static void AssertNotNull(DbDataReader reader)
  148. {
  149. if (reader == null)
  150. {
  151. throw new ArgumentNullException(nameof(reader));
  152. }
  153. }
  154. }
  155. }