Client initialization option set cloud config (#169)

Azure · Jul 7, 2022 · a606bb2 · a606bb2
1 parent 5325d88
commit a606bb2
Showing 1 changed file with 16 additions and 15 deletions.
diff --git a/internal/client/client.go b/internal/client/client.go
@@ -16,6 +16,7 @@ import (
 
 type ClientBuilder struct {
 	credential azcore.TokenCredential
+	opt        *arm.ClientOptions
 }
 
 func NewClientBuilder() (*ClientBuilder, error) {
@@ -54,41 +55,41 @@ func NewClientBuilder() (*ClientBuilder, error) {
 
 	return &ClientBuilder{
 		credential: cred,
-	}, nil
-}
-
-var clientOpt = &arm.ClientOptions{
-	ClientOptions: policy.ClientOptions{
-		Telemetry: policy.TelemetryOptions{
-			ApplicationID: "aztfy",
-			Disabled:      false,
+		opt: &arm.ClientOptions{
+			ClientOptions: policy.ClientOptions{
+				Cloud: cloudCfg,
+				Telemetry: policy.TelemetryOptions{
+					ApplicationID: "aztfy",
+					Disabled:      false,
+				},
+				Logging: policy.LogOptions{
+					IncludeBody: true,
+				},
+			},
 		},
-		Logging: policy.LogOptions{
-			IncludeBody: true,
-		},
-	},
+	}, nil
 }
 
 func (b *ClientBuilder) NewResourceGroupClient(subscriptionId string) (*armresources.ResourceGroupsClient, error) {
 	return armresources.NewResourceGroupsClient(
 		subscriptionId,
 		b.credential,
-		clientOpt,
+		b.opt,
 	)
 }
 
 func (b *ClientBuilder) NewKeyvaultKeysClient(subscriptionId string) (*armkeyvault.KeysClient, error) {
 	return armkeyvault.NewKeysClient(
 		subscriptionId,
 		b.credential,
-		clientOpt,
+		b.opt,
 	)
 }
 
 func (b *ClientBuilder) NewKeyvaultSecretsClient(subscriptionId string) (*armkeyvault.SecretsClient, error) {
 	return armkeyvault.NewSecretsClient(
 		subscriptionId,
 		b.credential,
-		clientOpt,
+		b.opt,
 	)
 }