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

test

Browse Source
This commit is contained in:
Abhishek Pandey 2023-11-06 23:44:54 -08:00
parent 4f6e60558a
commit 89914d8639
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/internal/m365/graph/concurrency_middleware.go
View File

@ -11,6 +11,7 @@ import (
khttp "github.com/microsoft/kiota-http-go" khttp "github.com/microsoft/kiota-http-go"
"golang.org/x/time/rate" "golang.org/x/time/rate"
limiters "github.com/alcionai/corso/src/internal/common/limiters"
"github.com/alcionai/corso/src/pkg/count" "github.com/alcionai/corso/src/pkg/count"
"github.com/alcionai/corso/src/pkg/logger" "github.com/alcionai/corso/src/pkg/logger"
"github.com/alcionai/corso/src/pkg/path" "github.com/alcionai/corso/src/pkg/path"
@ -102,6 +103,8 @@ var (
driveLimiter = rate.NewLimiter(drivePerSecond, driveMaxCap) driveLimiter = rate.NewLimiter(drivePerSecond, driveMaxCap)
// also used as the exchange service limiter // also used as the exchange service limiter
defaultLimiter = rate.NewLimiter(defaultPerSecond, defaultMaxCap) defaultLimiter = rate.NewLimiter(defaultPerSecond, defaultMaxCap)
exchLim, _ = limiters.NewSlidingWindowLimiter(610*time.Second, 1*time.Second, 9800)
) )
type LimiterCfg struct { type LimiterCfg struct {
@ -185,10 +188,14 @@ func ctxLimiterConsumption(ctx context.Context, defaultConsumption int) int {
// calls-per-minute rate. Otherwise, the call will wait in a queue until // calls-per-minute rate. Otherwise, the call will wait in a queue until
// the next token set is available. // the next token set is available.
func QueueRequest(ctx context.Context) { func QueueRequest(ctx context.Context) {
limiter := ctxLimiter(ctx) // limiter := ctxLimiter(ctx)
consume := ctxLimiterConsumption(ctx, defaultLC) // consume := ctxLimiterConsumption(ctx, defaultLC)
if err := limiter.WaitN(ctx, consume); err != nil { // if err := limiter.WaitN(ctx, consume); err != nil {
// logger.CtxErr(ctx, err).Error("graph middleware waiting on the limiter")
// }
if err := exchLim.Wait(ctx); err != nil {
logger.CtxErr(ctx, err).Error("graph middleware waiting on the limiter") logger.CtxErr(ctx, err).Error("graph middleware waiting on the limiter")
} }
} }