public async ValueTask <int> StageBatch(IReadOnlyList <TimeoutData> timeouts, int batchNumber) { await MsmqQueueCreator.TruncateTable(connection, MsmqSqlConstants.TimeoutMigrationStagingTable, schema, databaseName); foreach (var timeout in timeouts) { stagingDataTable.Rows.Add(timeout.Id, timeout.Destination, timeout.State, timeout.Time, SerializeHeaders(timeout.Headers)); } await using var transaction = (SqlTransaction) await connection.BeginTransactionAsync(); using var sqlBulk = new SqlBulkCopy(connection, SqlBulkCopyOptions.Default, transaction) { DestinationTableName = MsmqSqlConstants.TimeoutMigrationStagingTable, }; try { await sqlBulk.WriteToServerAsync(stagingDataTable); await transaction.CommitAsync(); } catch (Exception e) { logger.LogError(e, $"Unable to bulk copy batch '{batchNumber}' to the staging table. Exception occured: {e.Message}"); return(0); } finally { stagingDataTable.Clear(); } return(timeouts.Count); }
async ValueTask EnsureMigrationTableExists() { try { await MsmqQueueCreator.CreateStagingQueue(connection, MsmqSqlConstants.TimeoutMigrationStagingTable, schema, connection.Database, preview : false); } catch (Exception e) { logger.LogError($"Unable to create the staging queue '{MsmqSqlConstants.TimeoutMigrationStagingTable}'. The following exception occured: {e.Message}"); throw; } }
public async ValueTask <MigrationCheckResult> AbleToMigrate(EndpointInfo endpoint) { var migrationCheckResult = new MigrationCheckResult(); try { await EnsureConnectionOpen(); } catch (Exception e) { migrationCheckResult.Problems.Add($"Unable to connect to the server or database using the provided connection string. Verify the connection string. The following exception occured: {e.Message}"); return(migrationCheckResult); } var databaseName = connection.Database; try { await MsmqQueueCreator.CreateStagingQueue(connection, MsmqSqlConstants.TimeoutMigrationStagingTable, schema, databaseName, preview : true); } catch (Exception e) { migrationCheckResult.Problems.Add($"Attempt to verify whether the timeout migration staging table '{MsmqSqlConstants.TimeoutMigrationStagingTable}' could be created during migration mode failed. The following exception occured: {e.Message}"); } var endpointDelayedTableName = MsmqSqlConstants.DelayedTableName(endpoint.EndpointName); if (!await MsmqQueueCreator .DoesDelayedDeliveryTableExist(connection, endpointDelayedTableName, schema, databaseName) .ConfigureAwait(false)) { migrationCheckResult.Problems.Add($"Could not find delayed queue table with name '{endpointDelayedTableName}' for the endpoint '{endpoint.EndpointName}'"); } return(migrationCheckResult); }
Task RemoveMigrationTable() { return(MsmqQueueCreator.DeleteStagingQueue(connection, MsmqSqlConstants.TimeoutMigrationStagingTable, schema, connection.Database)); }
public async ValueTask <int> CompleteBatch(int batchNumber) { return(await MsmqQueueCreator.MoveFromTo(connection, MsmqSqlConstants.TimeoutMigrationStagingTable, schema, endpointDelayedTableName, schema, databaseName)); }