Streamlined the reading.

Author: drunkcod

DataBoss/SqlConnectionExtensions.cs

@@ -4,6 +4,7 @@
 using System;
 using System.Collections.Generic;
 using System.Linq;
+using System.Data;
 
 namespace DataBoss
 {
@@ -59,37 +60,36 @@ public static ICollection<int> InsertAndGetIdentities<T>(this SqlConnection conn
 				bulkCopy.WriteToServer(toInsert);
 
 			var columns = string.Join(",", Enumerable.Range(1, toInsert.FieldCount - 1).Select(toInsert.GetName));
-			var reader = new DbObjectReader(connection) {
-				CommandTimeout = null,
-			};
-			return reader.Read<IdRow>($@"
+			using(var cmd = connection.CreateCommand($@"
 				insert {destinationTable} with(tablock)({columns})
 				output inserted.$identity as {nameof(IdRow.Id)}
 				select {columns}
 				from {TempTableName}
 				order by [$]
 			
 				drop table {TempTableName}
-			").OrderBy(x => x.Id).Select(x => x.Id).ToList();
+			")) {
+				cmd.CommandTimeout = 0;
+				using(var reader = ObjectReader.For(cmd.ExecuteReader(CommandBehavior.SingleResult | CommandBehavior.SequentialAccess)))
+					return reader.Read<IdRow>().OrderBy(x => x.Id).Select(x => x.Id).ToList();
+			}
 		}
 
 		public static void WithCommand(this SqlConnection connection, Action<SqlCommand> useCommand) {
 			using(var cmd = connection.CreateCommand())
 				useCommand(cmd);
 		}
 
-		public static DatabaseInfo GetDatabaseInfo(this SqlConnection connection)
-		{
+		public static DatabaseInfo GetDatabaseInfo(this SqlConnection connection) {
 			var reader = new DbObjectReader(connection);
-			return reader.Single<DatabaseInfo>(
-				@"select 
-						ServerName = serverproperty('ServerName'),
-						ServerVersion = serverproperty('ProductVersion'),
-						DatabaseName = db.name,
-						DatabaseId = db.database_id,
-						CompatibilityLevel = db.compatibility_level
-					from sys.databases db where database_id = db_id()"
-				);
+			return reader.Single<DatabaseInfo>(@"
+				select 
+					ServerName = serverproperty('ServerName'),
+					ServerVersion = serverproperty('ProductVersion'),
+					DatabaseName = db.name,
+					DatabaseId = db.database_id,
+					CompatibilityLevel = db.compatibility_level
+				from sys.databases db where database_id = db_id()");
 		}
 	}
 }