diff --git a/pkg/azclient/factory_conf.go b/pkg/azclient/factory_conf.go index 883b191406..c7ac7d93fb 100644 --- a/pkg/azclient/factory_conf.go +++ b/pkg/azclient/factory_conf.go @@ -19,6 +19,7 @@ package azclient import ( "net/http" "sync" + "time" "github.com/Azure/azure-sdk-for-go/sdk/azcore/arm/policy" "github.com/Azure/go-armbalancer" @@ -37,6 +38,7 @@ func init() { Transport: utils.DefaultTransport, PoolSize: 100, }), + Timeout: time.Minute, } }) } diff --git a/pkg/azclient/utils/options.go b/pkg/azclient/utils/options.go index 295b611c1b..d83fbd8b44 100644 --- a/pkg/azclient/utils/options.go +++ b/pkg/azclient/utils/options.go @@ -50,6 +50,7 @@ func GetDefaultOption() *arm.ClientOptions { }, Transport: &http.Client{ Transport: DefaultTransport, + Timeout: time.Minute, }, TracingProvider: tracing.NewProvider(func(name, version string) tracing.Tracer { return tracing.NewTracer(NewOtlpSpan, nil) diff --git a/pkg/provider/azure_instance_metadata.go b/pkg/provider/azure_instance_metadata.go index c79fd6395e..5c42a99fd2 100644 --- a/pkg/provider/azure_instance_metadata.go +++ b/pkg/provider/azure_instance_metadata.go @@ -22,6 +22,7 @@ import ( "io" "net/http" "strings" + "time" "k8s.io/klog/v2" @@ -189,7 +190,7 @@ func (ims *InstanceMetadataService) getInstanceMetadata(key string) (*InstanceMe q.Add("api-version", consts.ImdsInstanceAPIVersion) req.URL.RawQuery = q.Encode() - client := &http.Client{} + client := &http.Client{Timeout: time.Minute} resp, err := client.Do(req) if err != nil { return nil, err @@ -227,7 +228,7 @@ func (ims *InstanceMetadataService) getLoadBalancerMetadata() (*LoadBalancerMeta q.Add("api-version", consts.ImdsLoadBalancerAPIVersion) req.URL.RawQuery = q.Encode() - client := &http.Client{} + client := &http.Client{Timeout: time.Minute} resp, err := client.Do(req) if err != nil { return nil, err