Display final row count for each table

Addresses #12.
BrentOzarULTD · Nov 20, 2017 · d6ba621 · d6ba621
1 parent 8dfb1db
commit d6ba621
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 2 deletions.
diff --git a/soddi/Loaders/BulkCopyTask.cs b/soddi/Loaders/BulkCopyTask.cs
@@ -137,8 +137,8 @@ public void Process()
                 }
 
                 _bc.RowsInserted += (s, e) =>
-                    {
-                        _count += _bc.NotifyAfter;
+                {
+                        _count = e.Count;
                         e.Abort = OnRowsInserted(CopyEventType.Active);
                         if (e.Abort)
                         {

diff --git a/soddi/Loaders/MsSql/MsSqlBulkInserter.cs b/soddi/Loaders/MsSql/MsSqlBulkInserter.cs
@@ -89,6 +89,17 @@ public override void WriteToServer(IDataReader reader)
             try
             {
                 _inner.WriteToServer(reader);
+                using (var connection = new SqlConnection(_connectionString))
+                {
+                    connection.Open();
+                    using (var command = connection.CreateCommand())
+                    {
+                        command.CommandType = CommandType.Text;
+                        command.CommandText = $"SELECT COUNT_BIG(*) FROM {DestinationTableName}";
+                        var rowCount = (long) command.ExecuteScalar();
+                        OnRowsInserted(new BulkCopyEventArgs {Count = rowCount, Type = CopyEventType.Complete});
+                    }
+                }
             }
             finally
             {

diff --git a/soddi/Loaders/MySql/MySqlBulkInserter.cs b/soddi/Loaders/MySql/MySqlBulkInserter.cs
@@ -131,6 +131,7 @@ public override void WriteToServer(IDataReader reader)
                         dbTrans.Commit();
                     }
                 }
+                OnRowsInserted(new BulkCopyEventArgs {Count = totalRecords, Type = CopyEventType.Complete});
             }
         }
     }

diff --git a/soddi/Loaders/SQLite/SQLiteBulkInserter.cs b/soddi/Loaders/SQLite/SQLiteBulkInserter.cs
@@ -130,6 +130,7 @@ public override void WriteToServer(IDataReader reader)
                         dbTrans.Commit();
                     }
                 }
+                OnRowsInserted(new BulkCopyEventArgs {Count = totalRecords, Type = CopyEventType.Complete});
             }
         }
     }