34 lines
584 B
Go
34 lines
584 B
Go
|
package mock
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
type taskStore struct {
|
||
|
|
store map[string]*mockTaskHandle
|
||
|
|
lock sync.RWMutex
|
||
|
|
}
|
||
|
|
|
||
|
|
func newTaskStore() *taskStore {
|
||
|
|
return &taskStore{store: map[string]*mockTaskHandle{}}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ts *taskStore) Set(id string, handle *mockTaskHandle) {
|
||
|
|
ts.lock.Lock()
|
||
|
|
defer ts.lock.Unlock()
|
||
|
|
ts.store[id] = handle
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ts *taskStore) Get(id string) (*mockTaskHandle, bool) {
|
||
|
|
ts.lock.RLock()
|
||
|
|
defer ts.lock.RUnlock()
|
||
|
|
t, ok := ts.store[id]
|
||
|
|
return t, ok
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ts *taskStore) Delete(id string) {
|
||
|
|
ts.lock.Lock()
|
||
|
|
defer ts.lock.Unlock()
|
||
|
|
delete(ts.store, id)
|
||
|
|
}
|