Add target service in header

Author: [email protected]

cluster-autoscaler/cloudprovider/azure/azure_config.go

@@ -65,6 +65,9 @@ type Config struct {
 	// It can override the default public ARM endpoint for VMs pool scale operations.
 	ARMBaseURLForAPClient string `json:"armBaseURLForAPClient" yaml:"armBaseURLForAPClient"`
 
+	// TargetServiceForAPClient is the service name for agent pool requests when using a custom ARMBaseURLForAPClient.
+	TargetServiceForAPClient string `json:"targetServiceForAPClient" yaml:"targetServiceForAPClient"`
+
 	// Hosted (on-behalf-of) system pool configuration for automatic cluster.
 	// HostedSubscriptionID is the subscription ID of the hosted resources under AKS internal tenant.
 	HostedSubscriptionID string `json:"hostedSubscriptionID" yaml:"hostedSubscriptionID"`
@@ -190,6 +193,9 @@ func BuildAzureConfig(configReader io.Reader) (*Config, error) {
 	if _, err = assignFromEnvIfExists(&cfg.ARMBaseURLForAPClient, "ARM_BASE_URL_FOR_AP_CLIENT"); err != nil {
 		return nil, err
 	}
+	if _, err = assignFromEnvIfExists(&cfg.TargetServiceForAPClient, "TARGET_SERVICE_FOR_AP_CLIENT"); err != nil {
+		return nil, err
+	}
 	if _, err = assignFromEnvIfExists(&cfg.Cloud, "ARM_CLOUD"); err != nil {
 		return nil, err
 	}

cluster-autoscaler/cloudprovider/azure/azure_vms_pool.go

@@ -164,6 +164,7 @@ func (vmPool *VMPool) IncreaseSize(delta int) error {
 		header := make(http.Header)
 		header.Set("Target-Count", fmt.Sprintf("%d", count))
 		header.Set("SKU", fmt.Sprintf("%s", vmPool.sku))
+		header.Set("X-Target-Service", vmPool.manager.config.TargetServiceForAPClient)
 		updateCtx = policy.WithHTTPHeader(updateCtx, header)
 	}