larryh/corso
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Change client timeouts (#834)

Browse Source 
Change HTTP client timeout from 30 to 90 seconds
This commit is contained in:
Vaibhav Kamra 2022-09-13 13:19:08 -07:00 committed by GitHub
parent 2b7d3877a4
commit a17d00c65f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/internal/connector/graph/service_helper.go
View File

@ -5,6 +5,7 @@ import (
nethttp "net/http"
"net/http/httputil"
"os"
"time"
az "github.com/Azure/azure-sdk-for-go/sdk/azidentity"
ka "github.com/microsoft/kiota-authentication-azure-go"
@ -20,7 +21,8 @@ const (
)
// CreateAdapter uses provided credentials to log into M365 using Kiota Azure Library
// with Azure identity package.
// with Azure identity package. An adapter object is a necessary to component
// to create *msgraphsdk.GraphServiceClient
func CreateAdapter(tenant, client, secret string) (*msgraphsdk.GraphRequestAdapter, error) {
// Client Provider: Uses Secret for access to tenant-level data
cred, err := az.NewClientSecretCredential(tenant, client, secret, nil)
@ -36,18 +38,16 @@ func CreateAdapter(tenant, client, secret string) (*msgraphsdk.GraphRequestAdapt
return nil, err
}
// If the "LOG_GRAPH_REQUESTS" environment variable is not set, return
// the default client
if os.Getenv(logGraphRequestsEnvKey) == "" {
return msgraphsdk.NewGraphRequestAdapter(auth)
clientOptions := msgraphsdk.GetDefaultClientOptions()
middlewares := msgraphgocore.GetDefaultMiddlewaresWithOptions(&clientOptions)
// When true, additional logging middleware support added for http request
if os.Getenv(logGraphRequestsEnvKey) != "" {
middlewares = append(middlewares, &LoggingMiddleware{})
}
// Create a client with logging middleware
clientOptions := msgraphsdk.GetDefaultClientOptions()
defaultMiddlewares := msgraphgocore.GetDefaultMiddlewaresWithOptions(&clientOptions)
middlewares := []khttp.Middleware{&LoggingMiddleware{}}
middlewares = append(middlewares, defaultMiddlewares...)
httpClient := msgraphgocore.GetDefaultClient(&clientOptions, middlewares...)
httpClient.Timeout = time.Second * 90
return msgraphsdk.NewGraphRequestAdapterWithParseNodeFactoryAndSerializationWriterFactoryAndHttpClient(
auth, nil, nil, httpClient)