Easee: Ensure token refresh doesn't race
This commit is contained in:
parent
3160f11a02
commit
a43bc0ce8b
1 changed files with 5 additions and 1 deletions
|
|
@ -2,6 +2,7 @@ package oauth
|
|||
|
||||
import (
|
||||
"errors"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/imdario/mergo"
|
||||
|
|
@ -13,15 +14,18 @@ type TokenRefresher interface {
|
|||
}
|
||||
|
||||
type TokenSource struct {
|
||||
mu sync.Mutex
|
||||
token *oauth2.Token
|
||||
refresher TokenRefresher
|
||||
}
|
||||
|
||||
func RefreshTokenSource(token *oauth2.Token, refresher TokenRefresher) oauth2.TokenSource {
|
||||
return &TokenSource{token, refresher}
|
||||
return &TokenSource{token: token, refresher: refresher}
|
||||
}
|
||||
|
||||
func (ts *TokenSource) Token() (*oauth2.Token, error) {
|
||||
ts.mu.Lock()
|
||||
defer ts.mu.Unlock()
|
||||
var err error
|
||||
if time.Until(ts.token.Expiry) < time.Minute {
|
||||
var token *oauth2.Token
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue