mirror of
https://github.com/Luzifer/ots.git
synced 2024-10-01 01:06:09 -04:00
31 lines
502 B
Go
31 lines
502 B
Go
|
package main
|
||
|
|
|
||
|
|
import "github.com/satori/uuid"
|
||
|
|
|
||
|
|
type storageMem struct {
|
||
|
|
store map[string]string
|
||
|
|
}
|
||
|
|
|
||
|
|
func newStorageMem() storage {
|
||
|
|
return &storageMem{
|
||
|
|
store: make(map[string]string),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s storageMem) Create(secret string) (string, error) {
|
||
|
|
id := uuid.NewV4().String()
|
||
|
|
s.store[id] = secret
|
||
|
|
|
||
|
|
return id, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s storageMem) ReadAndDestroy(id string) (string, error) {
|
||
|
|
secret, ok := s.store[id]
|
||
|
|
if !ok {
|
||
|
|
return "", errSecretNotFound
|
||
|
|
}
|
||
|
|
|
||
|
|
delete(s.store, id)
|
||
|
|
return secret, nil
|
||
|
|
}
|