33 lines
450 B
Go
33 lines
450 B
Go
|
package plancontext
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
bkgw "github.com/moby/buildkit/frontend/gateway/client"
|
||
|
|
)
|
||
|
|
|
||
|
|
type FS struct {
|
||
|
|
Result bkgw.Reference
|
||
|
|
}
|
||
|
|
|
||
|
|
type fsContext struct {
|
||
|
|
l sync.RWMutex
|
||
|
|
store map[ContextKey]*FS
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *fsContext) Register(fs *FS) ContextKey {
|
||
|
|
c.l.Lock()
|
||
|
|
defer c.l.Unlock()
|
||
|
|
|
||
|
|
id := hashID(fs)
|
||
|
|
c.store[id] = fs
|
||
|
|
return id
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *fsContext) Get(id ContextKey) *FS {
|
||
|
|
c.l.RLock()
|
||
|
|
defer c.l.RUnlock()
|
||
|
|
|
||
|
|
return c.store[id]
|
||
|
|
}
|