peridot/base/go/forge/caching.go
2023-10-04 02:55:49 +02:00

50 lines
1.1 KiB
Go

// Copyright 2023 Peridot Authors
//
// 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
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package forge
import "time"
type Cacher struct {
Forge
authenticator *Authenticator
}
func NewCacher(f Forge) *Cacher {
return &Cacher{
Forge: f,
}
}
func (c *Cacher) GetAuthenticator() (*Authenticator, error) {
if c.authenticator != nil {
// Check if the token is expired, if not return it
if !c.authenticator.Expires.Before(time.Now()) {
return c.authenticator, nil
}
}
// Otherwise, get a new token
c.authenticator = nil
a, err := c.Forge.GetAuthenticator()
if err != nil {
return nil, err
}
c.authenticator = a
return a, nil
}