mirror of
https://gitea.com/Lydanne/buildx.git
synced 2025-05-18 09:17:49 +08:00
64c5139ab6
Integrates vtproto into buildx. The generated files dockerfile has been modified to copy the buildkit equivalent file to ensure files are laid out in the appropriate way for imports. An import has also been included to change the grpc codec to the version in buildkit that supports vtproto. This will allow buildx to utilize the speed and memory improvements from that. Also updates the gc control options for prune. Signed-off-by: Jonathan A. Sternberg <jonathan.sternberg@docker.com>
59 lines
1.2 KiB
Go
59 lines
1.2 KiB
Go
// Copyright (c) 2021 PlanetScale Inc. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package generator
|
|
|
|
import (
|
|
"fmt"
|
|
"sort"
|
|
|
|
"google.golang.org/protobuf/compiler/protogen"
|
|
)
|
|
|
|
var defaultFeatures = make(map[string]Feature)
|
|
|
|
func findFeatures(featureNames []string) ([]Feature, error) {
|
|
required := make(map[string]Feature)
|
|
for _, name := range featureNames {
|
|
if name == "all" {
|
|
required = defaultFeatures
|
|
break
|
|
}
|
|
|
|
feat, ok := defaultFeatures[name]
|
|
if !ok {
|
|
return nil, fmt.Errorf("unknown feature: %q", name)
|
|
}
|
|
required[name] = feat
|
|
}
|
|
|
|
type namefeat struct {
|
|
name string
|
|
feat Feature
|
|
}
|
|
var sorted []namefeat
|
|
for name, feat := range required {
|
|
sorted = append(sorted, namefeat{name, feat})
|
|
}
|
|
sort.Slice(sorted, func(i, j int) bool {
|
|
return sorted[i].name < sorted[j].name
|
|
})
|
|
|
|
var features []Feature
|
|
for _, sp := range sorted {
|
|
features = append(features, sp.feat)
|
|
}
|
|
return features, nil
|
|
}
|
|
|
|
func RegisterFeature(name string, feat Feature) {
|
|
defaultFeatures[name] = feat
|
|
}
|
|
|
|
type Feature func(gen *GeneratedFile) FeatureGenerator
|
|
|
|
type FeatureGenerator interface {
|
|
GenerateFile(file *protogen.File) bool
|
|
}
|