You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
896 B
Go
48 lines
896 B
Go
package util
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/docker/buildx/store"
|
|
"github.com/google/uuid"
|
|
"github.com/pkg/errors"
|
|
"k8s.io/apiserver/pkg/storage/names"
|
|
)
|
|
|
|
func GenerateNodeName(builderName string, txn *store.Txn) (string, error) {
|
|
randomName := func() (string, error) {
|
|
u, err := uuid.NewRandom()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return names.SimpleNameGenerator.GenerateName(fmt.Sprintf("buildkit-%s-", u)), nil
|
|
}
|
|
|
|
ng, err := txn.NodeGroupByName(builderName)
|
|
if err != nil {
|
|
return randomName()
|
|
}
|
|
|
|
var name string
|
|
for i := 0; i < 6; i++ {
|
|
name, err = randomName()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
exists := func(name string) bool {
|
|
for _, n := range ng.Nodes {
|
|
if n.Name == name {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}(name)
|
|
if exists {
|
|
continue
|
|
}
|
|
return name, nil
|
|
}
|
|
|
|
return "", errors.Errorf("failed to generate random node name")
|
|
}
|