2019-08-14 19:02:52 +00:00
|
|
|
// Copyright 2019 Google LLC
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License"); you may not
|
|
|
|
// use this file except in compliance with the License. You may obtain a copy of
|
|
|
|
// the License at
|
|
|
|
//
|
|
|
|
// https://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
//
|
|
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
|
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
|
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
|
|
// License for the specific language governing permissions and limitations under
|
|
|
|
// the License.
|
|
|
|
package builder
|
|
|
|
|
|
|
|
import (
|
2019-09-08 21:21:14 +00:00
|
|
|
"github.com/google/nixery/config"
|
2019-08-14 19:02:52 +00:00
|
|
|
"sync"
|
|
|
|
)
|
|
|
|
|
|
|
|
type void struct{}
|
|
|
|
|
|
|
|
type BuildCache struct {
|
|
|
|
mmtx sync.RWMutex
|
2019-09-08 21:21:14 +00:00
|
|
|
mcache map[string]string
|
2019-08-14 19:02:52 +00:00
|
|
|
|
|
|
|
lmtx sync.RWMutex
|
|
|
|
lcache map[string]void
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewCache() BuildCache {
|
|
|
|
return BuildCache{
|
2019-09-08 21:21:14 +00:00
|
|
|
mcache: make(map[string]string),
|
2019-08-14 19:02:52 +00:00
|
|
|
lcache: make(map[string]void),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Has this layer hash already been seen by this Nixery instance? If
|
|
|
|
// yes, we can skip upload checking and such because it has already
|
|
|
|
// been done.
|
|
|
|
func (c *BuildCache) hasSeenLayer(hash string) bool {
|
|
|
|
c.lmtx.RLock()
|
|
|
|
defer c.lmtx.RUnlock()
|
|
|
|
_, seen := c.lcache[hash]
|
|
|
|
return seen
|
|
|
|
}
|
|
|
|
|
|
|
|
// Layer has now been seen and should be stored.
|
|
|
|
func (c *BuildCache) sawLayer(hash string) {
|
|
|
|
c.lmtx.Lock()
|
|
|
|
defer c.lmtx.Unlock()
|
|
|
|
c.lcache[hash] = void{}
|
|
|
|
}
|
|
|
|
|
2019-09-08 21:21:14 +00:00
|
|
|
// Retrieve a cached manifest if the build is cacheable and it exists.
|
|
|
|
func (c *BuildCache) manifestFromCache(src config.PkgSource, image *Image) (string, bool) {
|
|
|
|
key := src.CacheKey(image.Packages, image.Tag)
|
|
|
|
if key == "" {
|
|
|
|
return "", false
|
|
|
|
}
|
2019-08-14 19:02:52 +00:00
|
|
|
|
2019-09-08 21:21:14 +00:00
|
|
|
c.mmtx.RLock()
|
|
|
|
path, ok := c.mcache[key]
|
2019-08-14 19:02:52 +00:00
|
|
|
c.mmtx.RUnlock()
|
|
|
|
|
|
|
|
if !ok {
|
|
|
|
return "", false
|
|
|
|
}
|
|
|
|
|
2019-09-08 21:21:14 +00:00
|
|
|
return path, true
|
2019-08-14 19:02:52 +00:00
|
|
|
}
|
|
|
|
|
2019-09-08 21:21:14 +00:00
|
|
|
// Adds the result of a manifest build to the cache, if the manifest
|
|
|
|
// is considered cacheable.
|
|
|
|
func (c *BuildCache) cacheManifest(src config.PkgSource, image *Image, path string) {
|
|
|
|
key := src.CacheKey(image.Packages, image.Tag)
|
|
|
|
if key == "" {
|
|
|
|
return
|
2019-08-14 19:02:52 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
c.mmtx.Lock()
|
2019-09-08 21:21:14 +00:00
|
|
|
c.mcache[key] = path
|
2019-08-14 19:02:52 +00:00
|
|
|
c.mmtx.Unlock()
|
|
|
|
}
|