open-nomad/client/structs/allochook.go

30 lines
579 B
Go

package structs
import (
"sync"
"github.com/hashicorp/nomad/client/pluginmanager/csimanager"
)
// AllocHookResources contains data that is provided by AllocRunner Hooks for
// consumption by TaskRunners
type AllocHookResources struct {
CSIMounts map[string]*csimanager.MountInfo
mu sync.RWMutex
}
func (a *AllocHookResources) GetCSIMounts() map[string]*csimanager.MountInfo {
a.mu.RLock()
defer a.mu.RUnlock()
return a.CSIMounts
}
func (a *AllocHookResources) SetCSIMounts(m map[string]*csimanager.MountInfo) {
a.mu.Lock()
defer a.mu.Unlock()
a.CSIMounts = m
}