-
Notifications
You must be signed in to change notification settings - Fork 0
/
storage.go
69 lines (61 loc) · 1.47 KB
/
storage.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
package storage
import (
"sync"
)
// Storage is a simple object that provides thread safe
// methods to read and write into a map.
type Storage struct {
data map[string]map[string]interface{}
mux sync.RWMutex
}
// New returns an instance of Storage.
func New() *Storage {
return &Storage{
data: make(map[string]map[string]interface{}),
mux: sync.RWMutex{},
}
}
// Set writes a value interface to a string key.
func (s *Storage) Set(eventID, key string, value interface{}) {
s.mux.Lock()
defer s.mux.Unlock()
if s.data[eventID] == nil {
s.data[eventID] = make(map[string]interface{})
}
s.data[eventID][key] = value
}
// Get reads value by a key.
func (s *Storage) Get(eventID string, key string) interface{} {
s.mux.RLock()
defer s.mux.RUnlock()
if s.data[eventID] == nil {
return nil
}
return s.data[eventID][key]
}
// ListEventVariables returns the slice of variables created for EventID.
func (s *Storage) ListEventVariables(eventID string) []string {
s.mux.RLock()
defer s.mux.RUnlock()
list := []string{}
for k := range s.data[eventID] {
list = append(list, k)
}
return list
}
// ListEventIDs returns the list of stored event IDs.
func (s *Storage) ListEventIDs() []string {
s.mux.RLock()
defer s.mux.RUnlock()
list := []string{}
for k := range s.data {
list = append(list, k)
}
return list
}
// Flush removes variables by their parent event ID.
func (s *Storage) Flush(eventID string) {
s.mux.Lock()
defer s.mux.Unlock()
delete(s.data, eventID)
}