enhancement: Allow library user to provide some default GrpcChannelOp…

…tions. Signed-off-by: Mark Dinh <[email protected]>
cerbos · Aug 4, 2023 · 69b28c3 · 69b28c3
1 parent 595bdee
commit 69b28c3
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/src/Sdk/Cerbos/Sdk/Builder/CerbosClientBuilder.cs b/src/Sdk/Cerbos/Sdk/Builder/CerbosClientBuilder.cs
@@ -19,6 +19,7 @@ public sealed class CerbosClientBuilder
         private StreamReader CaCertificate { get; set; }
         private StreamReader TlsCertificate { get; set; }
         private StreamReader TlsKey { get; set; }
+        private GrpcChannelOptions GrpcChannelOptions { get; set; }
 
         private CerbosClientBuilder(string target) {
             Target = target;
@@ -55,6 +56,12 @@ public CerbosClientBuilder WithPlaygroundInstance(string playgroundInstanceId)
             return this;
         }
 
+        public CerbosClientBuilder WithGrpcChannelOptions(GrpcChannelOptions grpcChannelOptions)
+        {
+            GrpcChannelOptions = grpcChannelOptions;
+            return this;
+        }
+
         public CerbosClient Build()
         {
             if (string.IsNullOrEmpty(Target))
@@ -100,11 +107,18 @@ public CerbosClient Build()
             GrpcChannel channel;
             if (Plaintext)
             {
-                channel = GrpcChannel.ForAddress(Target);
+                if (GrpcChannelOptions != null)
+                {
+                    channel = GrpcChannel.ForAddress(Target, GrpcChannelOptions);
+                }
+                else
+                {
+                    channel = GrpcChannel.ForAddress(Target);
+                }
             }
             else
             {
-                GrpcChannelOptions grpcChannelOptions = new GrpcChannelOptions();
+                GrpcChannelOptions grpcChannelOptions = GrpcChannelOptions ?? new GrpcChannelOptions();
                 if (callCredentials != null && sslCredentials != null)
                 {
                     grpcChannelOptions.Credentials = ChannelCredentials.Create(sslCredentials, callCredentials);