Easee: Ensure token refresh doesn't race

This commit is contained in:
andig 2021-11-12 11:31:09 +01:00
parent 3160f11a02
commit a43bc0ce8b

View file

@ -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