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

Remove chaos handler

Browse Source
This commit is contained in:
Abhishek Pandey 2023-12-06 21:44:31 -08:00
parent eaf252a26c
commit b8930d436d
1 changed files with 26 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
src/pkg/services/m365/api/graph/service.go
View File

@ -2,9 +2,7 @@ package graph
import ( import (
"context" "context"
"io"
"net/http" "net/http"
"strings"
"time" "time"
"github.com/Azure/azure-sdk-for-go/sdk/azidentity" "github.com/Azure/azure-sdk-for-go/sdk/azidentity"
@ -288,26 +286,26 @@ func kiotaMiddlewares(
counter *count.Bus, counter *count.Bus,
) []khttp.Middleware { ) []khttp.Middleware {
// Add chaos handler to default middleware // Add chaos handler to default middleware
chaosOpt := &khttp.ChaosHandlerOptions{ // chaosOpt := &khttp.ChaosHandlerOptions{
ChaosStrategy: khttp.Random, // ChaosStrategy: khttp.Random,
ChaosPercentage: 50, // ChaosPercentage: 50,
StatusCode: 503, // StatusCode: 503,
ResponseBody: &http.Response{ // ResponseBody: &http.Response{
Status: "Service Unavailable", // Status: "Service Unavailable",
StatusCode: 503, // StatusCode: 503,
// Retry-After header // // Retry-After header
// Header: http.Header{ // // Header: http.Header{
// "Retry-After": []string{"1"}, // // "Retry-After": []string{"1"},
// }, // // },
// Dummy body // // Dummy body
Body: io.NopCloser(strings.NewReader("service is down. deal with it")), // Body: io.NopCloser(strings.NewReader("service is down. deal with it")),
}, // },
} // }
chaosHandler, err := khttp.NewChaosHandlerWithOptions(chaosOpt) // chaosHandler, err := khttp.NewChaosHandlerWithOptions(chaosOpt)
if err != nil { // if err != nil {
panic(err) // panic(err)
} // }
retryOptions := khttp.RetryHandlerOptions{ retryOptions := khttp.RetryHandlerOptions{
ShouldRetry: func( ShouldRetry: func(
@ -325,7 +323,7 @@ func kiotaMiddlewares(
mw := []khttp.Middleware{ mw := []khttp.Middleware{
msgraphgocore.NewGraphTelemetryHandler(options), msgraphgocore.NewGraphTelemetryHandler(options),
&RetryMiddleware{ &RetryMiddleware{
MaxRetries: 9, MaxRetries: cc.maxRetries,
Delay: cc.minDelay, Delay: cc.minDelay,
}, },
// We use default kiota retry handler for 503 and 504 errors // We use default kiota retry handler for 503 and 504 errors
@ -335,7 +333,7 @@ func kiotaMiddlewares(
khttp.NewParametersNameDecodingHandler(), khttp.NewParametersNameDecodingHandler(),
khttp.NewUserAgentHandler(), khttp.NewUserAgentHandler(),
&LoggingMiddleware{}, &LoggingMiddleware{},
chaosHandler, // chaosHandler,
} }
// Optionally add concurrency limiter middleware if it has been initialized. // Optionally add concurrency limiter middleware if it has been initialized.
@ -343,14 +341,14 @@ func kiotaMiddlewares(
mw = append(mw, concurrencyLimitMiddlewareSingleton) mw = append(mw, concurrencyLimitMiddlewareSingleton)
} }
// throttler := &throttlingMiddleware{ throttler := &throttlingMiddleware{
// tf: newTimedFence(), tf: newTimedFence(),
// counter: counter, counter: counter,
// } }
mw = append( mw = append(
mw, mw,
//throttler, throttler,
&RateLimiterMiddleware{}, &RateLimiterMiddleware{},
&MetricsMiddleware{ &MetricsMiddleware{
counter: counter, counter: counter,