static void ConfigureConnection(TransportExtensions <SqlServerTransport> transport, TransportSettings transportSettings)
{
var connectionString = transportSettings.ConnectionString
.RemoveCustomConnectionStringParts(out var customSchema, out var subscriptionsTableSetting);
var subscriptions = transport.SubscriptionSettings();
if (customSchema != null)
{
transport.DefaultSchema(customSchema);
subscriptions.SubscriptionTableName(defaultSubscriptionTableName, customSchema);
}
if (subscriptionsTableSetting != null)
{
var subscriptionsAddress = QueueAddress.Parse(subscriptionsTableSetting);
subscriptions.SubscriptionTableName(
tableName: subscriptionsAddress.Table,
schemaName: subscriptionsAddress.Schema ?? customSchema,
catalogName: subscriptionsAddress.Catalog
);
}
transport.ConnectionString(connectionString);
transport.EnableMessageDrivenPubSubCompatibilityMode();
}
public void PubSubCompatibilityModeWorks()
{
var extensions = new TransportExtensions <SqsTransport>(new SettingsHolder());
var result = extensions.EnableMessageDrivenPubSubCompatibilityMode();
Assert.IsTrue(result.GetSettings().Get <bool>("NServiceBus.Subscriptions.EnableMigrationMode"));
}
static void ConfigurePubSub(TransportExtensions <SqsTransport> transport, TransportSettings transportSettings)
{
// precaution in case we would ever use the subscription manager
transportSettings.Set("NServiceBus.AmazonSQS.DisableSubscribeBatchingOnStart", true);
// will be removed in next major
#pragma warning disable 618
transport.EnableMessageDrivenPubSubCompatibilityMode();
#pragma warning restore 618
}
static void ConfigureTransport(TransportExtensions <SqsTransport> transport, TransportSettings transportSettings)
{
var builder = new DbConnectionStringBuilder {
ConnectionString = transportSettings.ConnectionString
};
var alwaysLoadFromEnvironmentVariable = false;
if (builder.ContainsKey("AccessKeyId") || builder.ContainsKey("SecretAccessKey"))
{
PromoteEnvironmentVariableFromConnectionString(builder, "AccessKeyId", "AWS_ACCESS_KEY_ID");
PromoteEnvironmentVariableFromConnectionString(builder, "SecretAccessKey", "AWS_SECRET_ACCESS_KEY");
// if the user provided the access key and secret access key they should always be loaded from environment credentials
alwaysLoadFromEnvironmentVariable = true;
transport.ClientFactory(() => new AmazonSQSClient(new EnvironmentVariablesAWSCredentials()));
transport.ClientFactory(() => new AmazonSimpleNotificationServiceClient(new EnvironmentVariablesAWSCredentials()));
}
else
{
//See https://docs.aws.amazon.com/sdk-for-net/v3/developer-guide/net-dg-config-creds.html#creds-assign
log.Info("BasicAWSCredentials have not been supplied in the connection string. Attempting to use existing environment or IAM role credentials for SQS Client.");
}
var region = PromoteEnvironmentVariableFromConnectionString(builder, "Region", "AWS_REGION");
var awsRegion = RegionEndpoint.EnumerableAllRegions
.SingleOrDefault(x => x.SystemName == region);
if (awsRegion == null)
{
throw new ArgumentException($"Unknown region: \"{region}\"");
}
if (builder.TryGetValue("QueueNamePrefix", out var queueNamePrefix))
{
var queueNamePrefixAsString = (string)queueNamePrefix;
if (!string.IsNullOrEmpty(queueNamePrefixAsString))
{
transport.QueueNamePrefix(queueNamePrefixAsString);
}
}
if (builder.TryGetValue("TopicNamePrefix", out var topicNamePrefix))
{
var topicNamePrefixAsString = (string)topicNamePrefix;
if (!string.IsNullOrEmpty(topicNamePrefixAsString))
{
transport.TopicNamePrefix(topicNamePrefixAsString);
}
}
if (builder.TryGetValue("S3BucketForLargeMessages", out var bucketForLargeMessages))
{
var bucketForLargeMessagesAsString = (string)bucketForLargeMessages;
if (!string.IsNullOrEmpty(bucketForLargeMessagesAsString))
{
var keyPrefixAsString = string.Empty;
if (builder.TryGetValue("S3KeyPrefix", out var keyPrefix))
{
keyPrefixAsString = (string)keyPrefix;
}
var s3Settings = transport.S3(bucketForLargeMessagesAsString, keyPrefixAsString);
if (alwaysLoadFromEnvironmentVariable)
{
s3Settings.ClientFactory(() => new AmazonS3Client(new EnvironmentVariablesAWSCredentials()));
}
else
{
log.Info("BasicAWSCredentials have not been supplied in the connection string. Attempting to use existing environment or IAM role credentials for S3 Client.");
}
}
}
// precaution in case we would ever use the subscription manager
transportSettings.Set("NServiceBus.AmazonSQS.DisableSubscribeBatchingOnStart", true);
transport.EnableMessageDrivenPubSubCompatibilityMode();
//HINT: This is needed to make Core doesn't load a connection string value from the app.config.
// This prevents SQS from throwing on startup.
var connectionString = transport.GetSettings().Get("NServiceBus.TransportConnectionString");
connectionString.GetType()
.GetField("GetValue", BindingFlags.NonPublic | BindingFlags.Instance)
?.SetValue(connectionString, (Func <string>)(() => null));
transport.Transactions(TransportTransactionMode.ReceiveOnly);
}
