59 lines
1.7 KiB
Go
59 lines
1.7 KiB
Go
// Copyright (C) MongoDB, Inc. 2022-present.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
// not use this file except in compliance with the License. You may obtain
|
|
// a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
package internal
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
)
|
|
|
|
type timeoutKey struct{}
|
|
|
|
// MakeTimeoutContext returns a new context with Client-Side Operation Timeout (CSOT) feature-gated behavior
|
|
// and a Timeout set to the passed in Duration. Setting a Timeout on a single operation is not supported in
|
|
// public API.
|
|
//
|
|
// TODO(GODRIVER-2348) We may be able to remove this function once CSOT feature-gated behavior becomes the
|
|
// TODO default behavior.
|
|
func MakeTimeoutContext(ctx context.Context, to time.Duration) (context.Context, context.CancelFunc) {
|
|
// Only use the passed in Duration as a timeout on the Context if it
|
|
// is non-zero.
|
|
cancelFunc := func() {}
|
|
if to != 0 {
|
|
ctx, cancelFunc = context.WithTimeout(ctx, to)
|
|
}
|
|
return context.WithValue(ctx, timeoutKey{}, true), cancelFunc
|
|
}
|
|
|
|
func IsTimeoutContext(ctx context.Context) bool {
|
|
return ctx.Value(timeoutKey{}) != nil
|
|
}
|
|
|
|
// ZeroRTTMonitor implements the RTTMonitor interface and is used internally for testing. It returns 0 for all
|
|
// RTT calculations and an empty string for RTT statistics.
|
|
type ZeroRTTMonitor struct{}
|
|
|
|
// EWMA implements the RTT monitor interface.
|
|
func (zrm *ZeroRTTMonitor) EWMA() time.Duration {
|
|
return 0
|
|
}
|
|
|
|
// Min implements the RTT monitor interface.
|
|
func (zrm *ZeroRTTMonitor) Min() time.Duration {
|
|
return 0
|
|
}
|
|
|
|
// P90 implements the RTT monitor interface.
|
|
func (zrm *ZeroRTTMonitor) P90() time.Duration {
|
|
return 0
|
|
}
|
|
|
|
// Stats implements the RTT monitor interface.
|
|
func (zrm *ZeroRTTMonitor) Stats() string {
|
|
return ""
|
|
}
|