CLOUDP-118714 add IsTimeoutErr func (#291)

mongodb · May 6, 2022 · 0a87693 · 0a87693
1 parent 33f2542
commit 0a87693
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/auth/device_flow.go b/auth/device_flow.go
@@ -25,6 +25,8 @@ import (
 	atlas "go.mongodb.org/atlas/mongodbatlas"
 )
 
+const authExpiredError = "DEVICE_AUTHORIZATION_EXPIRED"
+
 // DeviceCode holds information about the authorization-in-progress.
 type DeviceCode struct {
 	UserCode        string `json:"user_code"`        // UserCode is the code presented to users
@@ -146,3 +148,8 @@ func (c Config) RevokeToken(ctx context.Context, token, tokenTypeHint string) (*
 
 	return c.Do(ctx, req, nil)
 }
+
+func IsTimeoutErr(err error) bool {
+	var target *atlas.ErrorResponse
+	return errors.Is(err, ErrTimeout) || (errors.As(err, &target) && target.ErrorCode == authExpiredError)
+}