Add command & bulkcopy timeout.

ctigeek · Oct 7, 2014 · 34109bd · 34109bd
1 parent ec195e2
commit 34109bd
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/DbSyncService/App.config b/DbSyncService/App.config
@@ -12,6 +12,7 @@
     <add key="ApiKey" value=""/>
     <add key="Sender" value=""/>
     <add key="Recipient" value=""/>
+    <add key="BulkCopyTimeoutInSeconds" value="500"/>
   </appSettings>
   <startup>
     <supportedRuntime version="v4.0" sku=".NETFramework,Version=v4.5" />

diff --git a/DbSyncService/SyncProvider/BulkLoader.cs b/DbSyncService/SyncProvider/BulkLoader.cs
@@ -20,6 +20,7 @@ public class BulkLoader
         private readonly ConnectionStringSettings SourceConnectionSetting;
         private readonly DbProviderFactory sourceDbFactory;
         private readonly SyncChangesData syncChangesData;
+        private readonly int BulkCopyTimeout;
 
         public BulkLoader(TableSet tableSet)
         {
@@ -28,6 +29,14 @@ public BulkLoader(TableSet tableSet)
             SourceConnectionSetting = ConfigurationManager.ConnectionStrings[tableSet.SourceConnectionStringName];
             syncChangesData = new SyncChangesData(tableSet);
             sourceDbFactory = DbProviderFactories.GetFactory(SourceConnectionSetting.ProviderName);
+            if (string.IsNullOrEmpty(ConfigurationManager.AppSettings["BulkCopyTimeoutInSeconds"]))
+            {
+                BulkCopyTimeout = 180;
+            }
+            else
+            {
+                BulkCopyTimeout = int.Parse(ConfigurationManager.AppSettings["BulkCopyTimeoutInSeconds"]);
+            }
         }
 
         public void TruncateDestinationTables()
@@ -59,7 +68,7 @@ public void BulkLoadDestinationTables()
                 {
                     bulkCopy.DestinationTableName = tableMap.FullyQualifiedDestinationTable;
                     bulkCopy.EnableStreaming = true;
-
+                    bulkCopy.BulkCopyTimeout = BulkCopyTimeout;
                     using (var conn = openSourceConnection())
                     {
                         var sql = "select * from " + tableMap.FullyQualifiedSourceTable + ";";
@@ -68,6 +77,7 @@ public void BulkLoadDestinationTables()
                             sql = tableMap.CustomSourceSQLForBulkLoadOnly;
                         }
                         var command = createCommand(conn, sql);
+                        command.CommandTimeout = BulkCopyTimeout;
                         using (var reader = command.ExecuteReader())
                         {
                             bulkCopy.WriteToServer(reader);