50 lines
880 B
Go
50 lines
880 B
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package docker
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/hashicorp/go-set"
|
|
)
|
|
|
|
type taskStore struct {
|
|
store map[string]*taskHandle
|
|
lock sync.RWMutex
|
|
}
|
|
|
|
func newTaskStore() *taskStore {
|
|
return &taskStore{store: map[string]*taskHandle{}}
|
|
}
|
|
|
|
func (ts *taskStore) Set(id string, handle *taskHandle) {
|
|
ts.lock.Lock()
|
|
defer ts.lock.Unlock()
|
|
ts.store[id] = handle
|
|
}
|
|
|
|
func (ts *taskStore) Get(id string) (*taskHandle, bool) {
|
|
ts.lock.RLock()
|
|
defer ts.lock.RUnlock()
|
|
t, ok := ts.store[id]
|
|
return t, ok
|
|
}
|
|
|
|
func (ts *taskStore) IDs() *set.Set[string] {
|
|
ts.lock.RLock()
|
|
defer ts.lock.RUnlock()
|
|
|
|
s := set.New[string](len(ts.store))
|
|
for _, handle := range ts.store {
|
|
s.Insert(handle.containerID)
|
|
}
|
|
return s
|
|
}
|
|
|
|
func (ts *taskStore) Delete(id string) {
|
|
ts.lock.Lock()
|
|
defer ts.lock.Unlock()
|
|
delete(ts.store, id)
|
|
}
|