Add proxy support to the Cosmos Extension

AzureCosmosDB · Jun 7, 2024 · e4ca5dc · e4ca5dc
1 parent 3fc3f32
commit e4ca5dc
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/Extensions/Cosmos/Cosmos.DataTransfer.CosmosExtension/CosmosExtensionServices.cs b/Extensions/Cosmos/Cosmos.DataTransfer.CosmosExtension/CosmosExtensionServices.cs
@@ -6,6 +6,7 @@
 using System.Reflection;
 using Azure.Core;
 using System.Text.RegularExpressions;
+using System.Net;
 
 namespace Cosmos.DataTransfer.CosmosExtension
 {
@@ -31,6 +32,10 @@ public static CosmosClient CreateClient(CosmosSettingsBase settings, string disp
                 EnableContentResponseOnWrite = false,
                 Serializer = cosmosSerializer,
             };
+
+            if (!string.IsNullOrEmpty(settings.WebProxy)){
+                clientOptions.WebProxy = new WebProxy(settings.WebProxy);
+            }
 
             CosmosClient? cosmosClient;
             if (settings.UseRbacAuth)

diff --git a/Extensions/Cosmos/Cosmos.DataTransfer.CosmosExtension/CosmosSettingsBase.cs b/Extensions/Cosmos/Cosmos.DataTransfer.CosmosExtension/CosmosSettingsBase.cs
@@ -11,7 +11,7 @@ public abstract class CosmosSettingsBase : IValidatableObject
         [Required]
         public string? Container { get; set; }
         public ConnectionMode ConnectionMode { get; set; } = ConnectionMode.Gateway;
-
+        public string? WebProxy { get; set; }
         public bool UseRbacAuth { get; set; }
         public string? AccountEndpoint { get; set; }
         public bool EnableInteractiveCredentials { get; set; } = true;