Skip to content

Commit bdf894b

Browse files
Copilotphilnach
andcommitted
Fix InitClientEncryption not being respected in CosmosDataSinkExtension
Co-authored-by: philnach <19275540+philnach@users.noreply.github.com>
1 parent 71f017e commit bdf894b

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

Extensions/Cosmos/Cosmos.DataTransfer.CosmosExtension/CosmosDataSinkExtension.cs

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -141,9 +141,18 @@ public async Task WriteAsync(IAsyncEnumerable<IDataItem> dataItems, IConfigurati
141141

142142
var client = CosmosExtensionServices.CreateClient(settings, DisplayName, dataSource.DisplayName);
143143

144-
Container container = settings.UseRbacAuth
145-
? await client.GetContainer(settings.Database, settings.Container).InitializeEncryptionAsync(cancellationToken)
146-
: await CreateDatabaseAndContainerAsync(client, settings, logger, cancellationToken);
144+
Container container;
145+
if (settings.UseRbacAuth)
146+
{
147+
var cosmosContainer = client.GetContainer(settings.Database, settings.Container);
148+
container = settings.InitClientEncryption
149+
? await cosmosContainer.InitializeEncryptionAsync(cancellationToken)
150+
: cosmosContainer;
151+
}
152+
else
153+
{
154+
container = await CreateDatabaseAndContainerAsync(client, settings, logger, cancellationToken);
155+
}
147156

148157
await CosmosExtensionServices.VerifyContainerAccess(container, settings.Container, logger, cancellationToken);
149158

0 commit comments

Comments
 (0)