mirror of
https://github.com/golang/oauth2.git
synced 2025-07-21 00:00:09 +08:00
d0670ef3b1
Introduce new AuthenticationError type returned by errWrappingTokenSource.Token. The new error wrapper exposes a boolean method Temporary, identifying the underlying network error as retryable based on the following status codes: 500, 503, 408, or 429. Bump go.mod version to 1.15 refs: https://github.com/googleapis/google-api-go-client/issues/1445 Change-Id: I27c76cb0c71b918c25a640f40d0bd515b2e488fc Reviewed-on: https://go-review.googlesource.com/c/oauth2/+/403846 Reviewed-by: Cody Oss <codyoss@google.com> Reviewed-by: Tyler Bui-Palsulich <tbp@google.com>
65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
// Copyright 2022 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package google
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"golang.org/x/oauth2"
|
|
)
|
|
|
|
// AuthenticationError indicates there was an error in the authentication flow.
|
|
//
|
|
// Use (*AuthenticationError).Temporary to check if the error can be retried.
|
|
type AuthenticationError struct {
|
|
err *oauth2.RetrieveError
|
|
}
|
|
|
|
func newAuthenticationError(err error) error {
|
|
re := &oauth2.RetrieveError{}
|
|
if !errors.As(err, &re) {
|
|
return err
|
|
}
|
|
return &AuthenticationError{
|
|
err: re,
|
|
}
|
|
}
|
|
|
|
// Temporary indicates that the network error has one of the following status codes and may be retried: 500, 503, 408, or 429.
|
|
func (e *AuthenticationError) Temporary() bool {
|
|
if e.err.Response == nil {
|
|
return false
|
|
}
|
|
sc := e.err.Response.StatusCode
|
|
return sc == 500 || sc == 503 || sc == 408 || sc == 429
|
|
}
|
|
|
|
func (e *AuthenticationError) Error() string {
|
|
return e.err.Error()
|
|
}
|
|
|
|
func (e *AuthenticationError) Unwrap() error {
|
|
return e.err
|
|
}
|
|
|
|
type errWrappingTokenSource struct {
|
|
src oauth2.TokenSource
|
|
}
|
|
|
|
func newErrWrappingTokenSource(ts oauth2.TokenSource) oauth2.TokenSource {
|
|
return &errWrappingTokenSource{src: ts}
|
|
}
|
|
|
|
// Token returns the current token if it's still valid, else will
|
|
// refresh the current token (using r.Context for HTTP client
|
|
// information) and return the new one.
|
|
func (s *errWrappingTokenSource) Token() (*oauth2.Token, error) {
|
|
t, err := s.src.Token()
|
|
if err != nil {
|
|
return nil, newAuthenticationError(err)
|
|
}
|
|
return t, nil
|
|
}
|