mirror of
https://github.com/Luzifer/ots.git
synced 2024-12-22 13:55:14 -05:00
77 lines
1.3 KiB
Go
77 lines
1.3 KiB
Go
|
// Package memory implements a pure in-memory store for secrets which
|
||
|
// is suitable for testing and should not be used for productive use
|
||
|
package memory
|
||
|
|
||
|
import (
|
||
|
"sync"
|
||
|
"time"
|
||
|
|
||
|
"github.com/Luzifer/ots/pkg/storage"
|
||
|
"github.com/gofrs/uuid"
|
||
|
)
|
||
|
|
||
|
type (
|
||
|
memStorageSecret struct {
|
||
|
Expiry time.Time
|
||
|
Secret string
|
||
|
}
|
||
|
|
||
|
storageMem struct {
|
||
|
sync.RWMutex
|
||
|
store map[string]memStorageSecret
|
||
|
}
|
||
|
)
|
||
|
|
||
|
// New creates a new In-Mem storage
|
||
|
func New() storage.Storage {
|
||
|
return &storageMem{
|
||
|
store: make(map[string]memStorageSecret),
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (s *storageMem) Count() (int64, error) {
|
||
|
s.RLock()
|
||
|
defer s.RUnlock()
|
||
|
|
||
|
return int64(len(s.store)), nil
|
||
|
}
|
||
|
|
||
|
func (s *storageMem) Create(secret string, expireIn time.Duration) (string, error) {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
var (
|
||
|
expire time.Time
|
||
|
id = uuid.Must(uuid.NewV4()).String()
|
||
|
)
|
||
|
|
||
|
if expireIn > 0 {
|
||
|
expire = time.Now().Add(expireIn)
|
||
|
}
|
||
|
|
||
|
s.store[id] = memStorageSecret{
|
||
|
Expiry: expire,
|
||
|
Secret: secret,
|
||
|
}
|
||
|
|
||
|
return id, nil
|
||
|
}
|
||
|
|
||
|
func (s *storageMem) ReadAndDestroy(id string) (string, error) {
|
||
|
s.Lock()
|
||
|
defer s.Unlock()
|
||
|
|
||
|
secret, ok := s.store[id]
|
||
|
if !ok {
|
||
|
return "", storage.ErrSecretNotFound
|
||
|
}
|
||
|
|
||
|
defer delete(s.store, id)
|
||
|
|
||
|
if !secret.Expiry.IsZero() && secret.Expiry.Before(time.Now()) {
|
||
|
return "", storage.ErrSecretNotFound
|
||
|
}
|
||
|
|
||
|
return secret.Secret, nil
|
||
|
}
|